:vj)dZddlmZddlZddlZddlZddlZddlZejrddl m Z ejdZ ejeej fZej$ejeej fZej$ej(eZej.egdfZej.degdfZGddZGdd Zdd gZy) z?Default classes for Comm and CommManager, for usage in IPython.) annotationsN) ZMQStreamCommBaseCommceZdZdZ d d fd Z d ddZddZ d ddZ d ddZ d ddZ ddZ dd Z dd Z dd Z xZS)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. c  t |di| |r|ntjj|_||_||_||_|r|nd|j jd|_ | r| ni|_ | r| ni|_ d|_ d|_d|_|j r|j!|||yd|_y)Nzcomm-asciiTdatametadatabuffersF)super__init__uuiduuid4hexcomm_idprimary target_name target_moduleencodetopic _open_data _close_data _msg_callback_close_callback_closedopen) selfrr r r rrrrrrkwargs __class__s _/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/comm/base_comm.pyrzBaseComm.__init__!s "6"")wtzz|/?/?  &*#UE$,,*@)H)H)Q (2**5;22648 << II4(GI D DLc d}t|)Nz*publish_msg Comm method is not implemented)NotImplementedError)r msg_typer r r keysmsgs r# publish_msgzBaseComm.publish_msgEs;!#&&r$ctjt5|jddddy#1swYyxYw)ztrigger close on gcT)deletingN) contextlibsuppress Exceptioncloser s r#__del__zBaseComm.__del__Ps,   + JJJ %, + +s6?c,| |j}tj}| d}t||j | |j d||||j |jd|_y#t$r|j|wxYw)z+Open the frontend-side version of this commNz.Comms cannot be opened without a comm_manager. comm_open)r r r rrF) rcommget_comm_manager RuntimeError register_commr*rrrr/unregister_comm)r r r r comm_managerr)s r#rz BaseComm.openYs <??D,,.  BCs# #""4(    ! ,,"00  !DL   ( ( .  s 2A77Bc|jryd|_| |j}|jd||||s$tjj |yy)z,Close the frontend-side version of this commNT comm_closer )rrr*r5r6r9)r r r r r,s r#r0zBaseComm.closetse <<  <##D      ! ! # 3 3D 9r$c.|jd|||y)z8Send a message to the frontend-side version of this commcomm_msgr N)r*)r r r r s r#sendz BaseComm.sends#    r$c||_y)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_closes  (r$c||_y)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)rrAs r#on_msgzBaseComm.on_msgs &r$ctjd|j||jr|j|yy)zHandle a comm_close messagezhandle_close[%s](%s)N)loggerdebugrr)r r)s r# handle_closezBaseComm.handle_closes3 +T\\3?      % r$ctjd|j||jrZddlm}|}|r|j jd|j||r|j jdyyy)zHandle a comm_msg messagezhandle_msg[%s](%s)r) get_ipython pre_execute post_executeN)rGrHrrIPythonrKeventstrigger)r r)rKshells r# handle_msgzBaseComm.handle_msgsl )4<<=    +ME $$]3   s # $$^4 r$) r5NNNNTNNNN)rstrr MaybeDictr rTr BuffersTyper str | NonerboolrrVrz bytes | NonerrTrrTr!t.AnyreturnNone)NNN) r'rSr rTr rTr rUr(rXrYrZrYrZ)r rTr rTr rUrYrZ)NNNF) r rTr rTr rUr,rWrYrZ)rBzCommCallback | NonerYrZ)r) MessageTyperYrZ)__name__ __module__ __qualname____doc__rr*r2rr0r?rCrErIrR __classcell__)r"s@r#rrs""#"$(" $!%"!"!"! "!  "!  "!"!""!"!"!"!"! "!N"# ' ' ' '  '  '  '&Z^09KV :"# ::: :  :  :2Z^    09  KV    (&& 5r$cXeZdZdZd dZd dZddZddZddZddZ ddZ dd Z dd Z y ) CommManagerzDDefault CommManager singleton implementation for Comms in the Kernelc i|_i|_y)N)commstargetsr1s r#rzCommManager.__init__s*, 68 r$cTt|trO|jdd}t|dk(r!|\}}t ||g} t ||}nt |d}tjt||j|<y#t $r}d|}t||d}~wwxYw)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. .)fromlistzNo module named Nr) isinstancerSrsplitlen __import__getattrAttributeError ImportErrortcastCommTargetCallbackrf) r rfpartspackageobjmodulee error_msgs r#register_targetzCommManager.register_targets a HHS!$E5zQ$ #Gse<8,A uQx($%FF+=q$A [!&8"23% 8I%i0a78s B B'B""B'c8|jj|S)z5Unregister a callable registered with register_target)rfpop)r rrvs r#unregister_targetzCommManager.unregister_targets|| ,,r$c<|j}||j|<|S)zRegister a new comm)rre)r r5rs r#r8zCommManager.register_comms,," 7r$cN|jj|j}y)z,Unregister a comm, and close its counterpartN)rerr)r r5s r#r9zCommManager.unregister_commszz~~dll+r$c  |j|S#t$rstjd|tj t j r7tjdt|jjYywxYw)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) reKeyErrorrGwarning isEnabledForloggingDEBUGrHlistr()r rs r#get_commzCommManager.get_commsh ::g& &  NN-w 7""7==1 0$tzz7H2IJ  sA9B  B cddlm}|d}|d}|d}|jj|d}||d|} |j | |t j d |n || |y | jy#t$rt j d |d Y6wxYw#t$rt j d d YywxYw)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.) r5rrfgetr8rGerrorr/r0) r streamidentr)rrrrrvr5s r#r4zCommManager.comm_opens$i.)$m, LL  [$ /#  4 9 LL={ K c$    DJJL  c E{]a b c   LLE   s$( B3B,!B)(B), CCc|d}|d}|j|}|y |j|y#t$rtj d|dYywxYw)zHandler for comm_msg messagesrrNzException in comm_msg for %sTr)rrRr/rGrr rrr)rrr5s r#r>zCommManager.comm_msg(sci.)$}}W% <  Q DOOC  Q LL74L P Qs3!AAc|d}|d}|j|}|yd|j|_|j|= |j|y#t$rt j d|dYywxYw)zHandler for comm_close messagesrrNTzException in comm_close for %sr)rrerrIr/rGrrs r#r<zCommManager.comm_close5si.)$}}W% < &* 7# JJw  S D  c " S LL97TL R SsA!A87A8Nr[)rrSrvzCommTargetCallback | strrYrZ)rrSrvrurYru)r5rrYrS)r5rrYrZ)rrSrYzBaseComm | None)rrrrSr)r\rYrZ) r]r^r_r`rr}rr8r9rr4r>r<rr$r#rcrcs6N9B6- , &B QSr$rc)r` __future__rr-rtypingrsrr5 TYPE_CHECKINGzmq.eventloop.zmqstreamr getLoggerrGDictrSAnyr\OptionalrTListbytesrUCallable CommCallbackrurrc__all__rr$r#rsE# ??1   6 "ffS!%%Z JJqvvc155j) * jj' zz;--. ZZ[ 94 ?@d5d5NBSBSJ } %r$