o BZh'@sdZddlmZddlZddlZddlZddlZddlm Z ddl Z ej r+ddl m Z edZejeejfZejejeejfZejejeZejegdfZejdegdfZGdddZGd d d Zd dgZdS) z@Default classes for Comm and CommManager, for usage in IPython. ) annotationsN) import_item) ZMQStreamZCommBaseCommcseZdZdZ          d5d6fdd Z   d7d8ddZd9dd Z d7d:d!d"Z    #d;dd1d2Z d>d3d4Z ZS)?rzClass 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. commNT target_namestrdata MaybeDictmetadatabuffers BuffersTypecomm_id str | Noneprimarybool target_moduletopic bytes | None _open_data _close_datakwargst.AnyreturnNonec  stjdi| |r |ntj|_||_||_||_|r |nd|j d|_ | r-| ni|_ | r4| ni|_ d|_ d|_d|_|jrM|j|||ddSd|_dS)Nzcomm-%sasciiTr r r F)super__init__uuiduuid4hexrrrrencoderrr _msg_callback_close_callback_closedopen) selfrr r r rrrrrrr __class__rE/var/www/html/lang_env/lib/python3.10/site-packages/comm/base_comm.pyr$s zBaseComm.__init__msg_typekeyscKs d}t|)Nz*publish_msg Comm method is not implemented)NotImplementedError)r(r,r r r r-msgrrr+ publish_msgHszBaseComm.publish_msgcCs<tt|jddWddS1swYdS)ztrigger close on gcT)deletingN) contextlibsuppress Exceptioncloser(rrr+__del__Ss "zBaseComm.__del__cCsv|dur|j}t}|durd}t|||z|jd||||j|jdd|_WdSt y:| |w)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_commr0rrr&r4unregister_comm)r(r r r Z comm_managerr/rrr+r'\s*    z BaseComm.openFr1cCsJ|jrdSd|_|dur|j}|jd|||d|s#t|dSdS)z,Close the frontend-side version of this commNT comm_closer)r&rr0rr9r<)r(r r r r1rrr+r5wszBaseComm.closecCs|jd|||ddS)z8Send a message to the frontend-side version of this commcomm_msgrN)r0)r(r r r rrr+sends  z BaseComm.sendcallbackCommCallback | NonecC ||_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)r%r(r@rrr+on_close zBaseComm.on_closecCrB)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)r$rCrrr+on_msgrEzBaseComm.on_msgr/ MessageTypecCs(td|j||jr||dSdS)zHandle a comm_close messagezhandle_close[%s](%s)N)loggerdebugrr%)r(r/rrr+ handle_closeszBaseComm.handle_closecCs^td|j||jr+ddlm}|}|r|jd|||r-|jddSdSdS)zHandle a comm_msg messagezhandle_msg[%s](%s)r) get_ipythonZ pre_executeZ post_executeN)rHrIrr$ZIPythonrKeventstrigger)r(r/rKshellrrr+ handle_msgs   zBaseComm.handle_msg) rNNNNTNNNN)rrr r r r r r rrrrrrrrrr rr rrrr)NNN) r,rr r r r r r r-rrrrr)r r r r r r rr)NNNF) r r r r r r r1rrr)r@rArr)r/rGrr)__name__ __module__ __qualname____doc__rr0r7r'r5r?rDrFrJrO __classcell__rrr)r+rs>'    c@sjeZdZdZd%ddZd&d d Zd'd dZd(ddZd)ddZd*ddZ d+ddZ d+d d!Z d+d"d#Z d$S), CommManagerzDDefault CommManager singleton implementation for Comms in the KernelrrcCsi|_i|_dS)N)commstargetsr6rrr+rs zCommManager.__init__rrfCommTargetCallback | strcCs(t|tr t|}tt||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) isinstancerrtcastCommTargetCallbackrXr(rrYrrr+register_targets zCommManager.register_targetr^cCs |j|S)z5Unregister a callable registered with register_target)rXpopr_rrr+unregister_targets zCommManager.unregister_targetrrcCs|j}||j|<|S)zRegister a new comm)rrW)r(rrrrr+r;s zCommManager.register_commcCs|j|j}dS)z,Unregister a comm, and close its counterpartN)rWrar)r(rrrr+r<szCommManager.unregister_commrBaseComm | Nonec CsPz|j|WSty'td|ttjr$tdt|j YdSw)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) rWKeyErrorrHwarning isEnabledForloggingDEBUGrIlistr-)r(rrrr+get_comms    zCommManager.get_commstreamridentr/rGc 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.) rrmrXgetr;rHerrorr4r5) r(rkrlr/rmrnrrrYrrrr+r8s6      zCommManager.comm_opencCs\|d}|d}||}|durdSz||WdSty-tjd|ddYdSw)zHandler for comm_msg messagesrnrNzException in comm_msg for %sTro)rjrOr4rHrrr(rkrlr/rnrrrrr+r>s  zCommManager.comm_msgcCsp|d}|d}||}|durdSd|j|_|j|=z||WdSty7tjd|ddYdSw)zHandler for comm_close messagesrnrNTzException in comm_close for %sro)rjrWr&rJr4rHrrrsrrr+r=,s   zCommManager.comm_closeNrP)rrrYrZrr)rrrYr^rr^)rrrr)rrrr)rrrrc)rkrrlrr/rGrr) rQrRrSrTrr`rbr;r<rjr8r>r=rrrr+rVs        ! rV)rT __future__rr2rgtypingr\r Ztraitlets.utils.importstringrr TYPE_CHECKINGZzmq.eventloop.zmqstreamr getLoggerrHDictrAnyrGOptionalr Listbytesr CallableZ CommCallbackr^rrV__all__rrrr+s(    ( y