U e @sZdZddlZddlZddlmZddlZedZGdddZGdddZ ddgZ dS) z@Default classes for Comm and CommManager, for usage in IPython. N) import_itemZCommc sreZdZdZdfdd 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  stjf| |r|ntj|_||_||_||_|r<|nd|j d|_ | rV| ni|_ | rd| ni|_ d|_ d|_d|_|jr|j|||dnd|_dS)Nzcomm-%sasciiTdatametadatabuffersF)super__init__uuiduuid4hexcomm_idprimary target_name target_moduleencodetopic _open_data _close_data _msg_callback_close_callback_closedopen) selfrrrr rrrrrrkwargs __class__Q/home/ankuromar296_gmail_com/.local/lib/python3.8/site-packages/comm/base_comm.pyr szBaseComm.__init__cKs tddS)Nz*publish_msg Comm method is not implemented)NotImplementedError)rmsg_typerrr keysrrr publish_msg<szBaseComm.publish_msgcCs|jdddS)ztrigger close on gcT)deletingN)closerrrr __del__?szBaseComm.__del__cCsz|dkr|j}t}|dkr&td||z$|jd||||j|jdd|_Wn t k rt| |YnXdS)z+Open the frontend-side version of this commNz.Comms cannot be opened without a comm_manager. comm_open)rrr rrF) rrget_comm_manager RuntimeError register_commr$rrr Exceptionunregister_comm)rrrr Z comm_managerrrr rEs&   z BaseComm.openFcCsF|jr dSd|_|dkr|j}|jd|||d|sBt|dS)z,Close the frontend-side version of this commNT comm_closer)rrr$rr*r.)rrrr r%rrr r&]szBaseComm.closecCs|jd|||ddS)z8Send a message to the frontend-side version of this commcomm_msgrN)r$)rrrr rrr sendos z BaseComm.sendcCs ||_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)rrcallbackrrr on_closezszBaseComm.on_closecCs ||_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)rr2rrr on_msgszBaseComm.on_msgcCs$td|j||jr ||dS)zHandle a comm_close messagezhandle_close[%s](%s)N)loggerdebugrr)rmsgrrr handle_closeszBaseComm.handle_closecCsVtd|j||jrRddlm}|}|r8|jd|||rR|jddS)zHandle a comm_msg messagezhandle_msg[%s](%s)r) get_ipythonZ pre_executeZ post_executeN)r6r7rrZIPythonr:eventstrigger)rr8r:shellrrr handle_msgs   zBaseComm.handle_msg) rNNNNTNNNN)NNN)NNN)NNNF)NNN)__name__ __module__ __qualname____doc__r r$r(rr&r1r4r5r9r> __classcell__rrrr 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'rrr r szCommManager.__init__cCs t|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) isinstancestrrrFrrfrrr register_targets zCommManager.register_targetcCs |j|S)z5Unregister a callable registered with register_target)rFpoprIrrr unregister_targetszCommManager.unregister_targetcCs|j}||j|<|S)zRegister a new comm)rrE)rrrrrr r,s zCommManager.register_commcCs|j|j}dS)z,Unregister a comm, and close its counterpartN)rErLr)rrrrr r.szCommManager.unregister_commc CsTz |j|WStk rNtd|ttjrJtdt|j YnXdS)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) rEKeyErrorr6warning isEnabledForloggingDEBUGr7listr#)rrrrr get_comms    zCommManager.get_commc Csddlm}|d}|d}|d}|j|d}||d|d} || |dkr`td |n6z|| |WdStk rtjd |d d YnXz | Wn"tk rtjd d d YnXdS)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.) rrUrFgetr,r6errorr-r&) rstreamidentr8rUrVrrrJrrrr r)s2    zCommManager.comm_opencCs^|d}|d}||}|dkr&dSz||Wn$tk rXtjd|ddYnXdS)zHandler for comm_msg messagesrVrNzException in comm_msg for %sTrW)rTr>r-r6rZrr[r\r8rVrrrrr r0s zCommManager.comm_msgcCsr|d}|d}||}|dkr&dSd|j|_|j|=z||Wn$tk rltjd|ddYnXdS)zHandler for comm_close messagesrVrNTzException in comm_close for %srW)rTrErr9r-r6rZr]rrr r/ s  zCommManager.comm_closeN) r?r@rArBr rKrMr,r.rTr)r0r/rrrr rDs! rD) rBrQr Ztraitlets.utils.importstringrr getLoggerr6rrD__all__rrrr s  x