U lu f@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 //tmp/pip-unpacked-wheel-h6ekxre8/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$win32c@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 selectors EVENT_WRITE_WRITE 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 r5r7r r r9}r:N) _get_selectorr3 add_readerr1Z add_writer)r-r#socketZevtr8selectorr r7r _watch_raw_socketws   zPoller._watch_raw_socketcGs*t|}|D]}||||q dS)z$Unschedule callback for a raw socketN)r; remove_readerZ remove_writer)r-r#Zsocketsr>r=r r r _unwatch_raw_socketss zPoller._unwatch_raw_socketsN)rr r r r?rAr r r r r4tsr4c@s0eZdZdZeZd ddZd ddZddZdS) Socketz.N)r;r<_fdrCr 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_closedrDr;r@)r-r#r r r _clear_io_stateszSocket._clear_io_state)N)N) rr r r r4Z _poller_classr;rErGr r r r rBs   rBc@seZdZdZeZdZdS)Contextz/Context for creating asyncio-compatible SocketsN)rr r r rB _socket_classZ _instancer r r r rHsrHcs"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 rMszZMQEventLoop.__init__)N)rr r r rM __classcell__r r rNr rJsrJcCs$tjr dSdt_tjdtdddS)NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.r'r)rKcalledrr DeprecationWarningr r r r rKsrKFcCs tdS)z(DEPRECATED: No longer needed in pyzmq 17N)rKr r r r installsrS)"r rr/r*rrrweakrefrZzmqZ_zmqrr__annotations__r!rZAbstractEventLoopr"r$platformr;r&Z _AsyncPollerr4Z _AsyncSocketrBrIrHrJZ_looprKrQrS__all__r r r r s@   9