B a@s@UdZddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl Z ddl m Z e Ze ed<Gdd d eZejd d d Zejd d dZejdkreZneZGdddeZGdddee jZGdddee jZee_Gddde jZGdddeZdZ ddZ!de!_"ddZ#dddddgZ$dS)z8AsyncIO support for zmq Requires asyncio and Python 3. N)SelectorEventLoopFuture)Union)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-xvwjx0np/zmq/asyncio.pyrsr)returncstkrtSttdrttjryddlm}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_tddS)N)closerpopr ) asyncio_loop loop_close selector_loopr r_close_selector_and_loopJs z7_get_selector_windows.._close_selector_and_loopN) rhasattrasyncio isinstancerZtornado.platform.asyncior ImportError RuntimeErrorwarningswarnRuntimeWarningr)rrrr )rrrr_get_selector_windowss&    r"cCs|S)zno-op on non-Windowsr )loopr r r_get_selector_noopWsr$win32c@s$eZdZeZejZejZ ddZ dS)_AsyncIOcCstS)N)rZget_event_loop)selfr r r _default_loopgsz_AsyncIO._default_loopN) r r r rZ_Future selectors EVENT_WRITE_WRITE EVENT_READ_READr(r r r rr&bsr&c@s eZdZdZddZddZdS)Pollerz1Poller returning asyncio.Future for poll results.csHt|}||j@r&||fdd||j@rD||fdddS)z"Schedule callback for a raw socketcsS)Nr )args)fr rrz*Poller._watch_raw_socket..csS)Nr )r/)r0r rr1tr2N) _get_selectorr- add_readerr+Z add_writer)r'r#socketZevtr0selectorr )r0r_watch_raw_socketns   zPoller._watch_raw_socketcGs.t|}x |D]}||||qWdS)z$Unschedule callback for a raw socketN)r3 remove_readerZ remove_writer)r'r#Zsocketsr6r5r r r_unwatch_raw_socketsvs  zPoller._unwatch_raw_socketsN)r r r r r7r9r r r rr.ksr.c@s4eZdZdZeZdZeddZddZ ddZ dS) Socketz.N)r=r4_fd)r'r )r'r_init_io_stateszSocket._init_io_statecCs|js|j|jdS)zEclear any ioloop event handler called once at close N)r<Z is_closedr=r8r>)r'r r r_clear_io_states zSocket._clear_io_state) r r r r r.Z _poller_classr;propertyr=r?r@r r r rr:~s  r:c@seZdZdZeZdZdS)Contextz/Context for creating asyncio-compatible SocketsN)r r r r r: _socket_classZ _instancer r r rrBsrBcs"eZdZdZdfdd ZZS) ZMQEventLoopz}DEPRECATED: AsyncIO eventloop using zmq_poll. pyzmq sockets should work with any asyncio event loop as of pyzmq 17. Ncsttt||S)N) _deprecatedsuperrD__init__)r'r6) __class__r rrGszZMQEventLoop.__init__)N)r r r r rG __classcell__r r )rHrrDsrDcCs$tjr dSdt_tjdtdddS)NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.)r)rEcalledrr DeprecationWarningr r r rrEsrEFcCs tdS)z(DEPRECATED: No longer needed in pyzmq 17N)rEr r r rinstallsrM)%r rr)sysrrrtypingrweakrefrZzmqZ_zmqrr__annotations__r!rZAbstractEventLoopr"r$platformr3objectr&Z _AsyncPollerr.Z _AsyncSocketr:rCrBrDZ_looprErKrM__all__r r r rs>   7