:vj+_@dZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZmZmZddlZdd lmZmZmZGd d eZGd dZGddeej6ZGddZeddZGddeej>eZ y)z%Future-returning APIs for coroutines.) annotationsN)Future)deque) Awaitable)partial)chain)AnyCallable NamedTupleTypeVarcast)EVENTSPOLLINPOLLOUTcJeZdZUded<ded<ded<ded<d ed <d ed <y ) _FutureEventrfuturestrkindtupleargsdictkwargsr msgtimerN)__name__ __module__ __qualname____annotations__\/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/zmq/_future.pyrrs N I K L H Jr!rcDeZdZUdZdZded<ded<d dZd dZd d d Zy) _AsynczMixin for common async logicNr _current_loopz type[Future]_Futurec|j<|j|_|j|j|jS|j}||jur||_|j||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_loop1sy    %!%!3!3!5D     2 2 3%% %))+ t11 1!-D     -r!ctd)Nz!Must be implemented in a subclassNotImplementedError)r*s r"r(z_Async._default_loopBs!"EFFr!cyNr r*loops r"r)z_Async._init_io_stateEs r!)returnr r1)r4None) rrr__doc__r%rr,r(r)r r!r"r$r$+s%&M3 "G r!r$c`eZdZUdZded<ded<ded<ded<d d Zd d Zddfd ZxZS) _AsyncPollerz:Poller that returns a Future on poll, instead of blocking.ztype[_AsyncSocket] _socket_classint_READ_WRITEz list[Any] raw_socketsct)z"Schedule callback for a raw socketr.)r*r3socketevtfs r"_watch_raw_socketz_AsyncPoller._watch_raw_socketQ !##r!ct)z$Unschedule callback for a raw socketr.)r*r3socketss r"_unwatch_raw_socketsz!_AsyncPoller._unwatch_raw_socketsUrCr!c j |dk(r# t d} j| Sj jgfd}jfdgfd}jD]\}}t|tjrt|js,jj|}j||tjzr|j!d|tj"zs|j%dj|d}|tjzr|j&z}|tj"zr|j(z}j+||| fd} j| rj||6|dkDr1fd } j-d |z| fd } j| fd } j|  S#t$r} j |Yd}~ Sd}~wwxYw) z Return a Future for a poll eventrNcJjsjdyyr1done set_result)rwatchers r"wake_rawz#_AsyncPoller.poll..wake_rawms<<>""4("r!c*jgSr1)rF)rAr3r=r*s r"z#_AsyncPoller.poll..rs/d//C{Cr!c4D]}|jyr1)_clear_io_state)rAswrapped_socketss r"_clear_wrapper_ioz,_AsyncPoller.poll.._clear_wrapper_iows$!!#%r!pollrcjryjr jyj r j j y t t#d}j|y#t$rYywxYw#t$r}j |Yd}~yd}~wwxYwNr) rJ cancelledcancel RuntimeError exception set_exceptionsuperr8rUrK Exception)rAresulte __class__rr*rLs r" on_poll_readyz(_AsyncPoller.poll..on_poll_readys{{}  "MMO  "$$W%6%6%89."<;A>F%%f-$!,((++,s)B &B BB B>#B99B>cJjsjdyyr1rI)rLsr"trigger_timeoutz*_AsyncPoller.poll..trigger_timeouts||~&&t,&r!MbP?cbtdrjyjy)NrZ)hasattrrZremove_timeout)rAr3timeout_handles r"cancel_timeoutz)_AsyncPoller.poll..cancel_timeouts'>84"))+''7r!cHjsjyyr1)rJrZ)rArLs r"cancel_watcherz)_AsyncPoller.poll..cancel_watchers<<> "r!)r&r^rUrKr_r]r,add_done_callbackrE isinstance_zmqSocketr9 from_socketappendr_add_recv_eventr_add_send_eventr;r<rB call_later)r*timeoutr`rarMrTr?maskr@rcrerkrmrr3r=rjrLrSrbs` @@@@@@r"rUz_AsyncPoller.pollYs a< *a!!&)M~~,,."$  ) !! C /1 $!LLLFD&$++.!&$*<*<=!//;;FCF#**62$++%**6'*B$,,&**6'*B""6*$++%4::%C$,,&4;;&C&&tVS(C#)& .( !!-0   % %&7 8  7Q; -"__TG^_MN 8  $ $^ 4 !   0 } ($$Q''M  (sH,, I5I  I) r3r r?r r@r:rAr r4r5)r3r rEr r4r5))r4z Awaitable[list[tuple[Any, int]]]) rrrr6rrBrFrU __classcell__rbs@r"r8r8Is2D%% J K$$ddr!r8ceZdZedZy)_NoTimercyr1r r r!r"rZz_NoTimer.cancels r!N)rrr staticmethodrZr r!r"r}r}s  r!r}T _AsyncSocket)boundcLeZdZUdZdZdZded<eZdZ d d!fd Z e d"d#dZ d"d$fd Z ejjj e _fdZejj"j e_ d% d&d Z d% d'd Zddd  d(d Z d% d)d Z d% d*dZdZdej0fd+dZdZdZed"dZdddd d,dZd-dZdZ dZ!d.dZ"d"dZ#dZ$dZ%dZ&d"dZ'dZ(xZ)S)/rNr _zmq.Socket _shadow_sockc t|tjrd|}}|"t||j ||_n?t|||fi|tjj|j |_|3tj|jjdtdt|_t|_d|_|j j"|_y)N)shadowz^(io_loop) argument is deprecated in pyzmq 22.2. The currently active loop will always be used.) stacklevelr)rorprqr^__init__ underlyingrrwarningswarnrbrDeprecationWarningr _recv_futures _send_futures_stateFD_fd)r*context socket_typeio_loop _from_socketrrbs r"rz_AsyncSocket.__init__s gt{{ +%)7\G  # G L$;$;  < ,D  G Wk >**+,BB"   #W"W $$''r!c|||S)z.Create an async socket from an existing Socket)rrr )clsr?rs r"rrz_AsyncSocket.from_sockets88r!cp|js|j~tt|jxsg|j xsg}|D]8}|j jr |j j:|jt|1|y#t$rYgwxYw)N)linger) closedrlistrrrrrJrZr[rQr^close)r*r event_listeventrbs r"rz_AsyncSocket.closes{{txx3-1d((.B0B0B0HbI.J$||((* ++-$  "  V $ (s,B)) B54B5cZt||}|tk(r|j||Sr1)r^getr_schedule_remaining_events)r*keyr`rbs r"rz_AsyncSocket.gets+S! &=  + +F 3 r!c@|jdt|||S)zvReceive a complete multipart zmq message. Returns a Future whose result will be a multipart message. recv_multipartflagscopytrackrrtrr*rrrs r"rz_AsyncSocket.recv_multiparts*## TD%N$  r!c@|jdt|||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.recvs*## 4e$eD$  r!nbytesrcB|jd|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&s,## sfTu-M$  r!c H||d<||d<||d<|jd||S)zqSend a complete multipart zmq message. Returns a Future that resolves when sending is complete. rrrsend_multipartrr)ru)r* msg_partsrrrrs r"rz_AsyncSocket.send_multipart1s8 wvw##$4)F#SSr!c ||d<||d<||d<|jt||||jd||S)zSend a single zmq frame. Returns a Future that resolves when sending is complete. Recommend using send_multipart instead. rrrrsendr)updaterru)r*datarrrrs r"rz_AsyncSocket.send=sL wvw dT?@##FV#DDr!c|jfd}j|fd}j|S)zDeserialize with FuturescjrBjs1j!tjdddt yjr j jyj} |}j|y#t$r}j |Yd}~yd}~wwxYw)z+Chain result through serialization to recvdNzFuture z completed while awaiting z. A message has been dropped!) rJrYr\rrRuntimeWarningr]r`rKr_)_rloadedrarAloadrecvds r"_chainz)_AsyncSocket._deserialize.._chainUsvvx(U__->-FMM"!$>ugEbc&    12lln)!#YFLL(!'OOA&&'sB// C8CCcjjryjrjyy)z"Chain cancellation from f to recvdN)rJrYrZ)rrArs r" _chain_cancelz0_AsyncSocket._deserialize.._chain_cancelps&zz|{{} r!)r&rn)r*rrrrrAs `` @r" _deserializez_AsyncSocket._deserializeQs= LLN )2 '  M*r!cjr#tjtjj }|j |t t|j|jfd}jr |nj|fd}j|S)zSpoll the socket for events returns a Future for the poll results. c^jryjr jy|j r j j yt j}j|jdy#t$rYywxYwrX) rJrYrZr[r\r]rr`rKr)rAevtsr poll_futurer*s r" unwrap_resultz(_AsyncSocket.poll..unwrap_results{{}$$&MMO{{}$$[%:%:%<=K..01!!$((4"34$sB B,+B,chjs jyy#t$rYywxYw)z4Cancel underlying poll if request has been cancelledN)rJrZr[)rrs r" cancel_pollz&_AsyncSocket.poll..cancel_polls9##%&&(&$s % 11) rrpZMQErrorENOTSUP _poller_classregisterr rrUr&rJrn)r*rwrprrrrs` @@r"rUz_AsyncSocket.poll{s ;;-- - -     46166'?3  5     + &  ) )- 8    - r!c2fd}|j||S)z'Add a timeout for a send or recv Futurecnjryjtjyr1)rJr]rpAgainrVsr"future_timeoutz1_AsyncSocket._add_timeout..future_timeouts%{{}   .r!) _call_later)r*rrwrs ` r" _add_timeoutz_AsyncSocket._add_timeouts /88r!cB|jj||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,rv)r*delaycallbacks r"rz_AsyncSocket._call_laters~~**5(;;r!cL|sy |j|y#t$rYywxYw)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_futures/     e $   s  ##)rrrcj|xs|j}|d}|i}|jdrU|jddtjzr2t |j |} ||i|}|j||St} ttdr0|j j} | dk\r|j|| dz} t||||d| } |jj!| |j jt"t$zr|j'|jrT| |jvrF|j)t+|j,|j| |j/t$|S#t$r}|j|Yd}~|Sd}~wwxYw) z4Add a recv event, returning the corresponding FutureNr rrrRCVTIMEOrfrrrrrr)r& startswithrrpDONTWAITgetattrrrKr_r]r}rhrcvtimeorrrrsrr _handle_recvrnrr _add_io_state) r*rrrrrArrrar timeout_ms _future_events r"rtz_AsyncSocket._add_recv_events  $dlln <D >F ??6 "vzz'1'= 'M4,,d3D $)&) QH 4 $**33JQ))!Z$->?% t$v4u  !!-0     (6 1       -43E3E"E   00#11'    v &E #""H  #s(F F2F--F2c|xs|j}|dvr|js|jdd}|j}|tj z|d<t |j|}d} ||fi|} |j| | r|jr|j|St} ttdrC|jjtj } | dk\r|j#|| dz} t%||d ||| }|jj'||j)t+|j,|j| |j/t0|S#tj$r2} |tj zr|j| nd} Yd} ~ -d} ~ wt$r} |j| Yd} ~ Pd} ~ wwxYw) z4Add a send event, returning the corresponding Future)rrrrTFNSNDTIMEOrfr rr)r&rrrrprrrrKrr]r_rrr}rhrrrrsrnrrrr)r*rrrrrAr nowait_kwargsr finish_earlyrrarrrs r"ruz_AsyncSocket._add_send_event s  $dlln - -d6H6HJJw*E"KKMM%*T]]%:M' "4,,d3D L . . Q%%335 4 $**..t}}=JQ))!Z$->?% t"VE  !!-0  ,,--#   7#S:: )4==(OOA&$)L #"" #s$6 E==G%'F== G% G  G%c|jjttzsyd}|jrB|jj \}}}}}}|j rd}nn |jrB|js|jt|yjdk(r|jdy|dk(r|jj}nF|dk(r|jj}n*|dk(r|jj}ntd|dxxtjzcc< |i|}|j|y#t $r} |j#| Yd} ~ yd} ~ wwxYw)zHandle recv eventsNrUrrrzUnhandled recv event type: r)rrrrrpopleftrJ_drop_io_staterZrKrrrrrprr_r]) r*rArrrrrrr`ras r"rz_AsyncSocket._handle_recvJs\  $$V,v5    .2.@.@.H.H.J +AtT61evvx   !!    ' 9   6> LL   % %$$33D V^$$))D [ $$..D:4(CD Dw4==( !4*6*F LL   OOA   sE E?$E::E?c|jjttzsyd}|jrB|jj \}}}}}}|j rd}nn |jrB|js|jt|yjdk(r|jdy|dk(r|jj}n*|dk(r|jj}ntd|dxxtjzcc< |fi|}|j|y#t$r} |j!| Yd} ~ yd} ~ wwxYw)NrUrrzUnhandled send event type: r)rrrrrrrJrrZrKrrrrprr_r]) r*rArrrrrrr`ras r" _handle_sendz_AsyncSocket._handle_sendusC  $$V,w6    040B0B0J0J0L -AtT63vvx   !!    ( 9   6> LL   % %$$33D V^$$))D:4(CD Dw4==( !#((F LL   OOA   s% E E$ EE$c|jjry|jjt}|tj zr|j |tjzr|j|jy)z(Dispatch IO events to _handle_recv, etc.N) rrrrrprrrrr)r*fdevents zmq_eventss r"_handle_eventsz_AsyncSocket._handle_eventssf    # # &&**62  #      $     '')r!c|jdk(ry||jjt}||jzr|j d|j yy)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_eventssU ;;!   >&&**62F DKK    Q 3 3 4 r!c|j|k7r|j|zx}|_|j|jy)zAdd io_state to poller.Nr_update_handlerr*states r"rz_AsyncSocket._add_io_states5 ;;% "&++"5 5EDK T[[)r!c|j|zr|j|z|_|j|jy)z&Stop poller from watching an io_state.Nrrs r"rz_AsyncSocket._drop_io_states2 ;; ++%0DK T[[)r!cH|r|j|jy)zOUpdate IOLoop handler with state. zmq FD is always read-only. N)r,rrs r"rz_AsyncSocket._update_handlers  NN  '')r!c||j}|j|j|j|j|j d|jy)z#initialize the ioloop event handlerNr)r, add_handlerrrr;rr2s r"r)z_AsyncSocket._init_io_statesK <>>#D **D,?,?L D//0r!c|j}|jjr |j}|j|jj |yy)zNunregister the ioloop event handler called once during close N)rrrr%remove_handler)r*rs r"rQz_AsyncSocket._clear_io_statesL       # #B    )    - -b 1 *r!)NryNN)rz_zmq.Socket | Noner4r5r1)rztype[T]r?rrr r4r)rz int | Noner4r5)rTF)rr:rboolrrr4z)Awaitable[list[bytes] | list[_zmq.Frame]])rr:rrrrr4zAwaitable[bytes | _zmq.Frame])rr:rr:r4Awaitable[int])rr rr:rrr4%Awaitable[_zmq.MessageTracker | None]) rr rr:rrrrrr r4r )r4r ) rrrz tuple | Nonerzdict[str, Any] | Nonerz Future | Noner4r)NNN)rr)*rrrrrrrr8rrr classmethodrrrrprqr6rrrrrrrrrUrrrrrtrurrrrrrrr)rQrzr{s@r"rrsBMM F M C+/ ( ) ( (:99 %KK%%--EM ++//))CK@E    $(  8<  2  @E    $(  8<  &  ()q  !$  14    HM T T%( T48 T . T EEE E  E  E /E((T t{{0d 9<*"(, $ 55 5 & 5  5 5n;z)!V&!R *5 * * *1 2r!)!r6 __future__rrasyncior collectionsrcollections.abcr functoolsr itertoolsrtypingr r r r r zmqrprrrrr$Pollerr8r}rrqrr r!r"rs+#%'':$