U Yc@s6UdZddlZddlZddlZddlZddlmZmZddlmZddl Z ddl m Z eZ ee d<GdddeZejd d d Zejd d d ZejdkreZneZGdddZGdddee jZGdddee jZee_Gddde jeZGdddeZdZddZde_ddZ dddddgZ!dS)z8AsyncIO support for zmq Requires asyncio and Python 3. N)FutureSelectorEventLoop)WeakKeyDictionary)_future _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 *lib/python3.8/site-packages/zmq/asyncio.pyrsr)returncstkrtSttdrttjrzddlm}Wntk rRtdYnXt j dt dd|t<j fdd }|_ SSd S) 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_loopZ loop_closeZ selector_loopr r _close_selector_and_loopIs z7_get_selector_windows.._close_selector_and_loopN) rhasattrasyncio isinstancerZtornado.platform.asyncior ImportError RuntimeErrorwarningswarnRuntimeWarningr)rrrr rr _get_selector_windowss.    r"cCs|S)zno-op on non-Windowsr )loopr r r _get_selector_noopVsr$Zwin32c@s$eZdZeZejZejZ ddZ dS)_AsyncIOcCsBtjdkr:z tWStk r8tjdtddYnXtS)N)uONo running event loop. zmq.asyncio should be used from within an asyncio loop.r) sys version_inforZget_running_looprrr r!Zget_event_loopselfr r r _default_loopfs   z_AsyncIO._default_loopN) rr r rZ_Future selectorsZ EVENT_WRITE_WRITEZ EVENT_READ_READr-r r r r r%asr%c@s eZdZdZddZddZdS)Pollerz1Poller returning asyncio.Future for poll results.csHt|}||j@r&||fdd||j@rD||fdddS)z"Schedule callback for a raw socketcsSrr argsfr r {z*Poller._watch_raw_socket..csSrr r2r4r r r6}r7N) _get_selectorr0 add_readerr/Z add_writer)r,r#socketZevtr5selectorr r4r _watch_raw_socketws   zPoller._watch_raw_socketcGs*t|}|D]}||||q dS)z$Unschedule callback for a raw socketN)r8 remove_readerZ remove_writer)r,r#Zsocketsr;r:r r r _unwatch_raw_socketss zPoller._unwatch_raw_socketsN)rr r r r<r>r r r r r1tsr1c@s0eZdZdZeZd ddZd ddZddZdS) Socketz.N)r8r9_fdr@r r+r _init_io_states  zSocket._init_io_statecCs2|j}|r.|s.|jdkr.|||jdS)zEclear any ioloop event handler called once at close N)Z _current_loopZ is_closedrAr8r=)r,r#r r r _clear_io_stateszSocket._clear_io_state)N)N) rr r r r1Z _poller_classr8rBrDr r r r r?s   r?c@seZdZdZeZdZdS)Contextz/Context for creating asyncio-compatible SocketsN)rr r r r? _socket_classZ _instancer r r r rEsrEcs"eZdZdZdfdd ZZS) ZMQEventLoopz}DEPRECATED: AsyncIO eventloop using zmq_poll. pyzmq sockets should work with any asyncio event loop as of pyzmq 17. Ncstt|Sr) _deprecatedsuper__init__)r,r; __class__r r rJszZMQEventLoop.__init__)N)rr r r rJ __classcell__r r rKr rGsrGcCs$tjr dSdt_tjdtdddS)NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.r&r)rHcalledrr DeprecationWarningr r r r rHsrHFcCs tdS)z(DEPRECATED: No longer needed in pyzmq 17N)rHr r r r installsrP)"r rr.r)rrrweakrefrZzmqZ_zmqrr__annotations__r!rZAbstractEventLoopr"r$platformr8r%Z _AsyncPollerr1Z _AsyncSocketr?rFrErGZ_looprHrNrP__all__r r r r s@   9