:vjUdZddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl Z ddl m Z e Zded<Gd d eZ dd Zdd Zej(d k(reZneZGddZGddee j.ZGddee j2Zee_Gdde j8eZGddeZdZddZde_ ddZ!gdZ"y)z8AsyncIO support for zmq Requires asyncio and Python 3. ) annotationsN)FutureSelectorEventLoop)WeakKeyDictionary)_futurer _selectorsceZdZdZy)ProactorSelectorThreadWarningzWarning class for notifying about the extra thread spawned by tornado We automatically support proactor via tornado's AddThreadSelectorEventLoopN)__name__ __module__ __qualname____doc__\/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/zmq/asyncio.pyr r sRrr cRtvr tSttdrjttjrP ddlm}tjdtd|xt<jfd}|_ SS#t$r tdwxYw) zGet selector-compatible loop Returns an object with ``add_reader`` family of methods, either the loop itself or a SelectorThread instance. Workaround Windows proactor removal of *reader methods, which we need for zmq sockets. ProactorEventLoopr)AddThreadSelectorEventLoopaProactor event loop does not implement add_reader family of methods required for zmq. zmq will work with proactor if tornado >= 6.1 can be found. Use `asyncio.set_event_loop_policy(WindowsSelectorEventLoopPolicy())` or install 'tornado>=6.1' to avoid this error.aProactor event loop does not implement add_reader family of methods required for zmq. Registering an additional selector thread for add_reader support via tornado. Use `asyncio.set_event_loop_policy(WindowsSelectorEventLoopPolicy())` to avoid this warning. stacklevelc`_tjdjyN)closerpop) asyncio_loop loop_close selector_loopsr_close_selector_and_loopz7_get_selector_windows.._close_selector_and_loopMs'",L  NN< .    !r) rhasattrasyncio isinstancertornado.platform.asyncior ImportError RuntimeErrorwarningswarnRuntimeWarningr)rrrrrs` @@r_get_selector_windowsr)sz!,''w+,!!2  K   &  4N 4  <0 "''  "6 G B  s BB&c|S)zno-op on non-Windowsr)loops r_get_selector_noopr,Zs Krwin32cHeZdZeZej ZejZ ddZ y)_AsyncIOc tjS#t$r2tjdt dYtj SwxYw)NuONo running event loop. zmq.asyncio should be used from within an asyncio loop.r)r!get_running_loopr%r&r'r(get_event_loopselfs r _default_loopz_AsyncIO._default_loopjsK ++- -  MMa  %%'' s%AANreturnzasyncio.AbstractEventLoop) r r r r_Future selectors EVENT_WRITE_WRITE EVENT_READ_READr6rrrr/r/es"G  " "F  E (rr/ceZdZdZdZdZy)Pollerz1Poller returning asyncio.Future for poll results.ct|}||jzr|j|fd||jzr|j |fdyy)z"Schedule callback for a raw socketcSrrargsfs rz*Poller._watch_raw_socket..~acrcSrrrCs rrFz*Poller._watch_raw_socket..rGrN) _get_selectorr> add_readerr< add_writer)r5r+socketevtrEselectors ` r_watch_raw_socketzPoller._watch_raw_socketzsL &     (9 :     (9 : rclt|}|D]$}|j||j|&y)z$Unschedule callback for a raw socketN)rI remove_reader remove_writer)r5r+socketsrNrLs r_unwatch_raw_socketszPoller._unwatch_raw_socketss2 &F  " "6 *  " "6 *rN)r r r rrOrTrrrr@r@ws;;+rr@c*eZdZdZeZddZddZdZy)Socketz.sd11!Q7rN)rIrJ_fdrYs` r_init_io_statezSocket._init_io_states& 7#.. HH7 rc|j}|rL|js;|jdk7r+|j|j |jyyyy)zEclear any ioloop event handler called once at close N) _current_loop is_closedr^rIrQ)r5r+s r_clear_io_statezSocket._clear_io_statesL !! (TXX^   t $ 2 2488 <.<(4rr) r r r rr@ _poller_classrIr_rdrrrrVrVsFM&  =rrVc@eZdZdZeZdZ d dfd ZxZS)Contextz/Context for creating asyncio-compatible SocketsNc&t|||yr)super__init__)r5 io_threadsshadow __class__s rrjzContext.__init__s V,r)r)r5rgrkzint | _zmq.Contextrlz_zmq.Context | intr8None) r r r rrV _socket_class _instancerj __classcell__rms@rrgrgsH9MI *+%&--&-#-  --rrgc&eZdZdZddfd ZxZS) ZMQEventLoopz}DEPRECATED: AsyncIO eventloop using zmq_poll. pyzmq sockets should work with any asyncio event loop as of pyzmq 17. c6tt| |Sr) _deprecatedrirj)r5rNrms rrjzZMQEventLoop.__init__s w))rr)rNzselectors.BaseSelector | Noner8ro)r r r rrjrrrss@rrurus **rructtjrydt_tjdtdy)NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.r)rwcalledr&r'DeprecationWarningrrrrwrws+K MM}rFcty)z(DEPRECATED: No longer needed in pyzmq 17N)rwrrrinstallr}sMr)rgrVr@rur}r7)r8ro)#r __future__rr!r:sysr&rrweakrefrzmq_zmqrr__annotations__r(r r)r,platformrIr/ _AsyncPollerr@ _AsyncSocketrVrprgru_looprwrzr}__all__rrrrs# -%!2 3 3RNR 99x <<7)M&M(($+Xw+++&=Xw++=4-dll6"-"*$*     r