o i!@sZdZddlZddlZddlZddlmZedZGdddZGdddZ ddgZ dS) z@Default classes for Comm and CommManager, for usage in IPython. N) import_itemCommcseZdZdZ          dfdd ZdddZd d Zdd d ZdddZdddZ ddZ ddZ ddZ ddZ ZS)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  stt|jdi| |r|ntj|_||_||_||_ |r"|nd|j d|_ | r/| ni|_ | r6| ni|_ d|_d|_d|_|jrO|j|||ddSd|_dS)Nzcomm-%sasciiTdatametadatabuffersF)superr__init__uuiduuid4hexcomm_idprimary target_name target_moduleencodetopic _open_data _close_data _msg_callback_close_callback_closedopen) selfrrr r rrrrrrkwargs __class__r I/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/comm/base_comm.pyr s zBaseComm.__init__cKstd)Nz*publish_msg Comm method is not implemented)NotImplementedError)rmsg_typerr r keysr r r! publish_msg<szBaseComm.publish_msgcCs|jdddS)ztrigger close on gcT)deletingN)closerr r r!__del__?szBaseComm.__del__cCsr|dur|j}t}|durtd||z|jd||||j|jdd|_WdSt y8| |w)z+Open the frontend-side version of this commNz.Comms cannot be opened without a comm_manager. comm_open)rr r rrF) rrget_comm_manager RuntimeError register_commr%rrr Exceptionunregister_comm)rrr r comm_managerr r r!rEs(    z BaseComm.openFcCsJ|jrdSd|_|dur|j}|jd|||d|s#t|dSdS)z,Close the frontend-side version of this commNT comm_closer)rrr%rr+r/)rrr r r&r r r!r']szBaseComm.closecCs|jd|||ddS)z8Send a message to the frontend-side version of this commcomm_msgrN)r%)rrr r r r r!sendos  z BaseComm.sendcC ||_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)rrcallbackr r r!on_closez zBaseComm.on_closecCr4)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)rr5r r r!on_msgr8zBaseComm.on_msgcCs(td|j||jr||dSdS)zHandle a comm_close messagezhandle_close[%s](%s)N)loggerdebugrr)rmsgr r r! handle_closeszBaseComm.handle_closecCs^td|j||jr+ddlm}|}|r|jd|||r-|jddSdSdS)zHandle a comm_msg messagezhandle_msg[%s](%s)r) get_ipython pre_execute post_executeN)r:r;rrIPythonr>eventstrigger)rr<r>shellr r r! handle_msgs   zBaseComm.handle_msg) rNNNNTNNNN)NNN)NNNF)__name__ __module__ __qualname____doc__r r%r)rr'r3r7r9r=rE __classcell__r r rr!rs, $      rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS) CommManagerzDDefault CommManager singleton implementation for Comms in the KernelcCsi|_i|_dS)N)commstargetsr(r r r!r s zCommManager.__init__cCs t|tr t|}||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) isinstancestrrrMrrfr r r!register_targets zCommManager.register_targetcCs |j|S)z5Unregister a callable registered with register_target)rMpoprPr r r!unregister_targets zCommManager.unregister_targetcCs|j}||j|<|S)zRegister a new comm)rrL)rrrr r r!r-s zCommManager.register_commcCs|j|j}dS)z,Unregister a comm, and close its counterpartN)rLrSr)rrr r r!r/szCommManager.unregister_commc CsVz|j|WSty*td|ttjr'tdt|j YdSYdSw)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) rLKeyErrorr:warning isEnabledForloggingDEBUGr;listr$)rrr r r!get_comms    zCommManager.get_commc Csddlm}|d}|d}|d}|j|d}||d|d} || |dur0td |nz|| |WdStyItjd |d d Ynwz| WdStybtjd d d YdSw)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.) rr\rMgetr-r:errorr.r') rstreamidentr<r\r]rrrQrr r r!r*s:       zCommManager.comm_opencCs\|d}|d}||}|durdSz||WdSty-tjd|ddYdSw)zHandler for comm_msg messagesr]rNzException in comm_msg for %sTr^)r[rEr.r:rarrbrcr<r]rrr r r!r2s  zCommManager.comm_msgcCsp|d}|d}||}|durdSd|j|_|j|=z||WdSty7tjd|ddYdSw)zHandler for comm_close messagesr]rNTzException in comm_close for %sr^)r[rLrr=r.r:rardr r r!r1 s   zCommManager.comm_closeN) rFrGrHrIr rRrTr-r/r[r*r2r1r r r r!rKs# rK) rIrrXrtraitlets.utils.importstringr getLoggerr:rrK__all__r r r r!s   z