B 2d;J@s0dZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZd dlmZd dlmZd dlmZd dlmZd dlmZGdddeZejejdddZ GdddeZ!Gddde!Z"dS)z%A kernel manager for multiple kernelsN)Any)Bool)default)Dict)DottedObjectName)Instance)observe)Unicode)LoggingConfigurable) import_item)KernelSpecManager)NATIVE_KERNEL_NAME) KernelManager) ensure_async)run_syncc@s eZdZdS)DuplicateKernelErrorN)__name__ __module__ __qualname__rrn/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/jupyter_client/multikernelmanager.pyrsr)freturncs2tjttjtjtjtjtjfdfdd }|S)zDdecorator for proxying MKM.method(kernel_id) to individual KMs by ID)self kernel_idargskwargsrcs6||}t|j}|||}||f|||S)N) get_kernelgetattrr)rrrrkmmethodr)rrrwrapped$s    zkernel_method..wrapped)trstrUnionCallable Awaitable)rr#r)rr kernel_method!s r)cseZdZdZeeddjddZee ddZ e dddjddZ e d d d Zed dZed ddZejdddZedddjddZedZedZeZeddZedejdddZ edZ!eZ"fddZ#ej$e%dddZ&e'dd d!Z(e%e)d"d#d$Z*ej+e%ejej,e-e%e%fd%d&d'Z.e%e-ej/d(d)d*d+Z0e%ej/d(d,d-d.Z1d/d0Z2dhej+e%eje%d%d1d2Z3e4e3Z5die%ej+e)ej+e)d(d3d4d5Z6e4e6Z7e8dje%ej+e)d(d6d7d8Z9e8dke%ej+e:ej+e:d(d:d;d<Z;e8dle%e)d(d6d=d>Ze4e>Z?e%d(d"dDdEZ@e8e%e'd(dFdGdHZAdne%e)d(dIdJdKZBe4eBZCe8e%e)d"dLdMZDe%d(d"dNdOZEe%e-d"dPdQZFe8doe%eje%d(dSdTdUZGe8dpe%eje%d(dSdVdWZHe8e%eje%ejfd"dXdYZIe8dqe%ej+eJeKjKdZd[d\ZLe8dre%ej+eJeKjKdZd]d^ZMe8dse%ej+eJeKjKdZd_d`ZNe8dte%ej+eJeKjKdZdadbZOe8due%ej+eJeKjKdZdcddZPeje%dedfdgZQZRS)vMultiKernelManagerz&A class for managing multiple kernels.z'The name of the default kernel to start)helpT)config) allow_nonez)jupyter_client.ioloop.IOLoopKernelManagerzThe kernel manager class. This is configurable to allow subclassing of the KernelManager for customized behavior. kernel_manager_classcCs||_dS)N)_create_kernel_manager_factorykernel_manager_factory)rZchangerrr_kernel_manager_class_changedEsz0MultiKernelManager._kernel_manager_class_changedz)this is kernel_manager_class after importr0cCs|S)N)r/)rrrr_kernel_manager_factory_defaultKsz2MultiKernelManager._kernel_manager_factory_default)rcs*tjtjtjtdfdd }|S)N)rrrcs4jr&jjr_|dj||}|S)Ncontext)shared_contextr3closed_context_default setdefault)rrr )kernel_manager_ctorrrrcreate_kernel_managerRs   zPMultiKernelManager._create_kernel_manager_factory..create_kernel_manager)r r.r$rr)rr9r)r8rrr/Os  z1MultiKernelManager._create_kernel_manager_factoryz4Share a single zmq.Context to talk to all my kernelsz zmq.ContextFcCs|jS)z#A shim for backwards compatibility.)_pending_kernels)rrrr_starting_kernelshsz$MultiKernelManager._starting_kernelsr3cCsd|_tS)NT)_created_contextzmqContext)rrrrr6msz#MultiKernelManager._context_defaultcs^|jr2|jr2|jjs2|jr(|jd||jy tj}Wntk rRYnX|dS)z:Handle garbage collection. Destroy context if applicable.zDestroying zmq context for %sN) r<r3r5logdebugdestroysuper__del__AttributeError)rZ super_del) __class__rrrDvs  zMultiKernelManager.__del__cCst|jS)z6Return a list of the kernel ids of the active kernels.)list_kernelskeys)rrrrlist_kernel_idssz"MultiKernelManager.list_kernel_idscCs t|S)z%Return the number of running kernels.)lenrJ)rrrr__len__szMultiKernelManager.__len__)rrcCs ||jkS)N)rH)rrrrr __contains__szMultiKernelManager.__contains__) kernel_namerrcCs|d|jf|}||kr(td||dkr6|j}i}|jrJ|j|d<|jftj|j d|||j |d|}|||fS)NrzKernel already exists: %skernel_spec_managerzkernel-%s.json)Zconnection_fileparentr@rN) pop new_kernel_idrdefault_kernel_namerOr0ospathjoinconnection_dirr@)rrNrrZconstructor_kwargsr rrrpre_start_kernels  z#MultiKernelManager.pre_start_kernelN)rr kernel_awaitablerc sZy&|IdH||j|<|j|dWn.tk rT}z|j|Wdd}~XYnXdS)N)rHr:rQ Exceptionr@ exception)rrr rYerrr_add_kernel_when_readys   z)MultiKernelManager._add_kernel_when_ready)rrYrc sZy&|IdH|||j|dWn.tk rT}z|j|Wdd}~XYnXdS)N) remove_kernelr:rQrZr@r[)rrrYr\rrr_remove_kernel_when_readys   z,MultiKernelManager._remove_kernel_when_readycCs t|ddS)zReturns a boolean; a clearer method for determining if this multikernelmanager is using pending kernels or not use_pending_kernelsF)r)rrrr_using_pending_kernelssz)MultiKernelManager._using_pending_kernelscs|||\}}}t|ts4|jdj|jjd||d<t|j f|}t | |||}||j |<|r~||j|<n|IdH|jr|j|S)zStart a new kernel. The caller can pick a kernel_id by passing one in as a keyword arg, otherwise one will be generated using new_kernel_id(). The kernel ID for the newly started kernel is returned. zHKernel manager class ({km_class}) is not an instance of 'KernelManager'!)Zkm_classrN)rX isinstancerr@warningformatr.rFr start_kernelasyncio create_taskr]r:rarHreadyr[)rrNrr rZstartertaskrrr_async_start_kernels       z&MultiKernelManager._async_start_kernel)rnowrestartrcs |jd|||jkr|j|}y.|IdH||}ttj|jIdHWn2tj k rhYnt k r| |dSX||}|j s|j r| |dSt|||}t|||}||j|<|s|IdH|j r|j dS)a3Shutdown a kernel by its kernel uuid. Parameters ========== kernel_id : uuid The id of the kernel to shutdown. now : bool Should the kernel be shutdown forcibly using a signal. restart : bool Will the kernel be restarted? zKernel shutdown: %sN)r@infor:rr$castrfFuturerhCancelledErrorrZr^ cancelledr[rshutdown_kernel ensure_futurer_ra)rrrkrlrir Zstopperfutrrr_async_shutdown_kernels.           z)MultiKernelManager._async_shutdown_kernel)rrlrcCsdS)z,Ask a kernel to shut down by its kernel uuidNr)rrrlrrrrequest_shutdownsz#MultiKernelManager.request_shutdown皙?)rwaittime pollintervalrcCs|jd|dS)zDWait for a kernel to finish shutting down, and kill it if it doesn'tzKernel shutdown: %sN)r@rm)rrrxryrrrfinish_shutdownsz"MultiKernelManager.finish_shutdowncCsdS)zClean up a kernel's resourcesNr)rrrlrrrcleanup_resources%sz$MultiKernelManager.cleanup_resourcescCs|j|dS)zremove a kernel from our mapping. Mainly so that a kernel can be removed if it is already dead, without having to call shutdown_kernel. The kernel object is returned, or `None` if not found. N)rHrQ)rrrrrr^)sz MultiKernelManager.remove_kernel)rkrc s}|tj7}tj}fddt|D}tj|IdHrxV|D]N}y|j IdHWqZtj k rj|j  YqZt k rYqZXqZWdS)zShutdown all kernels.csg|]}tj|dqS))rk)rrr).0kid)rkrrr 8sz:MultiKernelManager._async_shutdown_all..N)rJrGr:rHvaluessetrfgatherrarhrprcancelrZ)rrkZkidsZkmsZfutsr r)rkrr_async_shutdown_all3s z&MultiKernelManager._async_shutdown_allcCs8||}|jstd|}|jd||S)zInterrupt (SIGINT) the kernel by its uuid. Parameters ========== kernel_id : uuid The id of the kernel to interrupt. z/Kernel is in a pending state. Cannot interrupt.zKernel interrupted: %s)rrhdone RuntimeErrorinterrupt_kernelr@rm)rrkerneloutrrrrGs   z#MultiKernelManager.interrupt_kernel)rsignumrcCs|jd||fdS)aRSends a signal to the kernel by its uuid. Note that since only SIGTERM is supported on Windows, this function is only useful on Unix systems. Parameters ========== kernel_id : uuid The id of the kernel to signal. signum : int Signal number to send kernel. zSignaled Kernel %s with %sN)r@rm)rrrrrr signal_kernelVsz MultiKernelManager.signal_kernel)rrkrcsN||}|r$|js$tdt|j|dIdH}|jd||S)aERestart a kernel by its uuid, keeping the same ports. Parameters ========== kernel_id : uuid The id of the kernel to interrupt. 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. z-Kernel is in a pending state. Cannot restart.)rkNzKernel restarted: %s) rrarhrrrrestart_kernelr@rm)rrrkrrrrr_async_restart_kernelfs  z(MultiKernelManager._async_restart_kernelcCsdS)zIs the kernel alive. This calls KernelManager.is_alive() which calls Popen.poll on the actual kernel subprocess. Parameters ========== kernel_id : uuid The id of the kernel. Nr)rrrrris_alives zMultiKernelManager.is_alivecCs||krtd|dS)zcheck that a kernel id is validzKernel with id not found: %sN)KeyError)rrrrr_check_kernel_idsz#MultiKernelManager._check_kernel_idcCs|||j|S)zGet the single KernelManager object for a kernel by its uuid. Parameters ========== kernel_id : uuid The id of the kernel. )rrH)rrrrrrs zMultiKernelManager.get_kernelrl)rcallbackeventrcCsdS)z&add a callback for the KernelRestarterNr)rrrrrrradd_restart_callbacksz'MultiKernelManager.add_restart_callbackcCsdS)z)remove a callback for the KernelRestarterNr)rrrrrrrremove_restart_callbacksz*MultiKernelManager.remove_restart_callbackcCsdS)aReturn a dictionary of connection data for a kernel. Parameters ========== kernel_id : uuid The id of the kernel. Returns ======= connection_dict : dict A dict of the information needed to connect to a kernel. This includes the ip address and the integer port numbers of the different channels (stdin_port, iopub_port, shell_port, hb_port). Nr)rrrrrget_connection_infosz&MultiKernelManager.get_connection_info)ridentityrcCsdS)a6Return a zmq Socket connected to the iopub channel. Parameters ========== kernel_id : uuid The id of the kernel identity : bytes (optional) The zmq identity of the socket Returns ======= stream : zmq Socket or ZMQStream Nr)rrrrrr connect_iopubsz MultiKernelManager.connect_iopubcCsdS)a6Return a zmq Socket connected to the shell channel. Parameters ========== kernel_id : uuid The id of the kernel identity : bytes (optional) The zmq identity of the socket Returns ======= stream : zmq Socket or ZMQStream Nr)rrrrrr connect_shellsz MultiKernelManager.connect_shellcCsdS)a8Return a zmq Socket connected to the control channel. Parameters ========== kernel_id : uuid The id of the kernel identity : bytes (optional) The zmq identity of the socket Returns ======= stream : zmq Socket or ZMQStream Nr)rrrrrrconnect_controlsz"MultiKernelManager.connect_controlcCsdS)a6Return a zmq Socket connected to the stdin channel. Parameters ========== kernel_id : uuid The id of the kernel identity : bytes (optional) The zmq identity of the socket Returns ======= stream : zmq Socket or ZMQStream Nr)rrrrrr connect_stdinsz MultiKernelManager.connect_stdincCsdS)a3Return a zmq Socket connected to the hb channel. Parameters ========== kernel_id : uuid The id of the kernel identity : bytes (optional) The zmq identity of the socket Returns ======= stream : zmq Socket or ZMQStream Nr)rrrrrr connect_hbszMultiKernelManager.connect_hb)rrcKs ttS)z Returns the id to associate with the kernel for this request. Subclasses may override this method to substitute other sources of kernel ids. :param kwargs: :return: string-ized version 4 uuid )r%uuiduuid4)rrrrrrR sz MultiKernelManager.new_kernel_id)N)FF)F)Nrw)F)F)F)rl)rl)N)N)N)N)N)Srrr__doc__r rtagrSrr rOrr.rr1rr0rr2r$r'r/rr4r3r<rr:propertyr;r=r>r6rWrHrDListr%rJintrLboolrMOptionalTuplerrXr(r]r_rarjrrerurrr)rvfloatrzr{r^r shutdown_allrrrrrrrrrrbytessocketrrrrrrR __classcell__rr)rFrr*5s       !(     r*c@sFeZdZeddddZedddjddZej Z ej Z ej ZejZd S) AsyncMultiKernelManagerz.jupyter_client.ioloop.AsyncIOLoopKernelManagerTzThe kernel manager class. This is configurable to allow subclassing of the AsyncKernelManager for customized behavior. )r,r+FzWhether to make kernels available before the process has started. The kernel has a `.ready` future which can be awaited before connecting)r+)r,N)rrrrr.rrr`r*rjrerrrurrrrrrrrrsr)#rrfrTrtypingr$rr=Z traitletsrrrrrrrr Ztraitlets.config.configurabler Ztraitlets.utils.importstringr Z kernelspecr rmanagerrutilsrrrZrr'r)r*rrrrrs8               b