ó ßüÚ\c@sadZddlmZddlZddlZyddlZWnek r]ddlZnXddlm Z ddl Z ddl m Z ddl mZddlmZddlmZdd l mZdd lmZdd lmZmZdd lmZd Zd„Zdd„Zdd„Zie_ dd„Z!ddd„Z"dddd„Z#dS(sºUtilities for integrating with IPython These functions should probably reside in Jupyter and IPython repositories, after which we can import them instead of having our own definitions. iÿÿÿÿ(tprint_functionN(tPopen(tThread(tuuid4(t TimeoutError(tIOLoop(tEvent(t get_ipython(tBlockingKernelClienttwrite_connection_file(tjupyter_runtime_diri c Cs¥|j|ƒ}t|dtƒ}|rN|jj}|jj}|jj}nxPtr y|jdt ƒ}Wn t j k rt dƒ‚nX|dj dƒ|kr¯qQn|dd} |d} | d krê| d d krPqqQ| d krtt| d ƒ} | j| dƒqQ| dkrQ|rK|j|| | d|ƒq| dkrztdj| dƒdtjƒqtjj| dj ddƒƒqQqQWdS(siRun a cell on a KernelClient Any output from the cell will be redisplayed in the local session. tkernelttimeouts"Timeout waiting for IPython outputt parent_headertmsg_idtheadertmsg_typetcontenttstatustexecution_statetidletstreamtnamettextt display_datatexecute_resultterrortparents t tracebacktfiletdatas text/plaintN(RRR(texecutetgetattrtFalset display_pubt pub_sockettsessionR tTruet get_iopub_msgtOUTPUT_TIMEOUTtqueuetEmptyRtgettsystwritetsendtprinttjointstderrtstdout( tiptkctcellRt in_kerneltsocketR%R tmsgRRR((s9lib/python2.7/site-packages/distributed/_ipython_utils.pytrun_cell_remote"s8        ##tworkercsštƒ‰t|ƒ}|jdƒ}t|‰|ˆj_ˆjƒd‡‡fd†}ˆ|_ˆj |ddd|ƒˆj |ddd|ƒdS(s–Register a %worker magic, given connection_info. Both a line and cell magic are registered, which run the given cell in a remote kernel. tkeycs)|dkr|}ntˆˆ|ƒdS(s/Run the current cell on a remote IPython kernelN(tNoneR9(tlineR5(R3R4(s9lib/python2.7/site-packages/distributed/_ipython_utils.pytremoteYs  t magic_kindR=t magic_nameR5N( RtdicttpopRR%R;tstart_channelsR<tclienttregister_magic_function(tconnection_infoR@tinfoR;R>((R3R4s9lib/python2.7/site-packages/distributed/_ipython_utils.pytregister_worker_magicLs      c Cs/tƒ}|jddƒ}|d}||jkrCt|ƒ‚nt|j|ƒ}|sŠt|ƒdkr}tdƒ‚n|d}ndjt t t |j ƒƒƒƒ}|j dƒ}|tjkrÜtj|}n?t|}||j_|jƒ|jddƒ|tj|cCsOtƒ}|dkrdS|jtddd|ƒ|jtddd|ƒdS(sTDefine the parameterized %remote magic See remote_magic above for details. NR?R=R@R5(RR<RERS(R@R3((s9lib/python2.7/site-packages/distributed/_ipython_utils.pytregister_remote_magic”s   cs˜tƒ}|dkr$tƒj}ntjj||dƒ‰tˆ|dddˆg}|rr|j|ƒnt |ƒt j ‡fd†ƒ}dS(s¿Open a QtConsole connected to a worker who has the given future - identify worker with who_has - start IPython kernel on the worker - start qtconsole connected to the kernel s.jsontjupytert qtconsoles --existingcs)ytjˆƒWntk r$nXdS(s)Cleanup our connection file when we exit.N(tostremovetOSError((tpath(s9lib/python2.7/site-packages/distributed/_ipython_utils.pyt_cleanup_connection_file²s N( R R<RthexR\R_R0R textendRtatexittregister(RFRt extra_argst runtime_dirtcmdR`((R_s9lib/python2.7/site-packages/distributed/_ipython_utils.pytconnect_qtconsole s    c sAddlm}|ƒdk r.tdƒ‚nddlm}ddlm}tj ƒ}tj ƒ|ƒ‰ˆj ƒ|j d|ƒ‰dˆj j _|rª|ˆ_nd„‰ˆˆ_ˆˆ_tƒ‰‡‡‡‡‡fd †}td |ƒ}t|_|jƒˆjd d ƒs)td ƒ‚tj ƒ|j ƒˆS(sUStart an IPython kernel in a thread Parameters ---------- ip: str The IP address to listen on (likely the parent object's ip). ns: dict Any names that should be injected into the IPython namespace. log: logger instance Hook up IPython's logging to an existing logger instead of the default. iÿÿÿÿ(Rs+Cannot start IPython, it's already running.(t ZMQIOLoop(t IPKernelApptlogs:memory:cSsdS(N(R<(((s9lib/python2.7/site-packages/distributed/_ipython_utils.pytnoopàscs{ˆjgƒˆˆj_ˆˆj_ˆjjƒtjƒˆj_ˆrcˆjjj j ˆƒnˆj ƒˆjƒdS(N( t initializeR tpre_handler_hooktpost_handler_hooktstartRtinstancetlooptshellRKtupdatetset((tapptevtRltnstzmq_loop(s9lib/python2.7/site-packages/distributed/_ipython_utils.pyt_startës     ttargetR is4IPython didn't start in a reasonable amount of time.N(tIPythonRR<t RuntimeErrortzmq.eventloop.ioloopRitipykernel.kernelappRjRRqtclear_instancetinstalltconfigtHistoryManagert hist_fileR3t init_signaltlog_connection_infoRRR&tdaemonRptwaittAssertionError( R3RxRkRRiRjt save_instRztzmq_loop_thread((RvRwRlRxRys9lib/python2.7/site-packages/distributed/_ipython_utils.pyt start_ipython»s2               ($t__doc__t __future__RRcR\R)t ImportErrortQueuet subprocessRR,t threadingRtuuidRt tornado.genRttornado.ioloopRRR|Rtjupyter_clientRR tjupyter_core.pathsR R(R9RHR<RSRTRYRhRŒ(((s9lib/python2.7/site-packages/distributed/_ipython_utils.pyts2     *  ,