bc!dZddlZddlZddlZddlmZejdZGddZGddZ ddgZ dS) z@Default classes for Comm and CommManager, for usage in IPython. N) import_itemCommcveZdZdZ dfd ZddZdZddZdd Zdd Z d Z d Z dZ dZ xZS)BaseCommzClass for communicating between a Frontend and a Kernel Must be subclassed with a publish_msg method implementation which sends comm messages through the iopub channel. commNTc  tt|jdi| |r|ntjj|_||_||_||_ |r|nd|jz d|_ | r| ni|_ | r| ni|_ d|_d|_d|_|jr||||dSd|_dS)Nzcomm-%sasciiTdatametadatabuffersF)superr__init__uuiduuid4hexcomm_idprimary target_name target_moduleencodetopic _open_data _close_data _msg_callback_close_callback_closedopen) selfrr r r rrrrrrkwargs __class__s .lib/python3.11/site-packages/comm/base_comm.pyrzBaseComm.__init__s 'h&00000")?wwtz||/?  &*#SUU)dl*B)J)J7)S)S (2:***5=;;2!# < ! II4(GI D D D D D DLLLc td)Nz*publish_msg Comm method is not implemented)NotImplementedError)r msg_typer r r keyss r# publish_msgzBaseComm.publish_msg<s!"NOOOr$c2|ddS)ztrigger close on gcT)deletingN)closer s r#__del__zBaseComm.__del__?s D !!!!!r$c,||j}tj}|td|| |d||||j|jd|_dS#t$r| |wxYw)z+Open the frontend-side version of this commNz.Comms cannot be opened without a comm_manager. comm_open)r r r rrF) rrget_comm_manager RuntimeError register_commr)rrr Exceptionunregister_comm)r r r r comm_managers r#rz BaseComm.openEs  #?D,..  QOPP P""4(((    ! ,"0    !DLLL     ( ( . . .  s ,A22!BFc|jrdSd|_||j}|d||||s(tj|dSdS)z,Close the frontend-side version of this commNT comm_closer )rrr)rr1r5)r r r r r+s r#r,zBaseComm.close]s <  F  $#D       :  ! # # 3 3D 9 9 9 9 9 : :r$c8|d|||dS)z8Send a message to the frontend-side version of this commcomm_msgr N)r))r r r r s r#sendz BaseComm.sendos5        r$c||_dS)zRegister a callback for comm_close Will be called with the `data` of the close message. Call `on_close(None)` to disable an existing callback. N)rr callbacks r#on_closezBaseComm.on_closezs (r$c||_dS)zRegister a callback for comm_msg Will be called with the `data` of any comm_msg messages. Call `on_msg(None)` to disable an existing callback. N)rr=s r#on_msgzBaseComm.on_msgs&r$ctd|j||jr||dSdS)zHandle a comm_close messagezhandle_close[%s](%s)N)loggerdebugrr)r msgs r# handle_closezBaseComm.handle_closesJ +T\3???   &   % % % % % & &r$ctd|j||jr_ddlm}|}|r|jd|||r|jddSdSdS)zHandle a comm_msg messagezhandle_msg[%s](%s)r) get_ipython pre_execute post_executeN)rCrDrrIPythonrHeventstrigger)r rErHshells r# handle_msgzBaseComm.handle_msgs )4<===   5 + + + + + +KMME 4 $$]333   s # # # 5 $$^44444 5 5 5 5r$) rNNNNTNNNN)NNN)NNNF)__name__ __module__ __qualname____doc__rr)r.rr,r;r?rArFrO __classcell__)r"s@r#rrs "!"!"!"!"!"!HPPPP""" 0::::$    (((&&&&&& 5 5 5 5 5 5 5r$rcHeZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) CommManagerzDDefault CommManager singleton implementation for Comms in the Kernelc"i|_i|_dS)N)commstargetsr-s r#rzCommManager.__init__s  r$cbt|trt|}||j|<dS)aRegister a callable f for a given target name f will be called with two arguments when a comm_open message is received with `target`: - the Comm instance - the `comm_open` message itself. f can be a Python callable or an import string for one. N) isinstancestrrrYr rfs r#register_targetzCommManager.register_targets3 a   AA$% [!!!r$c6|j|S)z5Unregister a callable registered with register_target)rYpopr]s r#unregister_targetzCommManager.unregister_targets| ,,,r$c(|j}||j|<|S)zRegister a new comm)rrX)r rrs r#r3zCommManager.register_comms," 7r$cD|j|j}dS)z,Unregister a comm, and close its counterpartN)rXrar)r rs r#r5zCommManager.unregister_commsz~~dl++r$cB |j|S#t$rtd|tt jrBtdt|j YdSYdSwxYw)zGet a comm with a particular id Returns the comm if found, otherwise None. This will not raise an error, it will log messages if the comm cannot be found. zNo such comm: %szCurrent comms: %sN) rXKeyErrorrCwarning isEnabledForloggingDEBUGrDlistr()r rs r#get_commzCommManager.get_comms K:g& & K K K NN-w 7 7 7""7=11 K 0$tz7H7H2I2IJJJJJJ K K K Ks BBBcddlm}|d}|d}|d}|j|d}||d|} || |t d |n< || |dS#t$r t d |d YnwxYw | jdS#t$r t d d YdSwxYw)zHandler for comm_open messagesr) create_commcontentrrNF)rrrz"No such comm target registered: %sz&Exception opening comm with target: %sTexc_infozqCould not close comm during `comm_open` failure clean-up. The comm may not have been opened yet.) rrnrYgetr3rCerrorr4r,) r streamidentrErnrorrr^rs r#r0zCommManager.comm_opensb$$$$$$i.)$m, L  [$ / /{#    4    LL={ K K K K $     r|s   444444  6 " "N5N5N5N5N5N5N5N5bwSwSwSwSwSwSwSwSt * %r$