B \A&@sdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZddlmZdd lmZmZGd d d eZd d ZGdddeZdS)z%A kernel manager for multiple kernels)absolute_importN)LoggingConfigurable) import_item)InstanceDictListUnicodeAnyDottedObjectName) unicode_type)NATIVE_KERNEL_NAMEKernelSpecManagerc@s eZdZdS)DuplicateKernelErrorN)__name__ __module__ __qualname__rr@lib/python3.7/site-packages/jupyter_client/multikernelmanager.pyrsrcsfdd}|S)zDdecorator for proxying MKM.method(kernel_id) to individual KMs by IDcs6||}t|j}|||}||f|||S)N) get_kernelgetattrr)self kernel_idargskwargskmmethodr)frrwrappeds    zkernel_method..wrappedr)rrr)rr kernel_methods r c@s~eZdZdZeedddZeeddZ e ddddZ dd Z e d d Zd d ZedZddZedZeZddZddZddZdDddZedEddZedFddZedGd!d"ZedHd#d$Zd%d&ZdId'd(Zed)d*Z ed+d,Z!edJd-d.Z"ed/d0Z#d1d2Z$d3d4Z%edKd6d7Z&edLd8d9Z'ed:d;Z(edMdd?Z*edOd@dAZ+edPdBdCZ,dS)QMultiKernelManagerz&A class for managing multiple kernels.Tz'The name of the default kernel to start)Zconfighelp)Z allow_nonez)jupyter_client.ioloop.IOLoopKernelManagerzThe kernel manager class. This is configurable to allow subclassing of the KernelManager for customized behavior. cCst||_dS)N)rkernel_manager_factory)rnameoldnewrrr_kernel_manager_class_changed9sz0MultiKernelManager._kernel_manager_class_changedz)this is kernel_manager_class after import)r"cCs t|jS)N)rkernel_manager_class)rrrr_kernel_manager_factory_default=sz2MultiKernelManager._kernel_manager_factory_defaultz zmq.ContextcCs tjS)N)zmqZContextinstance)rrrr_context_defaultAsz#MultiKernelManager._context_defaultcCst|jS)z6Return a list of the kernel ids of the active kernels.)list_kernelskeys)rrrrlist_kernel_idsHsz"MultiKernelManager.list_kernel_idscCs t|S)z%Return the number of running kernels.)lenr1)rrrr__len__NszMultiKernelManager.__len__cCs ||jkS)N)r/)rrrrr __contains__RszMultiKernelManager.__contains__NcKs|dtt}||kr(td||dkr6|j}i}|jrJ|j|d<|jftj |j d|||j |d|}|j f|||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 picked using a uuid. The kernel ID for the newly started kernel is returned. rzKernel already exists: %sNkernel_spec_managerzkernel-%s.json)connection_fileparentlog kernel_name)popr uuidZuuid4rdefault_kernel_namer5r#ospathjoinconnection_dirr8 start_kernelr/)rr9rrZconstructor_kwargsrrrrrAUs       zMultiKernelManager.start_kernelFcCs|jd|||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)r8info remove_kernel)rrnowrestartrrrshutdown_kernelrs z"MultiKernelManager.shutdown_kernelcCsdS)z,Ask a kernel to shut down by its kernel uuidNr)rrrErrrrequest_shutdownsz#MultiKernelManager.request_shutdown皙?cCs|jd|dS)zMWait for a kernel to finish shutting down, and kill it if it doesn't zKernel shutdown: %sN)r8rB)rrZwaittimeZ pollintervalrrrfinish_shutdownsz"MultiKernelManager.finish_shutdowncCsdS)zClean up a kernel's resourcesNr)rrr6rrrcleanupszMultiKernelManager.cleanupcCs |j|S)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. )r/r:)rrrrrrCsz MultiKernelManager.remove_kernelcCsP|}x|D]}||qWx*|D]"}||||||q&WdS)zShutdown all kernels.N)r1rGrIrJrC)rrDZkidsZkidrrr shutdown_alls    zMultiKernelManager.shutdown_allcCs|jd|dS)zInterrupt (SIGINT) the kernel by its uuid. Parameters ========== kernel_id : uuid The id of the kernel to interrupt. zKernel interrupted: %sN)r8rB)rrrrrinterrupt_kernels z#MultiKernelManager.interrupt_kernelcCs|jd||fdS)aSends 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. zSignaled Kernel %s with %sN)r8rB)rrZsignumrrr signal_kernels z MultiKernelManager.signal_kernelcCs|jd|dS)zRestart a kernel by its uuid, keeping the same ports. Parameters ========== kernel_id : uuid The id of the kernel to interrupt. zKernel restarted: %sN)r8rB)rrrDrrrrestart_kernels z!MultiKernelManager.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. )rQr/)rrrrrrs zMultiKernelManager.get_kernelrEcCsdS)z&add a callback for the KernelRestarterNr)rrcallbackeventrrradd_restart_callbacksz'MultiKernelManager.add_restart_callbackcCsdS)z)remove a callback for the KernelRestarterNr)rrrRrSrrrremove_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_infocCsdS)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)rridentityrrr 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)rrrWrrr connect_shellsz MultiKernelManager.connect_shellcCsdS)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)rrrWrrr 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)rrrWrrr connect_hb/szMultiKernelManager.connect_hb)N)FF)F)NrH)T)F)F)rE)rE)N)N)N)N)-rrr__doc__rr r<rrr5r r(r'r r#r)contextr,r@rr/r1r3r4rAr rFrGrIrJrCrKrLrMrNrOrQrrTrUrVrXrYrZr[rrrrr!*s`                r!)r\Z __future__rr=r;r*Ztraitlets.config.configurablerZipython_genutils.importstringrZ traitletsrrrrr r Zipython_genutils.py3compatr Z kernelspecr r Exceptionrr r!rrrrs