Mh, UdZddlmZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl m Z ddl mZdd lmZee d re je jfZd ed <nd ZdZGddeZGdde eZdS)ziThis module wraps the :class:`Socket` and :class:`Context` found in :mod:`pyzmq ` to be non blocking) annotationsN) AsyncResult)get_hub)Context)Socket)_PollerRCVTIMEOtupleTIMEOScz |dS#t$r|YdSwxYw)zNsimple wrapper for stopping an Event, allowing for method rename in gevent 1.0N)stopAttributeErrorcancel)evts N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/zmq/green/core.py_stopr#sF   s ::ceZdZdZdZdZdZdZdZdZ dZ e Z dZ fdZdZdfd Zd Zd Zdfd Zd Zd Zdfd Zdfd Zdddfd ZfdZfdZfdZfdZxZS)_SocketaKGreen version of :class:`zmq.Socket` The following methods are overridden: * send * recv To ensure that the ``zmq.NOBLOCK`` flag is set and that sending or receiving is deferred to the hub if a ``zmq.EAGAIN`` (retry) error is raised. The `__state_changed` method is triggered when the zmq.FD for the socket is marked as readable and triggers the necessary read and write events (which are waited for in the recv and send methods). Some double underscore prefixes are used to minimize pollution of :class:`zmq.Socket`'s namespace. FNg333333'@zzmq.green.Socketc~tj|i|d|_d|_|dS)NF)super__init___Socket__in_send_multipart_Socket__in_recv_multipart_Socket__setup_events)selfakw __class__s rrz_Socket.__init__IsG!"r"""#( #(  c.|dSN)closers r__del__z_Socket.__del__Os r!crt||dSr#)rr$_Socket__cleanup_events)rlingerr s rr$z _Socket.closeRs1  f r!ct|ddrt|jd|_|j|jdS)N _state_event)getattrrr+_Socket__writableset_Socket__readabler%s r__cleanup_eventsz_Socket.__cleanup_eventsVs] 4 . . % $# $ $ $ $D   r!c t|_t|_|j|j t j|tj d|_ |j |j dS#t$r?ddlm}||tj |j d|_ YdSwxYw)Nrr) read_eventT)persist)rr/r-r.rloopio getsockoptzmqFDr+start_Socket__state_changedr gevent.corer2)rr2s r__setup_eventsz_Socket.__setup_events_s%--%--    '  1 1''!!D    # #D$8 9 9 9 9 9    . . . . . . * '')=t!!!D      sA(CAD  D c|jr|dS ttj}|tjzr|j|tj zr|j dSdS#tj $r?}|j ||j |Yd}~dSd}~wwxYwr#) closedr(rr6r7EVENTSPOLLOUTr-r.POLLINr/ZMQError set_exception)revent_evtypeeventsexcr s r__state_changedz_Socket.__state_changedrs ;   ! ! # # # F &WW'' 33F  # &##%%% " &##%%%%% & & | / / / O ) )# . . . O ) )# . . . . . . . . . /s+BC-.4C((C-c2|js Jdt|_tj}|j}|rt j|}nd} |r||jdn#t j$r}||urtj}|j r[|rY||z |krP| tj tj zr$td|jdt jYd}~nd}~wwxYw|r||jdS#|r||jwxYw)N.Only one greenlet can be waiting on this eventsecondsTblockz3BUG: gevent may have missed a libzmq send event on !file)r-readyrtime_gevent_bug_timeoutgeventTimeoutr9get _debug_geventr6r7r?r@printr8sysstderrr$r.rticdttimeoutttocs r _wait_writez_Socket._wait_writes$$&&XX(XXXX%--ikk  %  nR000GGG "   O  d  + + + +~   )++C"  #INNOOCJ//#+=#T$'TTT "   O   ! ! ! ! !   O   ! ! ! !1,1BE%D1-A:D,'E%,D11E%%1Fc2|js Jdt|_tj}|j}|rt j|}nd} |r||jdn#t j$r}||urtj}|j r[|rY||z |krP| tj tj zr$td|jdt jYd}~nd}~wwxYw|r||jdS#|r||jwxYw)NrJrKTrMz3BUG: gevent may have missed a libzmq recv event on rOrP)r/rRrrSrTrUrVr9rWrXr6r7r?rArYr8rZr[r$r.r\s r _wait_readz_Socket._wait_reads$$&&XX(XXXX%-- ikk  %  nR000GGG "   O  d  + + + +~   )++C"  #INNOOCJ//#*<#T$'TTT "   O   ! ! ! ! !   O   ! ! ! !rcrTc D|tjzr] tj||||fi|}|js|n!#|js|wwxYw|S|tjz} t||||}|js||S#tj$r;}|jtjkr|js|Yd}~nd}~wwxYw| )zsend, which will only block current greenlet state_changed always fires exactly once (success or fail) at the end of this method. TN) r7NOBLOCKrsendrr:rBerrnoEAGAINrb) rdataflagscopytrackkwargsmsger s rrhz _Socket.sends[ 3;   +"eggl4eFFvFF/+((***/+((****+J   ggll4e<</+((*** <   7cj((3/,,...)((((        s#A A)?$CD 1DD cN|tjzrc t|||}|js|n!#|js|wwxYw|S|tjz} t|||}|js||S#tj$r;}|jtjkr|js|Yd}~nd}~wwxYw| )zrecv, which will only block current greenlet state_changed always fires exactly once (success or fail) at the end of this method. TN) r7rgrrecvrr:rBrirjre)rrlrmrnrprqr s rrsz _Socket.recvsM 3;   +ggll5$66/+((***/+((****+J   ggll5$66/+((*** <   7cj((3/,,...)((((  OO    s##AA/#CD1D  Dnbytesrlc|tjzr$t|||S|tjz} t|||}||S#tj$r4}|jtjkr|Yd}~nd}~wwxYw|)z1recv_into, which will only block current greenletrtTN) r7DONTWAITr recv_intor:rBrirjre)rbufferrurlrecvdrqr s rrxz_Socket.recv_into s 3<  I77$$VF%$HH H   ))&u)MM $$&&& <   7cj((((***)((((  OO    s$BC*B>>Ccd|_ tj|i|}d|_|n #d|_|wxYw|S)zAwrap send_multipart to prevent state_changed on each partial sendTF)rrsend_multipartr:rargsrorpr s rr|z_Socket.send_multipartp#'  #(%''($9&99C',D $  " " " "(-D $  " " " " ?Acd|_ tj|i|}d|_|n #d|_|wxYw|S)zAwrap recv_multipart to prevent state_changed on each partial recvTF)rrrecv_multipartr:r}s rrz_Socket.recv_multipart&rrc|tvrtjdtt |}|t jkr||S)z+trigger state_changed on getsockopt(EVENTS)0TIMEO socket options have no effect in zmq.green) r warningswarn UserWarningrrWr7r?r:)roptoptvalr s rrWz _Socket.get0s` &== MBK   S!! #*    " " " r!c|tvrtjdtt ||S)zset socket optionr)r rrrrr.)rrvalr s rr.z _Socket.set;s? &== MBK   ww{{3$$$r!r#)NN)rTF)__name__ __module__ __qualname____doc__rrr-r/r+rTrXr _poller_class _repr_clsrr&r$r(rr:rbrerhrsrxr|rrWr. __classcell__)r s@rrr,s$ JJLMM"I       &&&&&&& """"""H$"$"$"L      D:./a"     %%%%%%%%%r!rceZdZdZeZdZdZdS)_ContextzrReplacement for :class:`zmq.Context` Ensures that the greened Socket above is used in calls to `socket`. zzmq.green.ContextN)rrrrr _socket_classr _instancer r!rrrDs+ M#IIIIr!r)r __future__rrZrSrrU gevent.eventr gevent.hubrr7r_original_Contextr_original_Socketpollr hasattrr SNDTIMEOr __annotations__rrrr r!rrs]poo""""""  $$$$$$ ,,,,,,****** 73 \3<0F00000 FU%U%U%U%U%U%U%U%p      )     r!