]c@@sUdZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZmZddlmZmZmZmZmZmZmZmZdd lmZmZdd lmZdd lm Z d efd YZ!e j"e!dddZ#edZ$dS(s%Base class to manage a running kerneli(tabsolute_import(tcontextmanagerN(t import_itemi(t is_local_ipt local_ips(tAnytFloattInstancetUnicodetListtBooltTypetDottedObjectName(t launch_kernelt kernelspec(tConnectionFileMixin(tKernelManagerABCt KernelManagercB@s eZdZeejZdZedZ e ddZ dZ dZ eZeejZdZdZed d ed d ZeejZd Zd*ZedZe ed ed dZ!dZ"edZ#eZ$eZ%eZ&e'ed ed dZ(dZ)dZ*dZ+ddZ,ddZ-dZ.d*dZ/dZ0dZ1dZ2dZ3e4dZ5d*d d!Z6ed"Z7e4e4d#Z8e4e4d$Z9ed%Z:d&Z;d'Z<d(Z=d)Z>RS(+sgManages a single kernel in a subprocess on this host. This version starts kernels with Popen. cC@s tjS(N(tzmqtContext(tself((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_context_default)ss,jupyter_client.blocking.BlockingKernelClienttklasssjupyter_client.KernelClientcC@s t|jS(N(Rt client_class(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_client_factory_default/scC@stt||_dS(N(Rtstrtclient_factory(Rtnametoldtnew((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_client_class_changed2scC@stjd|jS(Ntdata_dir(RtKernelSpecManagerR(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_kernel_spec_manager_default;scC@s d|_dS(N(tNonet _kernel_spec(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_kernel_spec_manager_changed>sg@tconfigthelpsETime to wait for a kernel to terminate before killing it, in seconds.cC@s(d|_|dkr$tj|_ndS(Ntpython(R"R#RtNATIVE_KERNEL_NAMEt kernel_name(RRRR((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_kernel_name_changedHs  cC@s@|jdkr9|jdk r9|jj|j|_n|jS(Nt(R#R"R)tkernel_spec_managertget_kernel_spec(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyt kernel_specOssDEPRECATED: Use kernel_name instead. The Popen Command to launch the kernel. Override this if you have a custom kernel. If kernel_cmd is specified in a configuration file, Jupyter does not pass any arguments to the kernel, because it cannot make any assumptions about the arguments that the kernel understands. In particular, this means that the kernel does not receive the option --debug if it given on the Jupyter command line. cC@stjddS(NsMSetting kernel_cmd is deprecated, use kernel_spec to start different kernels.(twarningstwarn(RRRR((s5lib/python2.7/site-packages/jupyter_client/manager.pyt_kernel_cmd_changedcscC@s|jdddhkS(NR'tpython2tpython3(R)(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyt ipykernelgss,Should we autorestart the kernel if it dies.cC@s|j|jdS(N(t_close_control_sockettcleanup_connection_file(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyt__del__us cC@sdS(N((R((s5lib/python2.7/site-packages/jupyter_client/manager.pytstart_restarter}scC@sdS(N((R((s5lib/python2.7/site-packages/jupyter_client/manager.pytstop_restarterstrestartcC@s*|jdkrdS|jj||dS(s;register a callback to be called when a kernel is restartedN(t _restarterR"t add_callback(Rtcallbacktevent((s5lib/python2.7/site-packages/jupyter_client/manager.pytadd_restart_callbackscC@s*|jdkrdS|jj||dS(s=unregister a callback to be called when a kernel is restartedN(R;R"tremove_callback(RR=R>((s5lib/python2.7/site-packages/jupyter_client/manager.pytremove_restart_callbackscK@sXi}|j|jdt|jtd|jd||j||j|S(s3Create a client configured to connect to our kerneltsessiontconnection_filetparent(tupdatetget_connection_infotTruetdictRCR(Rtkwargstkw((s5lib/python2.7/site-packages/jupyter_client/manager.pytclients    c@s|p g}|jr%|j|}n|jj|}|rz|dddtjddtjd hkrztj|dt|j r>td|jtfn|j|j|_|jdg}|j d|}|jdt j j}|jdd|j s|j|jjpin|jjd||j|d|||_|j|jdS(szStarts a kernel on this host in a separate process. If random ports (port=0) are being used, this method must be called before the channels are created. Parameters ---------- `**kw` : optional keyword arguments that are passed down to build the kernel_cmd and launching the kernel (e.g. Popen kwargs). ttcpsCan only launch a kernel on a local interface. This one is not: %s.Make sure that the '*_address' attributes are configured properly. Currently valid addresses are: %sR\tenvtPYTHONEXECUTABLEsStarting kernel: %sN(t transportRtipt RuntimeErrorRtwrite_connection_filetcopyRXtpopR`tostenvironR"RSRER.RitlogtdebugRatkernelR8Rf(RRJR\RSRi((s5lib/python2.7/site-packages/jupyter_client/manager.pyt start_kernels     cC@sKtd|}|jjdd|}|j|jj|j|dS(s4Send a shutdown request via control channel R:tshutdown_requesttcontentN(RHRBtmsgRftsendRc(RR:RxRy((s5lib/python2.7/site-packages/jupyter_client/manager.pytrequest_shutdowns g?cC@s|dkr!t|jd}nxatt||D]#}|jrZtj|q8Pq8W|jr|j j d|j ndS(sWait for kernel shutdown, then kill process if it doesn't shutdown. This does not send shutdown requests - use :meth:`request_shutdown` first. is,Kernel is taking too long to finish, killingN( R"tmaxtshutdown_wait_timetrangetinttis_alivettimetsleept has_kernelRsRtt _kill_kernel(Rtwaittimet pollintervalti((s5lib/python2.7/site-packages/jupyter_client/manager.pytfinish_shutdowns   cC@s+|r|jn|j|jdS(s/Clean up resources when the kernel is shut downN(R6tcleanup_ipc_filesR5(RRC((s5lib/python2.7/site-packages/jupyter_client/manager.pytcleanup#s  cC@sL|j|r|jn|jd||j|jd| dS(sXAttempts to stop the kernel process cleanly. This attempts to shutdown the kernels cleanly by: 1. Sending it a shutdown message over the shell channel. 2. If that fails, the kernel is shutdown forcibly by sending it a signal. Parameters ---------- now : bool Should the kernel be forcible killed *now*. This skips the first, nice shutdown attempt. restart: bool Will this kernel be restarted after it is shutdown. When this is True, connection files will not be cleaned up. R:RCN(R9RR{RR(RtnowR:((s5lib/python2.7/site-packages/jupyter_client/manager.pytshutdown_kernel+s    cK@sk|jdkrtdnI|jd|dt|rG|jn|jj||j|jdS(syRestarts a kernel with the arguments that were used to launch it. Parameters ---------- now : bool, optional If True, the kernel is forcefully restarted *immediately*, without having a chance to do any cleanup action. Otherwise the kernel is given 1s to clean up before a forceful restart is issued. In all cases the kernel is restarted, the only difference is whether it is given a chance to perform a clean shutdown or not. newports : bool, optional If the old kernel was launched with random ports, this flag decides whether the same ports and connection file will be used again. If False, the same ports and connection file are used. This is the default. If True, new random port numbers are chosen and a new connection file is written. It is still possible that the newly chosen random port numbers happen to be the same as the old ones. `**kw` : optional Any options specified here will overwrite those used to launch the kernel. s>Cannot restart the kernel. No previous call to 'start_kernel'.RR:N(RXR"RmRRGtcleanup_random_portsRERv(RRtnewportsRJ((s5lib/python2.7/site-packages/jupyter_client/manager.pytrestart_kernelKs cC@s |jdk S(s/Has a kernel been started that we are managing.N(RuR"(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyRrscC@s|jry3ttdr.|jtjn |jjWn_tk r}tj dkru|j dkrqqddl m }|j |krqnX|jj d|_n tddS(srKill the running kernel. This is a private method, callers should use shutdown_kernel(now=True). tSIGKILLtwin32ii(tESRCHs)Cannot kill kernel. No kernel is running!N(Rthasattrtsignalt signal_kernelRRutkilltOSErrorRUtplatformtwinerrorterrnoRtwaitR"Rm(RteR((s5lib/python2.7/site-packages/jupyter_client/manager.pyRws     cC@s|jr|jj}|dkrftjdkrSddlm}||jjq|j t j q|dkr|j j ddi}|j|j j|j|qn tdd S( sInterrupts the kernel by sending it a signal. Unlike ``signal_kernel``, this operation is well supported on all platforms. RRi(tsend_interrupttmessagetinterrupt_requestRxs.Cannot interrupt kernel. No kernel is running!N(RR.tinterrupt_modeRURt win_interruptRRutwin32_interrupt_eventRRtSIGINTRBRyRfRzRcRm(RRRRy((s5lib/python2.7/site-packages/jupyter_client/manager.pytinterrupt_kernels     cC@s|jr~ttdrkttdrky-tj|jj}tj||dSWqktk rgqkXn|jj|n t ddS(sSends a signal to the process group of the kernel (this usually includes the kernel and any subprocesses spawned by the kernel). Note that since only SIGTERM is supported on Windows, this function is only useful on Unix systems. tgetpgidtkillpgNs+Cannot signal kernel. No kernel is running!( RRRqRRutpidRRt send_signalRm(Rtsignumtpgid((s5lib/python2.7/site-packages/jupyter_client/manager.pyRs  cC@s1|jr)|jjdkr"tStSntSdS(s$Is the kernel process still running?N(RRutpollR"RGtFalse(R((s5lib/python2.7/site-packages/jupyter_client/manager.pyRs  N(?t__name__t __module__t__doc__RRRtcontextRR RR RRRRRuRR R,R!R$RRGR}RR(R)R*R"R#tpropertyR.R RSR1R4RXRcR;R t autorestartR7R8R9R?RARKR`RaRfR5RvRR{RRRRRRRRR(((s5lib/python2.7/site-packages/jupyter_client/manager.pyR!s`                      #   ,   ' !  i<R'cK@s{td|}|j||j}|jy|jd|Wn(tk rp|j|jnX||fS(s5Start a new kernel, and return its Manager and ClientR)ttimeout(RRvRKtstart_channelstwait_for_readyRmt stop_channelsR(tstartup_timeoutR)RItkmtkc((s5lib/python2.7/site-packages/jupyter_client/manager.pytstart_new_kernels      ck@s=t|\}}z |VWd|j|jdtXdS(sContext manager to create a kernel in a subprocess. The kernel is shut down when the context exits. Returns ------- kernel_client: connected KernelClient instance NR(RRRRG(RIRR((s5lib/python2.7/site-packages/jupyter_client/manager.pyt run_kernels   (%Rt __future__Rt contextlibRRqRYRRURR/Rtipython_genutils.importstringRtlocalinterfacesRRt traitletsRRRRR R R R tjupyter_clientR RtconnectRt managerabcRRtregisterRR(((s5lib/python2.7/site-packages/jupyter_client/manager.pyts(       :