Mh_PdZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZmZddlZdd lmZmZmZGd d eZGd d ZGddeejZGddZeddZGddeejeZdS)z%Future-returning APIs for coroutines.) annotationsN)Future)deque)partial)chain)Any AwaitableCallable NamedTupleTypeVarcast)EVENTSPOLLINPOLLOUTcLeZdZUded<ded<ded<ded<d ed <d ed <d S) _FutureEventrfuturestrkindtupleargsdictkwargsrmsgtimerN)__name__ __module__ __qualname____annotations__K/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/zmq/_future.pyrrsFNNN IIIKKKLLL HHHJJJJJr!rcFeZdZUdZdZded<ded<d dZd d Zd dd ZdS)_AsynczMixin for common async logicNr _current_loopz type[Future]_Futurereturnc|j:||_||j|jS|}||jur||_|||S)zGet event loop Notice if event loop has changed, and register init_io_state on activation of a new event loop )r% _default_loop_init_io_state)self current_loops r" _get_loopz_Async._get_loop1s   %!%!3!3!5!5D     2 3 3 3% %))++ t1 1 1!-D     - - -r!c td)Nz!Must be implemented in a subclassNotImplementedError)r+s r"r)z_Async._default_loopBs!"EFFFr!NonecdSNr r+loops r"r*z_Async._init_io_stateEs r!)r'rr3)r'r1) rrr__doc__r%rr-r)r*r r!r"r$r$+s~&&M"GGGG       r!r$c`eZdZUdZded<ded<ded<ded<ddZddZddfd ZxZS) _AsyncPollerz:Poller that returns a Future on poll, instead of blocking.ztype[_AsyncSocket] _socket_classint_READ_WRITEz list[Any] raw_socketsr5rsocketevtfr r'r1ct)z"Schedule callback for a raw socketr/)r+r5r>r?r@s r"_watch_raw_socketz_AsyncPoller._watch_raw_socketQ!###r!socketsct)z$Unschedule callback for a raw socketr/)r+r5rDs r"_unwatch_raw_socketsz!_AsyncPoller._unwatch_raw_socketsUrCr! Awaitable[list[tuple[Any, int]]]c   |dkrf td} |n,#t$r} |Yd}~nd}~wwxYw Sgfd}fdgfd}jD]\}}t|tj rt|j s/j |}||tjzr|d|tjzr|d|d}|tjzr |jz}|tjzr |jz}||| fd} | r||?|dkr9fd } d |z| fd } | fd } |  S) z Return a Future for a poll eventrNc^sddSdSr3done set_result)rwatchers r"wake_rawz#_AsyncPoller.poll..wake_rawms7<<>> )""4((((( ) )r!c jgRSr3)rF)r@r5r=r+s r"z#_AsyncPoller.poll..rs/d/C{CCCr!c:D]}|dSr3)_clear_io_state)r@swrapped_socketss r"_clear_wrapper_ioz,_AsyncPoller.poll.._clear_wrapper_iows1$ $ $!!#### $ $r!pollrcrdSr( n#t$rYnwxYwdSr)dS t td} |dS#t$r }|Yd}~dSd}~wwxYwNr) rL cancelledcancel RuntimeError exception set_exceptionsuperr8rWrM Exception)r@resulte __class__rr+rNs r" on_poll_readyz(_AsyncPoller.poll..on_poll_readys&{{}}   "" MMOOOO#D  "" .$$W%6%6%8%899999."<66;;A>>F%%f-----!,,,((+++++++++,s)A AA(C C;C66C;c^sddSdSr3rK)rNsr"trigger_timeoutz*_AsyncPoller.poll..trigger_timeouts7||~~-&&t,,,,,--r!MbP?c~tdrdSdS)Nr\)hasattrr\remove_timeout)r@r5timeout_handles r"cancel_timeoutz)_AsyncPoller.poll..cancel_timeoutsG>8448"))+++++''77777r!c\sdSdSr3)rLr\)r@rNs r"cancel_watcherz)_AsyncPoller.poll..cancel_watchers3<<>> !      ! !r!)r&r`rWrMrar_r-add_done_callbackrD isinstance_zmqSocketr9 from_socketappendr_add_recv_eventr_add_send_eventr;r<rB call_later)r+timeoutrbrcrOrVr>maskr?rergrmrorr5r=rlrNrUrds` @@@@@@r"rWz_AsyncPoller.pollYsU a<< *a!!&)))) ( ( ($$Q'''''''' (M~~,,.."$  ) ) ) ) ) !! C C C C C C   /1 $ $ $ $ $!L D DLFD&$+.. D!&$*<==3!/;;FCCF#**6222$+%C**6'*BBB$,&C**6'*BBB""6***$+%&4:%C$,&'4;&C&&tVS(CCCC . . . . . . . .( !!-000  9  % %&7 8 8 8  7Q;; - - - - -"__TG^_MMN 8 8 8 8 8 8  $ $^ 4 4 4 ! ! ! ! !   000 s!A B%A??B) r5rr>rr?r:r@r r'r1)r5rrDrr'r1)rG)r'rH) rrrr6rrBrFrW __classcell__rds@r"r8r8IsDD%%%%JJJKKK$$$$$$$$dddddddddddr!r8c$eZdZedZdS)_NoTimercdSr3r r r!r"r\z_NoTimer.cancels r!N)rrr staticmethodr\r r!r"r~r~s-  \   r!r~T _AsyncSocket)boundceZdZUdZdZdZded<eZdZ dDdEfd Z e dFdGdZ dFdHfd Z ejj je _fdZejjje_ dIdJdZ dIdKd Zddd!dLd$Z dIdMd'Z dIdNd*Zd+ZdejfdOd,Zd-Zd.ZedFd/Zdddd0dPd9ZdQd:Zd;Z d<Z!dRd=Z"dFd>Z#d?Z$d@Z%dAZ&dFdBZ'dCZ(xZ)S)SrNr _zmq.Socket _shadow_sockrG _from_socket_zmq.Socket | Noner'r1c t|tjrd|}}|/t|j||_nDtj||fi|tj|j|_|)tj |j j dtdt|_t|_d|_|jj|_dS)N)shadowz^(io_loop) argument is deprecated in pyzmq 22.2. The currently active loop will always be used.) stacklevelr)rqrrrsr`__init__ underlyingrrwarningswarnrdrDeprecationWarningr _recv_futures _send_futures_stateFD_fd)r+context socket_typeio_looprrrds r"rz_AsyncSocket.__init__s gt{ + + 4%)7\G  # GG  L$;  < < < ,D   EGG Wk < *BBB"      #WW"WW $'r!clstype[T]r>rrrc|||S)z.Create an async socket from an existing Socket)rrr )rr>rs r"rtz_AsyncSocket.from_socketss8888r!linger int | Nonecx|js|jtt|jpg|jpg}|D]F}|js+ |j6#t$rYBwxYwG| t |dS)N)r) closedrlistrrrrrLr\r]rSr`close)r+r event_listeventrds r"rz_AsyncSocket.closes{ #tx3-1d(.B0B0HbII..J$  |((** ++----'  " " "  V $$$$$sA33 B?Bct|}|tkr|||Sr3)r`getr_schedule_remaining_events)r+keyrbrds r"rz_AsyncSocket.gets:S!! &==  + +F 3 3 3 r!TFflagsr:copybooltrack)Awaitable[list[bytes] | list[_zmq.Frame]]cP|dt|||S)zvReceive a complete multipart zmq message. Returns a Future whose result will be a multipart message. recv_multipartrrrrrvrr+rrrs r"rz_AsyncSocket.recv_multiparts5## TD%N%N%N$   r!Awaitable[bytes | _zmq.Frame]cP|dt|||S)zReceive a single zmq frame. Returns a Future, whose result will be the received frame. Recommend using recv_multipart instead. recvrrrrs r"rz_AsyncSocket.recvs5## 4e$eDDD$   r!nbytesrrAwaitable[int]cR|d|ft||S)zReceive a single zmq frame into a pre-allocated buffer. Returns a Future, whose result will be the number of bytes received. recv_intor)rrr)r+bufrrs r"rz_AsyncSocket.recv_into&s7## sfTu-M-M-M$   r! msg_parts%Awaitable[_zmq.MessageTracker | None]c P||d<||d<||d<|d||S)zqSend a complete multipart zmq message. Returns a Future that resolves when sending is complete. rrrsend_multipartrr)rw)r+rrrrrs r"rz_AsyncSocket.send_multipart1s; wvw##$4)F#SSSr!datarc ||d<||d<||d<|t||||d||S)zSend a single zmq frame. Returns a Future that resolves when sending is complete. Recommend using send_multipart instead. rrrrsendr)updaterrw)r+rrrrrs r"rz_AsyncSocket.send=s[ wvw dT???@@@##FV#DDDr!c|fd}|fd}|S)zDeserialize with FuturescrKs5!tjdddt dSr)dS} |}|dS#t$r }|Yd}~dSd}~wwxYw)z+Chain result through serialization to recvdNzFuture z completed while awaiting z. A message has been dropped!) rLr[r^rrRuntimeWarningr_rbrMra)_rloadedrcr@loadrecvds r"_chainz)_AsyncSocket._deserialize.._chainUsvvxx ((U__->->-FMd!ccuccc&    ) 1 122222llnn)!T#YYFLL(((((!'''OOA&&&&&&&&&'s3 C C?C::C?crdSrdSdS)z"Chain cancellation from f to recvdN)rLr[r\)rr@rs r" _chain_cancelz0_AsyncSocket._deserialize.._chain_cancelpsCzz|| {{}}    r!)r&rp)r+rrrrr@s `` @r" _deserializez_AsyncSocket._deserializeQs LLNN ) ) ) ) ) ) )2 '''       M***r!cjrtjtj}||t t|| fd} r |n |fd} |S)zSpoll the socket for events returns a Future for the poll results. crdSr( n#t$rYnwxYwdS|r)dSt }| ddSrZ) rLr[r\r]r^r_rrbrMr)r@evtsr poll_futurer+s r" unwrap_resultz(_AsyncSocket.poll..unwrap_results{{}} $$&& MMOOOO#D{{}} 5$$[%:%:%<%<=====K..0011!!$((4"3"344444sA AAcs( dS#t$rYdSwxYwdS)z4Cancel underlying poll if request has been cancelledN)rLr\r])rrs r" cancel_pollz&_AsyncSocket.poll..cancel_polls`##%% &&(((((#DD  s - ;;) rrrZMQErrorENOTSUP _poller_classregisterr rrWr&rLrp)r+ryrprrrrs` @@r"rWz_AsyncSocket.poll{s ; .- -- -     46166'??33  5 5 5 5 5 5 5      9 M+ & & & &  ) )- 8 8 8        --- r!c:fd}|||S)z'Add a timeout for a send or recv FuturecrdStjdSr3)rLr_rrAgainrXsr"future_timeoutz1_AsyncSocket._add_timeout..future_timeouts:{{}}    . . . . .r!) _call_later)r+rryrs ` r" _add_timeoutz_AsyncSocket._add_timeouts4 / / / / /888r!cR|||S)zSchedule a function to be called later Override for different IOLoop implementations Tornado and asyncio happen to both have ioloop.call_later with the same signature. )r-rx)r+delaycallbacks r"rz_AsyncSocket._call_laters$~~**5(;;;r!c\|sdS ||dS#t$rYdSwxYw)zMake sure that futures are removed from the event list when they resolve Avoids delaying cleanup until the next send/recv event, which may never come. N)remove ValueError)rrrs r"_remove_finished_futurez$_AsyncSocket._remove_finished_futuresR  F    e $ $ $ $ $    FF s  ++)rrrrrr tuple | Nonedict[str, Any] | Noner Future | Nonerc|p|}|d}|i}|dr|ddtjzrbt |j|} ||i|}||n,#t$r}| |Yd}~nd}~wwxYw|St} ttdr+|jj } | dkr| || dz} t||||d| } |j| |jt"t$zr||jrR| |jvrI|t+|j|j| |t$|S) z4Add a recv event, returning the corresponding FutureNr rrrRCVTIMEOrhrrrrrr)r& startswithrrrDONTWAITgetattrrrMrar_r~rjrcvtimeorrrrurr _handle_recvrprr _add_io_state) r+rrrrr@rrrcr timeout_ms _future_events r"rvz_AsyncSocket._add_recv_events  $dllnn <D >F ??6 " " vzz'1'='= 'M 4,d33D D$)&)) Q # # #"""""""" #H 4 $ $ @*3JQ))!Z$->??% t$v4u    !!-000    ( (6 1         '-43E"E"E   0#1'      v & & &s-B B4B//B4c|p|}|dvr|js|dd}|}|tjz|d<t |j|}d} ||fi|} || nj#tj $r1} |tjzr| | nd} Yd} ~ n/d} ~ wt$r} | | Yd} ~ nd} ~ wwxYw| r|j r| |St} ttdrC|jtj} | dkr||| dz} t%||d ||| }|j||t+|j|j| |t0|S) z4Add a send event, returning the corresponding Future)rrrrTFNSNDTIMEOrhr rr)r&rrrrrrrrrMrr_rarrr~rjrrrrurprrrr)r+rrrrr@r nowait_kwargsr finish_earlyrrcrrrs r"rwz_AsyncSocket._add_send_event s@  $dllnn - - -d6H -JJw**E"KKMMM%*T]%:M' "4,d33D L D.. .. Q: ) ) )4=()OOA&&&&$)L # # #"""""""" #  %633555 4 $ $ @*..t}==JQ))!Z$->??% t"VE    !!-000  ,-#       7###s$7 BC=%'C C=C88C=c|jttzsdSd}|jr?|j\}}}}}}|rd}nn|j?|js|t|dS||dkr| ddS|dkr |jj }n8|dkr |jj }n%|dkr |jj }ntd||dxxtjzcc< ||i|}| |dS#t $r } || Yd} ~ dSd} ~ wwxYw)zHandle recv eventsNrWrrrzUnhandled recv event type: r)rrrrrpopleftrL_drop_io_stater\rMrrrrrrrrar_) r+r@rrrrrrrbrcs r"rz_AsyncSocket._handle_recvJs $$V,,v5  F   .2.@.H.H.J.J +AtT61evvxx    ! (    ' ' ' 9 F  6>> LL    F % % %$3DD V^^$)DD [ $.DDC4CCDD Dw4=( !T4*6**F LL     OOA          s.E E7E22E7c|jttzsdSd}|jr?|j\}}}}}}|rd}nn|j?|js|t|dS||dkr| ddS|dkr |jj }n%|dkr |jj }ntd||dxxtjzcc< ||fi|}| |dS#t$r } || Yd} ~ dSd} ~ wwxYw)NrWrrzUnhandled send event type: r)rrrrrrrLrr\rMrrrrrrrar_) r+r@rrrrrrrbrcs r" _handle_sendz_AsyncSocket._handle_sendus $$V,,w6  F   040B0J0J0L0L -AtT63vvxx    ! )    ( ( ( 9 F  6>> LL    F % % %$3DD V^^$)DDC4CCDD Dw4=( !T#((((F LL     OOA          s D;; E%E  E%c|jjrdS|jt}|tjzr||tjzr|| dS)z(Dispatch IO events to _handle_recv, etc.N) rrrrrrrrrrr)r+fdevents zmq_eventss r"_handle_eventsz_AsyncSocket._handle_eventss   #  F&**622  #        $       '')))))r!c|jdkrdS||jt}||jzr|d|jdSdS)zkSchedule a call to handle_events next loop iteration If there are still events to handle. rN)rrrrrr)r+rs r"rz'_AsyncSocket._schedule_remaining_eventssi ;!   F >&**622F DK  5   Q 3 4 4 4 4 4 5 5r!cr|j|kr|j|zx}|_||jdS)zAdd io_state to poller.Nr_update_handlerr+states r"rz_AsyncSocket._add_io_states> ;%  "&+"5 5EDK T[)))))r!cn|j|zr|j|z|_||jdS)z&Stop poller from watching an io_state.Nr r s r"rz_AsyncSocket._drop_io_states= ;  1+%0DK T[)))))r!cZ|r||dS)zOUpdate IOLoop handler with state. zmq FD is always read-only. N)r-rr s r"r z_AsyncSocket._update_handlers4   NN    '')))))r!c||}||j|j|j|d|jdS)z#initialize the ioloop event handlerNr)r- add_handlerrrr;rr4s r"r*z_AsyncSocket._init_io_statesU <>>##D *D,?LLL D/00000r!c|j}|jjr|j}|j|j|dSdS)zNunregister the ioloop event handler called once during close N)rrrr%remove_handler)r+rs r"rSz_AsyncSocket._clear_io_statesO     # B   )   - -b 1 1 1 1 1 * )r!)NrGNN)rrr'r1r3)rrr>rrrr'r)rrr'r1)rTF)rr:rrrrr'r)rr:rrrrr'r)rr:rr:r'r)rrrr:rrr'r) rrrr:rrrrrrr'r)r'r) rrrrrrrrr'r)NNN)rr)*rrrrrrrr8rrr classmethodrtrrrrsr6rrrrrrrrrWrrrrrvrwrrrrrrr r*rSr{r|s@r"rrsMM F M C+/ (((((((:9999[9 % % % % % % %K%-EM +/)CK@E      @E      ()q       HM T T T T T EEEEE((((T t{00000d 9 9 9<<<\*"(, $ 555555n;;;;z)!)!)!V&!&!&!R * * * *5555 *** *** * * *1111 2 2 2 2 2 2 2r!) r6 __future__rrasyncior collectionsr functoolsr itertoolsrtypingrr r r r r zmqrrrrrrr$Pollerr8r~rrsrr r!r"rs++#"""""'''''''''':$