:vje-dZddlZddlZddlZddlZddlmZddlm Z m Z ddl Z ddl mZddlmZddlmZdd lmZedZGd d eZGd d e Zej2eGddZGddeZy)zCBase classes to manage a Client's interaction with a running kernelN)Empty)EventThread) ensure_async)protocol_version_info) HBChannelABC)SessionceZdZdZy)InvalidPortNumberz/An exception raised for an invalid port number.N)__name__ __module__ __qualname____doc__h/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_client/channels.pyr r s9rr c XeZdZUdZdZdZdZdZdZe e d<dZ dZ dZ ddddejdzdedzd ej$ej&eefefd edzd df fd Zeej4dd ZddZddZddZddZddZ d e!fdZ"ddZ#ddZ$de d dfdZ%xZ&S) HBChannelzThe heartbeat channel which monitors the kernel heartbeat. Note that the heartbeat channel is paused by default. As long as you start this channel, the kernel manager will ensure that it is paused and un-paused as appropriate. NFg? time_to_dead)curve_serverkeycontextsessionaddressrreturnc.t|d|_||_||_||_t |tr|ddk(r d}t|d|z}n|}||_ d|_ t|_ d|_ tj|_y)a<Create the heartbeat monitor thread. Parameters ---------- context : :class:`zmq.Context` The ZMQ context to use. session : :class:`session.Session` The session to use. address : zmq url Standard (ip, port) tuple that the kernel is listening on. curve_serverkey : bytes, optional CurveZMQ server public key (Z85). When provided, the heartbeat REQ socket is configured as a CurveZMQ client so it can communicate with a CurveZMQ-enabled kernel. Trrz*The port number for a channel cannot be 0.z tcp://%s:%iFN)super__init__daemonrrr isinstancetupler r_runningr_exit_pausezmqPollerpoller)selfrrrrmessage address_str __class__s rrzHBChannel.__init__5s.    . gu %qzQF'00''1K!K"  W  jjl rc(t dt_yy)NT)r_exitingrrr _notice_exitzHBChannel._notice_exitbs  !%I  !rc|j?|jj|j|jj|jJ|jjt j |_d|j_|jTt j\}}||j_ ||j_ |j|j_|jJ|jj|j|jj|jt jy)N)socketr' unregistercloserr%REQlingerr curve_keypaircurve_secretkeycurve_publickeyrconnectregisterPOLLIN)r( client_pub client_secs r_create_socketzHBChannel._create_socketjs ;; " KK " "4;; / KK   ||'''ll))#''2 !     +&)%6%6%8 "J *4DKK '*4DKK '*.*>*>DKK '||''' DLL) T[[#**5rc&K|jd|_d|_|jJ|jrP|jr&|j j |j?d}t|jjdd{tj}|j j |jt|jjd|_|jr-t|jjd{|jr:tj|z }|j||jN|jrOyy77^w)z2The thread's main activity. Call start() instead.TNgspingr)r>r"_beatingr1r$r#waitrrsendtimeboolr'pollrecv call_handlers)r(since_last_heartbeat request_times r _async_runzHBChannel._async_run~s'   {{&&&mm{{  1 12#& t{{//89 9 999;L JJOOD-- . !1!1!!45DM}}"4;;#3#3#5666'+yy{\'A$""#78##%3mm :7s,BFF BF0F1AF FFctj}tj| |j|j |j y#|j wxYw)zRun the heartbeat thread.N)asyncionew_event_loopset_event_looprun_until_completerJr3)r(loops rrunz HBChannel.runsK%%'t$   # #DOO$5 6 JJLDJJLs AA-cd|_y)zPause the heartbeat.TNr$r(s rpausezHBChannel.pauses  rcd|_y)zUnpause the heartbeat.FNrSrTs runpausezHBChannel.unpauses  rcV|jr|js |jryy)z9Is the heartbeat running and responsive (and not paused).TF)is_aliver$r@rTs r is_beatingzHBChannel.is_beatings ==?4;;4==rcd|_|jj|j|j y)z2Stop the channel's event loop and join its thread.FN)r"r#setjoinr3rTs rstopzHBChannel.stops)    rc|j% |jjdd|_yy#t$rYwxYw)zClose the heartbeat thread.Nrr5r1r3 ExceptionrTs rr3zHBChannel.closeH ;; "  !!!+DK #   3 ??rHcy)aiThis method is called in the ioloop thread when a message arrives. Subclasses should override this method to handle incoming messages. It is important to remember that this method is called in the thread so that some logic must be done to ensure that the application level handlers are called in the application thread. Nr)r(rHs rrGzHBChannel.call_handlerss r)NNrN)'r rrrrr1rr-rfloat__annotations__r"r$r@r%Contextr tUnionTuplestrintbytesr staticmethodatexitr:r.r>rJrQrUrWrDrZr^r3rG __classcell__r+s@rrr#sG FGHL%H FH'+"&35 +# )- +#t#+#4+#c*C/0 +#  +# +#Z __&& 6( DD % D rrc eZdZdZddej dedejddffd Z dejdeje ejffd Z dd e dzdeje ejffd Zdejeje ejffd Zdefd ZddZeZdefdZdeje ejfddfdZddZxZS)ZMQSocketChannelzA ZMQ socket wrapperNr1rrPrc>t|||_||_y)aCreate a channel. Parameters ---------- socket : :class:`zmq.Socket` The ZMQ socket to use. session : :class:`session.Session` The session to use. loop Unused here, for other implementations N)rrr1r)r(r1rrPr+s rrzZMQSocketChannel.__init__s )/  rkwargsc |jJ|jjdi|}|jj|\}}|jj |SNrr1recv_multipartrfeed_identities deserialize)r(rxmsg_identsmsgs r_recvzZMQSocketChannel._recvsX{{&&&(dkk((262||33C8 ||''--rtimeoutc|jJ|dn t|dz}|jj|}|r|j}|Stz-Gets a message if there is one that is ready.Nr0r1rorErrr(r timeout_msreadyress rget_msgzZMQSocketChannel.get_msgsR{{&&&$_T#gn2E    , **,CJKrcjg} |j|j!#t$rY|SwxYw)*Get all messages that are currently ready.appendrrr(msgss rget_msgszZMQSocketChannel.get_msgss?  DLLN+   s % 22ch|jJt|jjdS)*Is there a message that has been received?rrr1rDrErTs r msg_readyzZMQSocketChannel.msg_readys/{{&&&DKK$$Q$/00rc|j% |jjdd|_yy#t$rYwxYw)zClose the socket channel.Nrr`rarTs rr3zZMQSocketChannel.close rcrdc|jduS)z"Test whether the channel is alive.N)r1rTs rrYzZMQSocketChannel.is_alives{{$&&rrcl|jJ|jj|j|y)z(Pass a message to the ZMQ socket to sendN)r1rrB)r(rs rrBzZMQSocketChannel.sends*{{&&& $++s+rcy)zStart the socket channel.NrrTs rstartzZMQSocketChannel.starts rNrg)r rrrr%Socketr rkAnyrDictrnrrhrListrrDrr3r^rYrBrrsrts@rrvrvsszzG155TX".aee.sAEEz(:. ut| qvvc155j7I !&&QUU !34141  D'$',sAEEz*,t,  rrvc eZdZUdZej j ed<ddej j dede jddffd Z de jde je e jffd Z dd edzde je e jffd Zde j"e je e jffd Zdefd ZxZS)AsyncZMQSocketChannelzA ZMQ socket in an async APIr1NrrPrct|tjjs d}t |t |||y)aCreate a channel. Parameters ---------- socket : :class:`zmq.asyncio.Socket` The ZMQ socket to use. session : :class:`session.Session` The session to use. loop Unused here, for other implementations zSocket must be asyncioN)r r%rLr ValueErrorrr)r(r1rrPrr+s rrzAsyncZMQSocketChannel.__init__(s7&#++"4"45*CS/ ! )rrxcK|jJ|jjdi|d{}|jj|\}}|jj |S7=wrzr{)r(rxr_rs rrzAsyncZMQSocketChannel._recv9sd{{&&&.DKK..888,,..s34||''--9s.A0A.>A0rcK|jJ|dn t|dz}|jj|d{}|r|jd{}|St7&7wrrrs rrzAsyncZMQSocketChannel.get_msg?sg{{&&&$_T#gn2E kk&&z22  $CJK 3$s#?A,A(A,A* A,*A,cKg} |j|jd{)7 #t$rY|SwxYww)rNrrs rrzAsyncZMQSocketChannel.get_msgsLsF  $,,.010   s,A1/1A1 >A>AcK|jJt|jjdd{S7w)rNrrrrTs rrzAsyncZMQSocketChannel.msg_readyVs9{{&&&$++**1*55665s3A> Ar)r rrrr%rLrrir rkrrrrnrrhrrrrDrrsrts@rrr#s& KK  *s{{11*G*155*\`*".AEE.affS!%%Z.@.'+ t|  QUU  qvvc155j'9 :77rr)rrLrrrCtypingrkqueuer threadingrr zmq.asyncior%jupyter_core.utilsr_versionr channelsabcr rr major_protocol_versionrbr rr:rvrrrrrsI #++%/q1  m m ` i J J Z67,67r