Mh'dZddlmZddlZddlZddlZddlZddlm Z ddl Z ej rddl m Z ejdZejeejfZejejeejfZejejeZejegdfZejdegdfZGddZGd d Zd dgZdS) z@Default classes for Comm and CommManager, for usage in IPython. ) annotationsN) import_item) ZMQStreamCommBaseCommceZdZdZ d+d,fd Z d-d.dZd/dZ d-d0dZ d1d2d!Z d-d0d"Z d3d%Z d3d&Z d4d)Z d4d*Z xZS)5rzClass 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  tjdi| |r|ntjj|_||_||_||_|r|nd|jz d|_ | r| ni|_ | r| ni|_ d|_ d|_d|_|jr||||dSd|_dS)Nzcomm-%sasciiTr rrF)super__init__uuiduuid4hexrrr rencoderrr _msg_callback_close_callback_closedopen) selfr r rrrrrrrrr __class__s N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/comm/base_comm.pyr#zBaseComm.__init__$s ""6"""")?wwtz||/?  &*#SUU)dl*B)J)J7)S)S (2:***5=;;22648 < ! II4(GI D D D D D DLLLmsg_typekeysc $d}t|)Nz*publish_msg Comm method is not implemented)NotImplementedError)r,r0r rrr1msgs r. publish_msgzBaseComm.publish_msgHs;!#&&&r/ctjt5|dddddS#1swxYwYdS)ztrigger close on gcT)deletingN) contextlibsuppress Exceptioncloser,s r.__del__zBaseComm.__del__Ss   + + & & JJJ % % % & & & & & & & & & & & & & & & & & &s>AAc0||j}tj}|d}t||| |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 rrr rF) rr get_comm_manager RuntimeError register_commr5r rr*r:unregister_comm)r,r rr comm_managerr4s r.r+z BaseComm.open\s <?D,..  BCs## #""4(((    ! ,"0    !DLLL     ( ( . . .  s ,A44!BFr7c|jrdSd|_||j}|d||||s(tj|dSdS)z,Close the frontend-side version of this commNT comm_closer )r*rr5r r@rC)r,r rrr7s r.r;zBaseComm.closews <  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)r5)r,r rrs r.sendz BaseComm.sends7        r/callbackCommCallback | Nonec||_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,rJs r.on_closezBaseComm.on_closes (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)r(rMs r.on_msgzBaseComm.on_msgs&r/r4 MessageTypectd|j||jr||dSdS)zHandle a comm_close messagezhandle_close[%s](%s)N)loggerdebugrr))r,r4s 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)rSrTrr(IPythonrWeventstrigger)r,r4rWshells r. handle_msgzBaseComm.handle_msgs )4<===   5 + + + + + +KMME 4 $$]333   s # # # 5 $$^44444 5 5 5 5r/) r NNNNTNNNN)r r r r rr rrrrrrrrrrrr rr rrrr)NNN) r0r r r rr rrr1rrrrr)r r rr rrrr)NNNF) r r rr rrr7rrr)rJrKrr)r4rQrr)__name__ __module__ __qualname____doc__r#r5r=r+r;rIrNrPrUr^ __classcell__)r-s@r.rrs\""#"$(" $!%"!"!"!"!"!"!"!N"# ' ' ' ' '&&&&Z^:"# :::::2Z^      ((((&&&&&&&& 5 5 5 5 5 5 5 5r/cZeZdZdZddZdd Zdd ZddZd dZd!dZ d"dZ d"dZ d"dZ dS)# CommManagerzDDefault CommManager singleton implementation for Comms in the Kernelrrc"i|_i|_dS)N)commstargetsr<s r.r#zCommManager.__init__s*, 68 r/r r fCommTargetCallback | strct|trt|}tjt ||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) isinstancer rtcastCommTargetCallbackrir,r rjs r.register_targetzCommManager.register_targets@ a   AA$%F+=q$A$A [!!!r/rpc6|j|S)z5Unregister a callable registered with register_target)ripoprqs r.unregister_targetzCommManager.unregister_targets| ,,,r/r rc(|j}||j|<|S)zRegister a new comm)rrh)r,r rs r.rBzCommManager.register_comms," 7r/cD|j|j}dS)z,Unregister a comm, and close its counterpartN)rhrtr)r,r s r.rCzCommManager.unregister_commsz~~dl++r/rBaseComm | Nonec< |j|S#t$rtd|tt jr?tdt|j YdSwxYw)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) rhKeyErrorrSwarning isEnabledForloggingDEBUGrTlistr1)r,rs r.get_commzCommManager.get_comms :g& &    NN-w 7 7 7""7=11 K 0$tz7H7H2I2IJJJ44  s BBBstreamridentr4rQcddlm}|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_commcontentrr NF)rrr z"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.) r rrigetrBrSerrorr:r;) r,rrr4rrrr rjr s r.r?zCommManager.comm_opensW$$$$$$i.)$m, L  [$ / /{#    4   9 LL={ K K K K c$  c c c E{]a bbbbb c  DJLLLLL    LLE        s$< B 'B43B48C &C32C3c|d}|d}||}|dS |j|dS#t$r!td|dYdSwxYw)zHandler for comm_msg messagesrrNzException in comm_msg for %sTr)rr^r:rSrr,rrr4rrr s r.rHzCommManager.comm_msgsi.)$}}W%% < F Q DOC  Q Q Q LL74L P P P P P P Qs='A('A(c |d}|d}||}|dSd|j|_|j|= |j|dS#t$r!t d|dYdSwxYw)zHandler for comm_close messagesrrNTzException in comm_close for %sr)rrhr*rUr:rSrrs r.rFzCommManager.comm_close,si.)$}}W%% < F&* 7# Jw  S D c " " " " " S S S LL97TL R R R R R R SsA'BBNr_)r r rjrkrr)r r rjrprrp)r rrr )r rrr)rr rrx)rrrr r4rQrr) r`rarbrcr#rrrurBrCrr?rHrFr!r/r.rfrfsNN9999 B B B B---- ,,,, &B Q Q Q QSSSSSSr/rf)rc __future__rr8r}typingrnr$traitlets.utils.importstringrr TYPE_CHECKINGzmq.eventloop.zmqstreamr getLoggerrSDictr AnyrQOptionalr ListbytesrCallable CommCallbackrprrf__all__r!r/r.rsw #""""" 444444 ?2111111  6 " "fS!%Z Jqvc15j) * j' z;--. Z[ 94 ?@d5d5d5d5d5d5d5d5NvSvSvSvSvSvSvSvSr * %r/