EfUdZddlZddlZddlZddlZddlmZmZddlmZddl Z ddl m Z eZ ee d<GddeZd ej fd Zd ej fd Zej&d k(reZneZGd dZGddee j,ZGddee j0Zee_Gdde j6eZGddeZdZdZde_dZ gdZ!y)z8AsyncIO support for zmq Requires asyncio and Python 3. N)FutureSelectorEventLoop)WeakKeyDictionary)_future _selectorsceZdZdZy)ProactorSelectorThreadWarningzWarning class for notifying about the extra thread spawned by tornado We automatically support proactor via tornado's AddThreadSelectorEventLoopN)__name__ __module__ __qualname____doc__-/usr/lib/python3/dist-packages/zmq/asyncio.pyr r sRrr returncRtvr 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_loopIs'",L  NN< .    !r) rhasattrasyncio isinstancertornado.platform.asyncior ImportError RuntimeErrorwarningswarnRuntimeWarningr)rrrrrs` @@r_get_selector_windowsr)sz!,''w+,g//2  K   &  4Nl3[[ <0"''  "6 C B  s BB&c|S)zno-op on non-Windowsr)loops r_get_selector_noopr,Vs Krwin32cFeZdZeZej ZejZ dZ y)_AsyncIOctjdk\r tjStjS#t$r2t j dtdYtjSwxYw)N)uONo running event loop. zmq.asyncio should be used from within an asyncio loop.r) sys version_infor!get_running_loopr%r&r'r(get_event_loopselfs r _default_loopz_AsyncIO._default_loopfsl   v % //11%%''   e"  %%'' s=%A87A8N) r r r r_Future selectors EVENT_WRITE_WRITE EVENT_READ_READr:rrrr/r/as"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..{ acrcSrrrEs rrHz*Poller._watch_raw_socket..}rIrN) _get_selectorr@ add_readerr> add_writer)r9r+socketevtrGselectors ` r_watch_raw_socketzPoller._watch_raw_socketwsL &     (9 :     (9 : rclt|}|D]$}|j||j|&y)z$Unschedule callback for a raw socketN)rK remove_reader remove_writer)r9r+socketsrPrNs r_unwatch_raw_socketszPoller._unwatch_raw_socketss7 & +F  " "6 *  " "6 * +rN)r r r r rQrVrrrrBrBts;;+rrBc*eZdZdZeZddZddZdZy)Socketz.sd11!Q7rN)rKrL_fdr[s` 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`rKrS)r9r+s r_clear_io_statezSocket._clear_io_statesL !! (TXX^   t $ 2 2488 <.<(4rr) r r r r rB _poller_classrKrarfrrrrXrXsFM&  =rrXceZdZdZeZdZy)Contextz/Context for creating asyncio-compatible SocketsN)r r r r rX _socket_class _instancerrrriris9MIrric$eZdZdZdfd ZxZS) ZMQEventLoopz}DEPRECATED: AsyncIO eventloop using zmq_poll. pyzmq sockets should work with any asyncio event loop as of pyzmq 17. c6tt| |Sr) _deprecatedsuper__init__)r9rP __class__s rrqzZMQEventLoop.__init__s w))rr)r r r r rq __classcell__)rrs@rrmrms **rrmcttjrydt_tjdtdy)NTzsZMQEventLoop and zmq.asyncio.install are deprecated in pyzmq 17. Special eventloop integration is no longer needed.r1r)rocalledr&r'DeprecationWarningrrrroros+K MM}rFcty)z(DEPRECATED: No longer needed in pyzmq 17N)rorrrinstallrxsMr)rirXrBrmrx)"r r!r<r4r&rrweakrefrzmq_zmqrr__annotations__r(r AbstractEventLoopr)r,platformrKr/ _AsyncPollerrB _AsyncSocketrXrjrirm_looprorurx__all__rrrrs -%!2 3 3RNR 6 6r 9 9 <<7)M&M((&+Xw+++&=Xw++=4dll6"*$*     r