ó L]c@s™dZddlmZmZddlmZddlmZmZm Z ddlZ eddƒZ d e j fd „ƒYZ d e jfd„ƒYZdS(s%Future-returning APIs for coroutines.iÿÿÿÿ(t namedtupletdeque(tchain(tEVENTStPOLLOUTtPOLLINNt _FutureEventtfuturetkindtkwargstmsgt _AsyncPollercBseZdZdd„ZRS(s:Poller that returns a Future on poll, instead of blocking.iÿÿÿÿc sQˆjƒ‰|dkrhyttˆƒjdƒ}Wn tk rV}ˆj|ƒnXˆj|ƒˆSˆjƒ‰ˆjƒ‰g‰‡fd†}ˆj‡‡‡fd†ƒxüˆj D]ñ\}}t |t j ƒrIt |ˆj ƒsˆj j|ƒ}n|t j@r#|jddˆƒn|t j@r¬|jddˆƒq¬q»ˆj|ƒd}|t j@ry|ˆjO}n|t j@r–|ˆjO}nˆjˆ|||ƒq»W‡‡‡fd†}ˆj|ƒ|d k r1|dkr1‡fd†} ˆjd|| ƒ‰‡‡fd †} ˆj| ƒn‡fd †} ˆj| ƒˆS( s Return a Future for a poll eventics ˆjƒsˆjdƒndS(N(tdonet set_resulttNone(targs(twatcher(s*lib/python2.7/site-packages/zmq/_future.pytwake_raw-s csˆjˆˆŒS(N(t_unwatch_raw_sockets(tf(tloopt raw_socketstself(s*lib/python2.7/site-packages/zmq/_future.pyt1ttpollRcs´ˆjƒrdSˆjƒrByˆjƒWntk r=nXdSˆjƒrdˆjˆjƒƒnLyttˆƒjdƒ}Wn t k r¢}ˆj|ƒnXˆj |ƒdS(Ni( R t cancelledtcancelt RuntimeErrort exceptiont set_exceptiontsuperR Rt ExceptionR (Rtresultte(RRR(s*lib/python2.7/site-packages/zmq/_future.pyt on_poll_readyEs    cs ˆjƒsˆjdƒndS(N(R R R((R(s*lib/python2.7/site-packages/zmq/_future.pyttrigger_timeout\s gü©ñÒMbP?cs-tˆdƒrˆjƒn ˆjˆƒdS(NR(thasattrRtremove_timeout(R(Rttimeout_handle(s*lib/python2.7/site-packages/zmq/_future.pytcancel_timeoutds csˆjƒsˆjƒndS(N(R R(R(R(s*lib/python2.7/site-packages/zmq/_future.pytcancel_watcherks N(t_FutureRR RR RR t _default_looptadd_done_callbacktsocketst isinstancet_zmqtSockett _socket_classt from_socketRt_add_recv_eventRt_add_send_eventtappendt_READt_WRITEt_watch_raw_socketRt call_later( RttimeoutR!R"RtsockettmasktevtR#R$R(R)((RRRRR'Rs*lib/python2.7/site-packages/zmq/_future.pyRsP             (t__name__t __module__t__doc__R(((s*lib/python2.7/site-packages/zmq/_future.pyR st _AsyncSocketcBsŽeZdZdZdZdZeZdZ dZ dddd„Z e dd„ƒZ dd„Zejjje_d„Zejjje_deed„Zdeed„Zdeed„Zdeed „Zd „Zdejd „Zd „Zd „Zed„ƒZddd„Z dddd„Z!d„Z"d„Z#ddd„Z$dd„Z%d„Z&d„Z'd„Z(d„Z)d„Z*RS(iiÿÿÿÿcKsït|tjƒr"d|}}n|jddƒ}|dk rhtt|ƒjd|jƒ||_ n4tt|ƒj|||tjj |jƒ|_ |p«|j ƒ|_ t ƒ|_t ƒ|_d|_|j j|_|jƒdS(Nt _from_sockettshadowi(R.R/R0RtpopRRAt__init__t underlyingt _shadow_sockRCR+tio_loopRt _recv_futurest _send_futurest_statetFDt_fdt_init_io_state(Rtcontextt socket_typeRHR R2((s*lib/python2.7/site-packages/zmq/_future.pyRE}s     cCs|d|d|ƒS(s.Create an async socket from an existing SocketRBRH((tclsR;RH((s*lib/python2.7/site-packages/zmq/_future.pyR2scCs|jssxZtt|j|jƒƒD]=}|jjƒs%y|jjƒWqbtk r^qbXq%q%W|j ƒnt t |ƒj d|ƒdS(Ntlinger( tclosedtlistRRIRJRR RRt_clear_io_stateRRAtclose(RRRtevent((s*lib/python2.7/site-packages/zmq/_future.pyRV”s "   cCs8tt|ƒj|ƒ}|tkr4|j|ƒn|S(N(RRAtgetRt_schedule_remaining_events(RtkeyR!((s*lib/python2.7/site-packages/zmq/_future.pyRX¡s c Cs%|jdtd|d|d|ƒƒS(s~Receive a complete multipart zmq message. Returns a Future whose result will be a multipart message. trecv_multiparttflagstcopyttrack(R3tdict(RR\R]R^((s*lib/python2.7/site-packages/zmq/_future.pyR[¨s c Cs%|jdtd|d|d|ƒƒS(s™Receive a single zmq frame. Returns a Future, whose result will be the received frame. Recommend using recv_multipart instead. trecvR\R]R^(R3R_(RR\R]R^((s*lib/python2.7/site-packages/zmq/_future.pyR`±s cKs7||d<||d<||d<|jdd|d|ƒS(sqSend a complete multipart zmq message. Returns a Future that resolves when sending is complete. R\R]R^tsend_multipartR R (R4(RR R\R]R^R ((s*lib/python2.7/site-packages/zmq/_future.pyRa¼s   cKsY||d<||d<||d<|jtd|d|d|ƒƒ|jdd|d|ƒS(s”Send a single zmq frame. Returns a Future that resolves when sending is complete. Recommend using send_multipart instead. R\R]R^tsendR R (tupdateR_R4(RR R\R]R^R ((s*lib/python2.7/site-packages/zmq/_future.pyRbÆs    "csQ|jƒ‰‡‡‡fd†}ˆj|ƒ‡‡fd†}ˆj|ƒˆS(sDeserialize with Futurescs‚ˆjƒrdSˆjƒr2ˆjˆjƒƒnLˆjƒ}yˆ|ƒ}Wn tk rp}ˆj|ƒnXˆj|ƒdS(s+Chain result through serialization to recvdN(R RRR!R R (t_tbuftloadedR"(Rtloadtrecvd(s*lib/python2.7/site-packages/zmq/_future.pyt_chainÖs   cs-ˆjƒrdSˆjƒr)ˆjƒndS(s"Chain cancellation from f to recvdN(R RR(Rd(RRh(s*lib/python2.7/site-packages/zmq/_future.pyt _chain_cancelæs  (R*R,(RRhRgRiRj((RRgRhs*lib/python2.7/site-packages/zmq/_future.pyt _deserializeÓs    cs‘ˆjrtjtjƒ‚nˆjƒ}|jˆ|ƒ|j|ƒ}ˆjƒ‰‡‡fd†}|jƒr€||ƒn |j |ƒˆS(sSpoll the socket for events returns a Future for the poll results. cs“ˆjƒrdS|jƒrByˆjƒWntk r=nXdS|jƒrdˆj|jƒƒn+t|jƒƒ}ˆj|j ˆdƒƒdS(Ni( R RRRRRR_R!R RX(Rtevts(RR(s*lib/python2.7/site-packages/zmq/_future.pyt unwrap_resultþs    ( RSR/tZMQErrortENOTSUPt _poller_classtregisterRR*R R,(RR:R\tpRRm((RRs*lib/python2.7/site-packages/zmq/_future.pyRðs      cs#‡fd†}|j||ƒdS(s'Add a timeout for a send or recv Futurecs'ˆjƒrdSˆjtjƒƒdS(N(R RR/tAgain((R(s*lib/python2.7/site-packages/zmq/_future.pytfuture_timeouts N(t _call_later(RRR:Rt((Rs*lib/python2.7/site-packages/zmq/_future.pyt _add_timeoutscCs|jj||ƒdS(sÊSchedule a function to be called later Override for different IOLoop implementations Tornado and asyncio happen to both have ioloop.call_later with the same signature. N(RHR9(Rtdelaytcallback((s*lib/python2.7/site-packages/zmq/_future.pyRu scCsqt|dtƒrdSx:t|ƒD](\}\}}}}||kr#Pq#q#WdSt|_|j||ƒdS(s°Make sure that futures are removed from the event list when they resolve Avoids delaying cleanup until the next send/recv event, which may never come. t _pyzmq_poppedN(tgetattrtFalset enumeratetTrueRytremove(Rt event_listtf_idxRRR Rd((s*lib/python2.7/site-packages/zmq/_future.pyt_remove_finished_future*s%  c sI|pˆjƒ}|jdƒr|jddƒtj@rtˆj|ƒ}y||}Wn tk r~}|j|ƒnX|j |ƒ|Sˆj j t |||ddƒƒ|j‡fd†ƒttdƒr ˆjj}|dkr ˆj||dƒq nˆjjtƒt@r,ˆjƒnˆj rEˆjtƒn|S( s4Add a recv event, returning the corresponding FutureR`R\iR csˆj|ˆjƒS(N(RRI(R(R(s*lib/python2.7/site-packages/zmq/_future.pyRPRtRCVTIMEOgü©ñÒMbP?N(R*t startswithRXR/tDONTWAITRzRGR RR RIR5RRR,R%trcvtimeoRvRRt _handle_recvt _add_io_state( RRR RRR`trR"t timeout_ms((Rs*lib/python2.7/site-packages/zmq/_future.pyR3;s*(      c s£|pˆjƒ}|d kr ˆj r |jddƒ}|jƒ}|tjB|d}|j|ƒnX|j |ƒdS(NRRaRbsUnhandled send event type: %rR\(RGRXRRRRJRR}RyR RŽR RaRbRR/R„R R(RRRR R RbR!R"((s*lib/python2.7/site-packages/zmq/_future.pyt _handle_send¾s6          cCsT|jjtƒ}|tj@r,|jƒn|tj@rF|jƒn|jƒdS(s(Dispatch IO events to _handle_recv, etc.N( RGRXRR/RR†RRRY(Rtfdteventst zmq_events((s*lib/python2.7/site-packages/zmq/_future.pyt_handle_eventsæs     cCs[|jdkrdS|dkr4|jjtƒ}n||j@rW|jd|jƒndS(skSchedule a call to handle_events next loop iteration If there are still events to handle. iN(RKRRGRXRRuR”(RR’((s*lib/python2.7/site-packages/zmq/_future.pyRYïs   cCs:|j|kr&|j|B}|_n|j|jƒdS(sAdd io_state to poller.N(RKt_update_handler(Rtstate((s*lib/python2.7/site-packages/zmq/_future.pyR‡ÿscCs5|j|@r!|j|@|_n|j|jƒdS(s&Stop poller from watching an io_state.N(RKR•(RR–((s*lib/python2.7/site-packages/zmq/_future.pyRŽs cCs|jƒdS(sOUpdate IOLoop handler with state. zmq FD is always read-only. N(RY(RR–((s*lib/python2.7/site-packages/zmq/_future.pyR• scCs6|jj|j|j|jƒ|jd|jƒdS(s#initialize the ioloop event handleriN(RHt add_handlerRGR”R6Ru(R((s*lib/python2.7/site-packages/zmq/_future.pyRNscCs5|j}|jjr!|j}n|jj|ƒdS(sNunregister the ioloop event handler called once during close N(RGRSRMRHtremove_handler(RR‘((s*lib/python2.7/site-packages/zmq/_future.pyRUs   N(+R>R?RRIRJRKRGR RpRHRMREt classmethodR2RVR/R0R@RXR}R{R[R`RaRbRkRRRvRut staticmethodRR3R4R†RR”RYR‡RŽR•RNRU(((s*lib/python2.7/site-packages/zmq/_future.pyRAssD      % #8 ( (     (RRR R (R@t collectionsRRt itertoolsRtzmqRRRR/RtPollerR R0RA(((s*lib/python2.7/site-packages/zmq/_future.pyts  \