o iL@sdZddlZddlZddlZddlZddlZddlmZddl Z ddl m Z m Z m Z mZmZmZmZmZddlmZddlmZddlmZmZdd lmZdd lmZmZGd d d eZ d ej!dej!fddZ"GdddeZ#Gddde#Z$dS)z%A kernel manager for multiple kernelsN)wraps)AnyBoolDictDottedObjectNameInstanceUnicodedefaultobserve)LoggingConfigurable) import_item)NATIVE_KERNEL_NAMEKernelSpecManager) KernelManager) ensure_asyncrun_syncc@s eZdZdS)DuplicateKernelErrorN)__name__ __module__ __qualname__rr\/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/jupyter_client/multikernelmanager.pyrsrfreturnc sBtdtjdtdtjdtjdtjtjtjff fdd }|S)zDdecorator for proxying MKM.method(kernel_id) to individual KMs by IDself kernel_idargskwargsrcs@||}t|j}||i|}||g|Ri||SN) get_kernelgetattrr)rrrrkmmethodrrrrwrappeds  zkernel_method..wrapped)rtrstrUnionCallable Awaitable)rr&rr%r kernel_methodsr,c sBeZdZdZeeddjddZee ddZ e dddjddZ e d d d Zed dZed ddZdejfddZedddjddZedZedZeZeddZeddejfddZ edZ!eZ"fddZ#dej$e%fddZ&de'fd d!Z(d"e%de)fd#d$Z*d%ej+e%d&ejdej,e-e%e%ffd'd(Z.d"e%d)e-d*ej/dd+fd,d-Z0d"e%d*ej/dd+fd.d/Z1d0d1Z2d+d2d%ej+e%d&ejde%fd3d4Z3e4e3Z5  dhd"e%d5ej+e)d6ej+e)dd+fd7d8Z6e4e6Z7e8did"e%d6ej+e)dd+fd9d:Z9e8 + ;djd"e%dd?Z;e8did"e%d6e)dd+fd@dAZe4e>Z?d"e%dd+fdFdGZ@e8d"e%dHe'dd+fdIdJZAdid"e%d5e)dd+fdKdLZBe4eBZCe8d"e%de)fdMdNZDd"e%dd+fdOdPZEd"e%de-fdQdRZFe8 6dkd"e%dSejdTe%dd+fdUdVZGe8 6dkd"e%dSejdTe%dd+fdWdXZHe8d"e%deje%ejffdYdZZIe8 +dld"e%d[ej+eJdeKjKfd\d]ZLe8 +dld"e%d[ej+eJdeKjKfd^d_ZMe8 +dld"e%d[ej+eJdeKjKfd`daZNe8 +dld"e%d[ej+eJdeKjKfdbdcZOe8 +dld"e%d[ej+eJdeKjKfdddeZPd&ejde%fdfdgZQZRS)mMultiKernelManagerz&A class for managing multiple kernels.z'The name of the default kernel to starthelpTconfig) allow_nonez)jupyter_client.ioloop.IOLoopKernelManagerzThe kernel manager class. This is configurable to allow subclassing of the KernelManager for customized behavior. kernel_manager_classcCs||_dSr)_create_kernel_manager_factorykernel_manager_factory)rchangerrr_kernel_manager_class_changed>sz0MultiKernelManager._kernel_manager_class_changedz)this is kernel_manager_class after importr5cCs|Sr)r4rrrr_kernel_manager_factory_defaultDsz2MultiKernelManager._kernel_manager_factory_defaultrcs.tjdtjdtjdtffdd }|S)Nrrrcs8jrjjr _|dj|i|}|S)Ncontext)shared_contextr:closed_context_default setdefault)rrr"kernel_manager_ctorrrrcreate_kernel_managerKs  zPMultiKernelManager._create_kernel_manager_factory..create_kernel_manager)r r3r'rr)rrArr?rr4Hs   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_kernelsr8rrr_starting_kernelsasz$MultiKernelManager._starting_kernelsr:cCsd|_tSNT)_created_contextzmqContextr8rrrr=fsz#MultiKernelManager._context_defaultcs^|jr|jr|jjs|jr|jd||jztj}Wn ty)YdSw|dS)z:Handle garbage collection. Destroy context if applicable.zDestroying zmq context for %sN) rEr:r<logdebugdestroysuper__del__AttributeError)r super_del __class__rrrMos    zMultiKernelManager.__del__cCst|jS)z6Return a list of the kernel ids of the active kernels.)list_kernelskeysr8rrrlist_kernel_ids|sz"MultiKernelManager.list_kernel_idscCs t|S)z%Return the number of running kernels.)lenrUr8rrr__len__s zMultiKernelManager.__len__rcCs ||jvSr)rSrrrrr __contains__s zMultiKernelManager.__contains__ kernel_namercCs|d|jdi|}||vrtd||dur|j}i}|jr'|j|d<|jdtj|j d|||j |d|}|||fS)NrzKernel already exists: %skernel_spec_managerzkernel-%s.json)connection_fileparentrIrZr) pop new_kernel_idrdefault_kernel_namer[r5ospathjoinconnection_dirrI)rrZrrconstructor_kwargsr"rrrpre_start_kernels"   z#MultiKernelManager.pre_start_kernelr"kernel_awaitableNc s^z|IdH||j|<|j|dWdSty.}z |j|WYd}~dSd}~wwr)rSrBr^ ExceptionrI exception)rrr"rgerrr_add_kernel_when_ready  z)MultiKernelManager._add_kernel_when_readyc s^z|IdH|||j|dWdSty.}z |j|WYd}~dSd}~wwr) remove_kernelrBr^rhrIri)rrrgrjrrr_remove_kernel_when_readyrlz,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!r8rrr_using_pending_kernelss z)MultiKernelManager._using_pending_kernels)rZcs|||\}}}t|ts|jdj|jjd||d<t|j di|}t | |||}||j |<|rC||j|<|S|IdH|jrR|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'!)km_classrNr)rf isinstancerrIwarningformatr3rQr start_kernelasyncio create_taskrkrBrprSreadyri)rrZrr"rstartertaskrrr_async_start_kernels&     z&MultiKernelManager._async_start_kernelnowrestartcs |jd|||jvrB|j|}z|IdH||}ttj|jIdHWntj y3Ynt yA| |YdSw||}|j sX|j rX| |dSt|||}t|||}||j|<|s|IdH|j r|j dSdS)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? Kernel shutdown: %sN)rIinforBr r'castrvFuturerxCancelledErrorrhrm cancelledrirshutdown_kernel ensure_futurernrp)rrr|r}rzr"stopperfutrrr_async_shutdown_kernels6            z)MultiKernelManager._async_shutdown_kernelcCdS)z,Ask a kernel to shut down by its kernel uuidNrrrr}rrrrequest_shutdownz#MultiKernelManager.request_shutdown皙?waittime pollintervalcCs|jd|dS)zDWait for a kernel to finish shutting down, and kill it if it doesn'tr~NrIr)rrrrrrrfinish_shutdownsz"MultiKernelManager.finish_shutdowncCr)zClean up a kernel's resourcesNrrrrrcleanup_resourcesrz$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)rSr^rXrrrrm"sz MultiKernelManager.remove_kernelc s}|tj7}tj}fddt|D}tj|IdHrR|D]&}z|j IdHWq-tj yIj|j  Yq-t yQYq-wdSdS)zShutdown all kernels.csg|] }j|dqS)r|)r).0kidr|rrr 1sz:MultiKernelManager._async_shutdown_all..N)rUrRrBrSvaluessetrvgatherrprxrrcancelrh)rr|kidskmsfutsr"rrr_async_shutdown_all,s" z&MultiKernelManager._async_shutdown_allcCs:||}|jsd}t||}|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)r rxdone RuntimeErrorinterrupt_kernelrIr)rrkernelmsgoutrrrr@s  z#MultiKernelManager.interrupt_kernelsignumcCs|jd||dS)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 %sNr)rrrrrr signal_kernelPsz MultiKernelManager.signal_kernelcsR||}|r|jsd}t|t|j|dIdH|jd|dS)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.rNzKernel restarted: %s) r rprxrrrrestart_kernelrIr)rrr|rrrrr_async_restart_kernel`s z(MultiKernelManager._async_restart_kernelcCr)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. NrrXrrris_alivexrzMultiKernelManager.is_alivecCs||vr td|dS)zcheck that a kernel id is validzKernel with id not found: %sN)KeyErrorrXrrr_check_kernel_ids z#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. )rrSrXrrrr s  zMultiKernelManager.get_kernelcallbackeventcCr)z&add a callback for the KernelRestarterNrrrrrrrradd_restart_callbackrz'MultiKernelManager.add_restart_callbackcCr)z)remove a callback for the KernelRestarterNrrrrrremove_restart_callbackrz*MultiKernelManager.remove_restart_callbackcCr)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). NrrXrrrget_connection_inforz&MultiKernelManager.get_connection_infoidentitycCr)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 Nrrrrrrr connect_iopubrz MultiKernelManager.connect_iopubcCr)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 Nrrrrr connect_shellrz MultiKernelManager.connect_shellcCr)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 Nrrrrrconnect_controlrz"MultiKernelManager.connect_controlcCr)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 Nrrrrr connect_stdinrz MultiKernelManager.connect_stdincCr)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 Nrrrrr connect_hbrzMultiKernelManager.connect_hbcKs 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)rrrrrr_ s z MultiKernelManager.new_kernel_id)FF)F)Nr)r}r)Srrr__doc__rrtagr`rrr[rr3r r7rr5r r9r'r*r4rr;r:rErrBpropertyrCrFrGr=rdrSrMListr(rUintrWboolrYOptionalTuplerrfr+rkrnrpr{rrurrr,rfloatrrrmr shutdown_allrrrrrrr rrrbytessocketrrrrrr_ __classcell__rrrPrr-.sl             # -              r-c@seZdZUeddddZedddjddZed Z e d d e j j fd d ZejZejdejfed<ejZejdejfed<ejZejdejfed<ejZejdejfed<dS)AsyncMultiKernelManagerz.jupyter_client.ioloop.AsyncIOLoopKernelManagerTzThe kernel manager class. This is configurable to allow subclassing of the AsyncKernelManager for customized behavior. )r1r/FzWhether to make kernels available before the process has started. The kernel has a `.ready` future which can be awaited before connectingr.r0zzmq.asyncio.Contextr:rcCsd|_tjSrD)rErFrvrGr8rrrr=(s z(AsyncMultiKernelManager._context_default.rurrrN)rrrrr3rrrorr:r rFrvrGr=r-r{rur'r*r+__annotations__rrrrrrrrrrrs>  r)%rrvrartypingr'r functoolsrrF traitletsrrrrrrr r traitlets.config.configurabler traitlets.utils.importstringr kernelspecrrmanagerrutilsrrrhrr*r,r-rrrrrs* (   l