ó Û¤[c@sndZddlmZddlZddlmZddlmZyddlm Z Wn!e k ryddl m Z nXddl Z ddlZddlZddlZddlmZmZddlZddlmZdd lmZdd lmZdd lmZdd lmZd ZdZ de!fd„ƒYZ"de!fd„ƒYZ#defd„ƒYZ$dS(s.Wrappers for forwarding stdout/stderr over zmqiÿÿÿÿ(tprint_functionN(tb2a_hex(tdeque(t lock_held(tStringIOt TextIOBase(tIOLoop(t ZMQStream(textract_header(t py3compat(t unicode_typeiit IOPubThreadcBs¶eZdZed„Zd„Zd„Zed„ƒZd„Z d„Z d„Z d„Z d „Z d „Zd „Zd „Zd „Zed„ƒZd„Zd„Zd„ZRS(sAn object for sending IOPub messages in a background thread Prevents a blocking main thread from delaying output from threads. IOPubThread(pub_socket).background_socket is a Socket-API-providing object whose IO is always run in a thread. cCs¢||_t|ƒ|_tjƒ|_||_tdtƒ|_ |rU|j ƒnt j ƒ|_ tƒ|_|jƒt jd|jƒ|_t|j_dS(sCreate IOPub thread Parameters ---------- socket: zmq.PUB Socket the socket on which messages will be sent. pipe: bool Whether this process should listen for IOPub messages piped from subprocesses. t make_currentttargetN(tsockettBackgroundSockettbackground_sockettostgetpidt _master_pidt _pipe_flagRtFalsetio_loopt_setup_pipe_int threadingtlocalt_localRt_eventst_setup_event_pipetThreadt _thread_maintthreadtTruetdaemon(tselfRtpipe((s1lib/python2.7/site-packages/ipykernel/iostream.pyt__init__2s     cCs1|jjƒ|jjƒ|jjdtƒdS(s.The inner loop that's actually run in a threadtall_fdsN(RR tstarttcloseR (R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyRKs  cCs|jj}|jtjƒ}d|_ttjdƒƒjdƒ}d|}|_ |j |ƒt ||j ƒ|_ |j j|jƒdS(sLCreate the PULL socket listening for events that should fire in this thread.iitasciis inproc://%sN(RtcontexttzmqtPULLtlingerRRturandomtdecodet_event_interfacetbindRRt _event_pullerton_recvt _handle_event(R"tctxtpipe_int_uuidtiface((s1lib/python2.7/site-packages/ipykernel/iostream.pyRQs   cCsky|jj}WnTtk rf|jj}|jtjƒ}d|_|j|j ƒ||j_nX|S(sSthread-local event pipe for signaling events that should be processed in the threadi( Rt event_pipetAttributeErrorRR)R*tPUSHR,tconnectR/(R"R8R4((s1lib/python2.7/site-packages/ipykernel/iostream.pyt _event_pipe]s   cCs@t|jƒ}x*t|ƒD]}|jjƒ}|ƒqWdS(sÅHandle an event on the event pipe Content of the message is ignored. Whenever *an* event arrives on the event stream, *all* waiting events are processed in order. N(tlenRtrangetpopleft(R"tmsgtn_eventstitevent_f((s1lib/python2.7/site-packages/ipykernel/iostream.pyR3ks cCs½|jj}tjdƒ|_|jtjƒ}d|_y|jdƒ|_ Wn?tj k r}t j d|dƒt |_|jƒdSXt||jƒ|_|jj|jƒdS(s7setup listening pipe for IOPub from forked subprocessesiistcp://127.0.0.1s)Couldn't bind IOPub Pipe to 127.0.0.1: %ss' subprocess output will be unavailable.N(RR)RR-t _pipe_uuidR*R+R,tbind_to_random_portt _pipe_porttZMQErrortwarningstwarnRRR'RRt_pipe_inR2t_handle_pipe_msg(R"R4R5te((s1lib/python2.7/site-packages/ipykernel/iostream.pyRzs     cCs]|j s|jƒ rdS|d|jkrHtd|dtjƒdS|j|dƒdS(s'handle a pipe message from a subprocessNisBad pipe message: %stfilei(Rt_is_master_processRDtprinttsyst __stderr__tsend_multipart(R"R@((s1lib/python2.7/site-packages/ipykernel/iostream.pyRKs cCsEtjƒ}|jtjƒ}d|_|jd|jƒ||fS(Ni¸ stcp://127.0.0.1:%i(R*tContextRR:R,R;RF(R"R4tpipe_out((s1lib/python2.7/site-packages/ipykernel/iostream.pyt_setup_pipe_out™s   cCstjƒ|jkS(N(RRR(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyRN¡scCs"|j s|jƒrtStSdS(s8check for forks, and switch to zmq pipeline if necessaryN(RRNtMASTERtCHILD(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyt_check_mp_mode¤scCs!|jjƒtj|jƒdS(sStart the IOPub threadN(RR&tatexittregistertstop(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyR&«s cCs_|jjƒsdS|jj|jjƒ|jjƒt|jdƒr[|jjj ƒndS(sStop the IOPub threadNR8( Rtis_aliveRt add_callbackR[tjointhasattrRR8R'(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyR[²s  cCs|jjƒd|_dS(N(RR'tNone(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyR'»s cCs |jdkS(N(RR`(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pytclosed¿scCs=|jjƒr2|jj|ƒ|jjdƒn|ƒdS(stSchedule a function to be called in our IO thread. If the thread is not running, call immediately. tN(RR\RtappendR<tsend(R"tf((s1lib/python2.7/site-packages/ipykernel/iostream.pytscheduleÃscs ˆj‡‡‡fd†ƒdS(s”send_multipart schedules actual zmq send in my thread. If my thread isn't running (e.g. forked process), send immediately. csˆjˆˆŽS(N(t _really_send((targstkwargsR"(s1lib/python2.7/site-packages/ipykernel/iostream.pytÔsN(Rf(R"RhRi((RhRiR"s1lib/python2.7/site-packages/ipykernel/iostream.pyRRÏscOsx|jƒ}|tkr1|jj|||ŽnC|jƒ\}}|j|jg|||Ž|jƒ|jƒdS(s)The callback that actually sends messagesN(RXRWRRRRURDR'tterm(R"R@RhRitmp_modeR4RT((s1lib/python2.7/site-packages/ipykernel/iostream.pyRgÖs   (t__name__t __module__t__doc__RR$RRtpropertyR<R3RRKRURNRXR&R[R'RaRfRRRg(((s1lib/python2.7/site-packages/ipykernel/iostream.pyR )s$           RcBsAeZdZdZd„Zd„Zd„Zd„Zd„Z RS(s>Wrapper around IOPub thread that provides zmq send[_multipart]cCs ||_dS(N(t io_thread(R"Rq((s1lib/python2.7/site-packages/ipykernel/iostream.pyR$ëscCs“|jdƒr7|jdƒr7tt|ƒj|ƒnt|jj|ƒrytj d|t ddƒt |jj|ƒStt|ƒj|ƒdS(s2Wrap socket attr access for backward-compatibilityt__s5Accessing zmq Socket attribute %s on BackgroundSockett stackleveliN( t startswithtendswithtsuperRt __getattr__R_RqRRHRItDeprecationWarningtgetattr(R"tattr((s1lib/python2.7/site-packages/ipykernel/iostream.pyRwîs  cCsz|dks*|jdo$|jdƒƒrFtt|ƒj||ƒn0tjd|tddƒt|j j ||ƒdS(NRqRrs3Setting zmq Socket attribute %s on BackgroundSocketRsi( RtRuRvRt __setattr__RHRIRxtsetattrRqR(R"Rztvalue((s1lib/python2.7/site-packages/ipykernel/iostream.pyR{ùs *  cOs|j|g||ŽS(N(RR(R"R@RhRi((s1lib/python2.7/site-packages/ipykernel/iostream.pyRdscOs|jj||ŽS(sSchedule send in IO thread(RqRR(R"RhRi((s1lib/python2.7/site-packages/ipykernel/iostream.pyRRsN( RmRnRoR`RqR$RwR{RdRR(((s1lib/python2.7/site-packages/ipykernel/iostream.pyRçs   t OutStreamcBs§eZdZdZdZdZdZddd„Zd„Z d„Z d„Z e d„ƒZ d „Zd „Zd „Zd „Zd „Zd„Zd„Zd„ZRS(stA file like object that publishes the stream to a 0MQ PUB socket. Output is handed off to an IO Thread i gš™™™™™É?sUTF-8cCs|dk rtjdtƒn||_t|tƒsjtjd|tddƒt|ƒ}|jƒn||_||_ dt j |ƒ|_ i|_ tjƒ|_t|_|j|_|jƒd|_|rt|dƒrt|dƒr||_qtdƒ‚ndS( Ns4pipe argument to OutStream is deprecated and ignoreds4OutStream should be created with IOPubThread, not %rRsisstream.treadtwrites(echo argument must be a file like object(R`RHRIRxtsessiont isinstanceR R&t pub_threadtnameR t cast_bytesttopict parent_headerRRRRt_flush_pendingRt_io_loopt _new_buffertechoR_t ValueError(R"RRƒR„R#R‹((s1lib/python2.7/site-packages/ipykernel/iostream.pyR$s,                cCstjƒ|jkS(N(RRR(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyRN3scCst|ƒ|_dS(N(RR‡(R"tparent((s1lib/python2.7/site-packages/ipykernel/iostream.pyt set_parent6scCs d|_dS(N(R`Rƒ(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyR'9scCs |jdkS(N(RƒR`(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyRa<scs9ˆjr dStˆ_‡fd†}ˆjj|ƒdS(suschedule a flush in the IO thread call this on write, to indicate that flush should be called soon. NcsˆjjˆjˆjƒdS(N(R‰t call_latertflush_intervalt_flush((R"(s1lib/python2.7/site-packages/ipykernel/iostream.pyt_schedule_in_threadJs(RˆR RƒRf(R"R’((R"s1lib/python2.7/site-packages/ipykernel/iostream.pyt_schedule_flush@s   cCs‰|jjjƒr{|jj|jƒtƒs…tjƒ}|jj|jƒ|j |j ƒsxt ddt j ƒqxq…n |jƒdS(sStrigger actual zmq send send will happen in the background thread sIOStream.flush timed outRMN(RƒRR\RfR‘timport_lock_heldRtEventtsettwaitt flush_timeoutRORPRQ(R"tevt((s1lib/python2.7/site-packages/ipykernel/iostream.pytflushNs  c Csãt|_|jd k rsy|jjƒWqstk ro}|jtjk rptdj |ƒdtjƒqpqsXn|j ƒ}|rßt j ƒ|j _i|jd6|d6}|j j|jdd|d|jd|jƒnd S( s³This is where the actual send happens. _flush should generally be called in the IO thread, unless the thread has been destroyed (e.g. forked subprocess). sFlush failed: {}RMunameutextustreamtcontentRtidentN(RRˆR‹R`RštOSErrorRPRQROtformatt _flush_bufferRRRtpidR„RdRƒR‡R†(R"RLtdataR›((s1lib/python2.7/site-packages/ipykernel/iostream.pyR‘ds  cs ˆjdk rmyˆjjˆƒWqmtk ri}ˆjtjk rjtdj|ƒdtjƒqjqmXnˆjdkr‹t dƒ‚n|t ˆt ƒs²ˆj ˆj dƒ‰nˆjƒ }ˆjj‡‡fd†ƒ|rýdˆkrˆjƒqn ˆjƒdS(NsWrite failed: {}RMsI/O operation on closed filetreplacecsˆjjˆƒS(N(t_bufferR€((R"tstring(s1lib/python2.7/site-packages/ipykernel/iostream.pyRjss (R‹R`R€RRPRQRORžRƒRŒR‚R R.tencodingRNRfRšR“(R"R¤RLtis_child((R"R¤s1lib/python2.7/site-packages/ipykernel/iostream.pyR€~s"  cCs@|jdkrtdƒ‚nx|D]}|j|ƒq%WdS(NsI/O operation on closed file(RƒR`RŒR€(R"tsequenceR¤((s1lib/python2.7/site-packages/ipykernel/iostream.pyt writelinesšs cCstS(N(R (R"((s1lib/python2.7/site-packages/ipykernel/iostream.pytwritable¡scCsEd}|jdk rA|j}|jƒ|jƒ}|jƒn|S(sƒclear the current buffer and return the current buffer data. This should only be called in the IO thread. uN(R£R`RŠtgetvalueR'(R"R¡tbuf((s1lib/python2.7/site-packages/ipykernel/iostream.pyRŸ¤s    cCstƒ|_dS(N(RR£(R"((s1lib/python2.7/site-packages/ipykernel/iostream.pyRŠ±sN(RmRnRoR˜RR`R†R¥R$RNRŽR'RpRaR“RšR‘R€R¨R©RŸRŠ(((s1lib/python2.7/site-packages/ipykernel/iostream.pyR~ s$          (%Rot __future__RRYtbinasciiRt collectionsRt importlibRR”t ImportErrortimpRRPRRHtioRRR*tzmq.eventloop.ioloopRtzmq.eventloop.zmqstreamRtjupyter_client.sessionRtipython_genutilsR tipython_genutils.py3compatR RVRWtobjectR RR~(((s1lib/python2.7/site-packages/ipykernel/iostream.pyts0       ¾"