o ṏiv@s0UdZddlmZddlZddlZddlZddlZddlmZmZddl m Z ddl Z ddl m Z e Zded<Gd d d eZd"d dZd"ddZejdkrReZneZGdddZGdddee jZGdddee jZee_Gddde jeZGdddeZdZddZde_ dd Z!gd!Z"dS)#z8AsyncIO support for zmq Requires asyncio and Python 3. ) annotationsN)FutureSelectorEventLoop)WeakKeyDictionary)_futurer _selectorsc@seZdZdZdS)ProactorSelectorThreadWarningzWarning class for notifying about the extra thread spawned by tornado We automatically support proactor via tornado's AddThreadSelectorEventLoopN)__name__ __module__ __qualname____doc__r r H/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/zmq/asyncio.pyrsrreturnasyncio.AbstractEventLoopcstvrtSttdrGttjrGzddlm}Wn ty&tdwt j dt dd|t<j fdd }|_ SS) 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. stacklevelcs_tddSN)closerpopr  asyncio_loop loop_close selector_loopr r_close_selector_and_loopMs  z7_get_selector_windows.._close_selector_and_loop) rhasattrasyncio isinstancertornado.platform.asyncior ImportError RuntimeErrorwarningswarnRuntimeWarningr)rrrr rr_get_selector_windowss4     r'cCs|S)zno-op on non-Windowsr )loopr r r_get_selector_noopZsr)win32c@s$eZdZeZejZejZ ddZ dS)_AsyncIOcCs4ztWStytjdtddYtSw)NuONo running event loop. zmq.asyncio should be used from within an asyncio loop.r)rget_running_loopr#r$r%r&get_event_loopselfr r r _default_loopjs  z_AsyncIO._default_loopN) r r r r_Future selectors EVENT_WRITE_WRITE EVENT_READ_READr1r r r rr+es  r+c@s eZdZdZddZddZdS)Pollerz1Poller returning asyncio.Future for poll results.csLt|}||j@r||fdd||j@r$||fdddSdS)z"Schedule callback for a raw socketcSrr argsfr r~z*Poller._watch_raw_socket..cr9rr r:r<r rr>r?N) _get_selectorr7 add_readerr5 add_writer)r0r(socketevtr=selectorr r<r_watch_raw_socketzs   zPoller._watch_raw_socketcGs*t|}|D] }||||qdS)z$Unschedule callback for a raw socketN)r@ remove_reader remove_writer)r0r(socketsrErCr r r_unwatch_raw_socketss   zPoller._unwatch_raw_socketsN)r r r r rFrJr r r rr8ws r8c@s0eZdZdZeZd ddZd ddZddZdS) Socketzs z'Socket._init_io_state..N)r@rA_fdrMr r/r_init_io_states zSocket._init_io_statecCs>|j}|r|s|jdkr|||jdSdSdSdS)zEclear any ioloop event handler called once at close N) _current_loop is_closedrPr@rG)r0r(r r r_clear_io_states zSocket._clear_io_stater) r r r r r8 _poller_classr@rQrUr r r rrKs    rKcs0eZdZdZeZdZ  ddfd d ZZS)Contextz/Context for creating asyncio-compatible SocketsNrr0 io_threadsint | _zmq.Contextshadow_zmq.Context | intrNonecst||dSr)super__init__)r0rYr[ __class__r rr_szContext.__init__)rXr)r0rWrYrZr[r\rr]) r r r r rK _socket_class _instancer_ __classcell__r r r`rrWsrWcs"eZdZdZdfdd ZZS) ZMQEventLoopz}DEPRECATED: AsyncIO eventloop using zmq_poll. pyzmq sockets should work with any asyncio event loop as of pyzmq 17. Ncstt|Sr) _deprecatedr^r_)r0rEr`r rr_s zZMQEventLoop.__init__r)r r r r r_rdr r r`rresrecCs$tjrdSdt_tjdtdddS)NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.r)rfcalledr$r%DeprecationWarningr r r rrfs rfFcCs tdS)z(DEPRECATED: No longer needed in pyzmq 17N)rfr r r rinstalls rj)rWrKr8rerj)rr)#r __future__rrr3sysr$rrweakrefrzmq_zmqrr__annotations__r&rr'r)platformr@r+ _AsyncPollerr8 _AsyncSocketrKrbrWre_looprfrhrj__all__r r r rs8     <