|tf'dZddlmZddlZddlZddlZddlZddlm Z ddl Z ejrddl m Z ejdZej eej$fZej(ej eej$fZej(ej,eZej2egdfZej2degdfZGddZGd d Zd dgZy) z@Default classes for Comm and CommManager, for usage in IPython. ) annotationsN) import_item) 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 zjd|_ | r| ni|_ | r| ni|_ d|_ d|_d|_|j r|j!|||yd|_y)Nzcomm-%sasciiTdatametadatabuffersF)super__init__uuiduuid4hexcomm_idprimary target_name target_moduleencodetopic _open_data _close_data _msg_callback_close_callback_closedopen) selfrr r rrrrrrrkwargs __class__s X/var/www/html/software/conda/envs/higlass/lib/python3.12/site-packages/comm/base_comm.pyrzBaseComm.__init__$s "6"")wtzz|/?/?  &*#U)dll*B)J)J7)S (2**5;22648 << II4(GI D DLc d}t|)Nz*publish_msg Comm method is not implemented)NotImplementedError)r!msg_typer r rkeysmsgs r$ publish_msgzBaseComm.publish_msgHs;!#&&r%ctjt5|jddddy#1swYyxYw)ztrigger close on gcT)deletingN) contextlibsuppress Exceptioncloser!s r$__del__zBaseComm.__del__Ss5   + & 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 rrrF) rcommget_comm_manager RuntimeError register_commr+rrrr0unregister_comm)r!r r r comm_managerr*s r$r z BaseComm.open\s <??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+r6r7r:)r!r r rr-s r$r1zBaseComm.closewse <<  <##D      ! ! # 3 3D 9r%c.|jd|||y)z8Send a message to the frontend-side version of this commcomm_msgr N)r+)r!r r rs 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)rrBs 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)rHrIrrIPythonrLeventstrigger)r!r*rLshells r$ handle_msgzBaseComm.handle_msgsl )4<<=    +ME $$]3   s # $$^4 r%) r6NNNNTNNNN)rstrr MaybeDictr rUr BuffersTyper str | NonerboolrrWrz bytes | NonerrUrrUr"t.AnyreturnNone)NNN) r(rTr rUr rUrrVr)rYrZr[rZr[)r rUr rUrrVrZr[)NNNF) r rUr rUrrVr-rXrZr[)rCzCommCallback | NonerZr[)r* MessageTyperZr[)__name__ __module__ __qualname____doc__rr+r3r r1r@rDrFrJrS __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)commstargetsr2s r$rzCommManager.__init__s*, 68 r%ct|tr t|}tjt ||j |<y)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) isinstancerTrtcastCommTargetCallbackrgr!rfs r$register_targetzCommManager.register_targets1 a AA$%FF+=q$A [!r%c8|jj|S)z5Unregister a callable registered with register_target)rgpoprms r$unregister_targetzCommManager.unregister_targets|| ,,r%c<|j}||j|<|S)zRegister a new comm)rrf)r!r6rs r$r9zCommManager.register_comms,," 7r%cN|jj|j}y)z,Unregister a comm, and close its counterpartN)rfrqr)r!r6s r$r:zCommManager.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) rfKeyErrorrHwarning isEnabledForloggingDEBUGrIlistr))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.) r6r~rggetr9rHerrorr0r1) r!streamidentr*r~rrrrnr6s r$r5zCommManager.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)r|rSr0rHrr!rrr*rrr6s r$r?zCommManager.comm_msgsci.)$}}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)r|rfrrJr0rHrrs r$r=zCommManager.comm_close,si.)$}}W% < &* 7# JJw  S D  c " S LL97TL R SsA!A87A8Nr\)rrTrnzCommTargetCallback | strrZr[)rrTrnrlrZrl)r6rrZrT)r6rrZr[)rrTrZzBaseComm | None)rrrrTr*r]rZr[) r^r_r`rarrorrr9r:r|r5r?r=rr%r$rdrds6N9 B- , &B QSr%rd)ra __future__rr.rytypingrjrtraitlets.utils.importstringrr6 TYPE_CHECKINGzmq.eventloop.zmqstreamr getLoggerrHDictrTAnyr]OptionalrUListbytesrVCallable CommCallbackrlrrd__all__rr%r$rs # 4 ??1   6 "ffS!%%Z JJqvvc155j) * jj' zz;--. ZZ[ 94 ?@d5d5NvSvSr * %r%