o ZhIX@sdZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlZddlmZmZmZmZmZmZmZmZddlmZddlmZd d lmZd d lmZm Z d d l!m"Z"d d l#m$Z$m%Z%m&Z&Gddde'Z(dddZ)GdddeZ*Gddde*Z+dS)z%A kernel manager for multiple kernels) annotationsNwraps)Path)AnyBoolDictDottedObjectNameInstanceUnicodedefaultobserve)LoggingConfigurable) import_item)KernelConnectionInfo)NATIVE_KERNEL_NAMEKernelSpecManager) KernelManager) ensure_asyncrun_syncutcnowc@s eZdZdS)DuplicateKernelErrorN)__name__ __module__ __qualname__rrX/var/www/html/lang_env/lib/python3.10/site-packages/jupyter_client/multikernelmanager.pyrsrf t.Callablereturncstd fd d }|S) zDdecorator for proxying MKM.method(kernel_id) to individual KMs by IDselft.Any kernel_idstrargskwargsr t.Callable | t.Awaitablecs@||}t|j}||i|}||g|Ri||SN) get_kernelgetattrr)r!r#r%r&kmmethodrrrrwrapped!s  zkernel_method..wrappedN) r!r"r#r$r%r"r&r"r r'r)rr/rr.r kernel_methodsr0cseZdZdZeeddjddZee ddZ e dddjddZ e d dddZeddZeddddZdddZedddjddZedZedZeZedddZeddddZed Zed!ddZeZdfd$d% Z dfd&d' Z!dd)d*Z"dd,d-Z#dd1d2Z$dd6d7Z%dd:d;Z&dd@dAZ'ddBdCZ(ddDdEZ)d!dFddGdHZ*e+e*Z,  dddLdMZ-e+e-Z.e/dddNdOZ0e/ ! PdddTdUZ1e/dddVdWZ2ddXdYZ3dddZd[Z4e+e4Z5dd\d]Z6e/dd_d`Z7dddadbZ8e+e8Z9e/ddcddZ:ddedfZ;ddgdhZe/ddpdqZ?e/ !dddudvZ@e/ !dddwdxZAe/ !dddydzZBe/ !ddd{d|ZCe/ !ddd}d~ZDdddZEZFS)MultiKernelManagerz&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_classchanger"r NonecCs||_dSr()_create_kernel_manager_factorykernel_manager_factory)r!r8rrr_kernel_manager_class_changedCsz0MultiKernelManager._kernel_manager_class_changedz)this is kernel_manager_class after importr;rcCs|Sr()r:r!rrr_kernel_manager_factory_defaultIsz2MultiKernelManager._kernel_manager_factory_defaultcstjdfdd }|S) Nr%r"r&r rcs8jrjjr _|dj|i|}|S)Ncontext)shared_contextr?closed_context_default setdefault)r%r&r+Zkernel_manager_ctorr!rrcreate_kernel_managerPs  zPMultiKernelManager._create_kernel_manager_factory..create_kernel_manager)r%r"r&r"r r)rr7)r!rErrDrr:Ms  z1MultiKernelManager._create_kernel_manager_factoryz4Share a single zmq.Context to talk to all my kernels zmq.ContextFdictcCs|jS)z#A shim for backwards compatibility.)_pending_kernelsr=rrr_starting_kernelsfsz$MultiKernelManager._starting_kernelsr?cCsd|_tSNT)_created_contextzmqContextr=rrrrBksz#MultiKernelManager._context_defaultNr%r&cstj|i|i|_dSr()super__init__kernel_id_to_connection_file)r!r%r& __class__rrrPus zMultiKernelManager.__init__cs^|jr|jr|jjs|jr|jd||jztj}Wn ty)YdSw|dS)z:Handle garbage collection. Destroy context if applicable.zDestroying zmq context for %sN) rKr?rAlogdebugdestroyrO__del__AttributeError)r!Z super_delrRrrrWys    zMultiKernelManager.__del__ list[str]c CsZ|jdurt|j}|rdd|D}t|j}t|j}t|jD]}||vrA|||}|j|=|j |=q,|D]a}||jvrNqDz t | }Wn t y`YqDw|jd|d|vrpd|vsqqD|}||j|<|j||jdd}||t|_d |_d |_||_|d|_|jd||j |<qDt|j S) z6Return a list of the kernel ids of the active kernels.NcSsg|]}|r|qSr)is_file).0prrr sz6MultiKernelManager.list_kernel_ids..zLoading connection file %s kernel_namekeyF)parentrTZ owns_kernelidler)external_connection_dirris_diriterdirlistrQkeysvaluesindex_kernelsjsonloads read_text ExceptionrTrU new_kernel_idr;Zload_connection_inforZ last_activityZexecution_state connectionsr#r^ready set_result) r!rbZconnection_fileskvconnection_filer#Zconnection_infor+rrrlist_kernel_idssR        z"MultiKernelManager.list_kernel_idsintcCs t|S)z%Return the number of running kernels.)lenrur=rrr__len__s zMultiKernelManager.__len__r#r$boolcCs ||jvSr()rir!r#rrr __contains__s zMultiKernelManager.__contains__r^ str | Nonetuple[KernelManager, str, str]cCs|d|jdi|}||vrtd||dur|j}i}|jr'|j|d<|jdtj|j d|||j |d|}|||fS)Nr#zKernel already exists: %skernel_spec_managerzkernel-%s.json)rtr`rTr^r) poprnrdefault_kernel_namer~r;ospathjoinconnection_dirrT)r!r^r&r#Zconstructor_kwargsr+rrrpre_start_kernels"   z#MultiKernelManager.pre_start_kernelenvt.Dict[str, str]cCs"||vr|j|j|ddSdS)z Allow to update the environment of the given kernel. Forward the update env request to the corresponding kernel. .. version-added: 8.5 )rN)ri update_env)r!r#rrrrrszMultiKernelManager.update_envr+rkernel_awaitable t.Awaitablec s^z|IdH||j|<|j|dWdSty.}z |j|WYd}~dSd}~wwr()rirHrrmrT exception)r!r#r+rerrr_add_kernel_when_ready  z)MultiKernelManager._add_kernel_when_readyc s^z|IdH|||j|dWdSty.}z |j|WYd}~dSd}~wwr() remove_kernelrHrrmrTr)r!r#rrrrr_remove_kernel_when_readyrz,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*r=rrr_using_pending_kernelss z)MultiKernelManager._using_pending_kernels)r^cs|||\}}}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'!)Zkm_classr#Nr)r isinstancerrTwarningformatr7rSr start_kernelasyncio create_taskrrHrrirpr)r!r^r&r+r#Zstartertaskrrr_async_start_kernels&     z&MultiKernelManager._async_start_kernelnow bool | Nonerestartcs |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)rTinforHr)tcastrFuturerpCancelledErrorrmr cancelledrrshutdown_kernel ensure_futurerr)r!r#rrrr+Zstopperfutrrr_async_shutdown_kernel!s6            z)MultiKernelManager._async_shutdown_kernelcCdS)z,Ask a kernel to shut down by its kernel uuidNrr!r#rrrrrequest_shutdownPz#MultiKernelManager.request_shutdown皙?waittime float | None pollintervalcCs|jd|dS)zDWait for a kernel to finish shutting down, and kill it if it doesn'trNrTr)r!r#rrrrrfinish_shutdownTsz"MultiKernelManager.finish_shutdowncCr)zClean up a kernel's resourcesNrrrrrcleanup_resources^rz$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)rirrzrrrrbsz 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)r[kidrr!rrr]qsz:MultiKernelManager._async_shutdown_all..N)rurerHrirgsetrgatherrrprr#cancelrm)r!rZkidsZkmsZfutsr+rrr_async_shutdown_allls" 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)rpdone RuntimeErrorinterrupt_kernelrTr)r!r#kernelmsgoutrrrrs  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)r!r#rrrr signal_kernelsz 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)rrprrrrestart_kernelrTr)r!r#rrrrrr_async_restart_kernels 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. Nrrzrrris_aliverzMultiKernelManager.is_alivecCs||vr td|dS)zcheck that a kernel id is validzKernel with id not found: %sN)KeyErrorrzrrr_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. )rrirzrrrr)s  zMultiKernelManager.get_kernelcallbackeventcCr)z&add a callback for the KernelRestarterNrr!r#rrrrradd_restart_callbackrz'MultiKernelManager.add_restart_callbackcCr)z)remove a callback for the KernelRestarterNrrrrrremove_restart_callbackrz*MultiKernelManager.remove_restart_callbackdict[str, t.Any]cCr)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). Nrrzrrrget_connection_inforz&MultiKernelManager.get_connection_infoidentity bytes | None socket.socketcCr)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 Nrr!r#rrrr 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_stdin)rz 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_hb;rzMultiKernelManager.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)r!r&rrrrnMs z MultiKernelManager.new_kernel_id)r8r"r r9)r r)r rG)r rF)r%r"r&r"r r9)r r9)r rY)r rv)r#r$r ry)r^r|r&r"r r})r#r$rrr r9)r#r$r+rrrr r9)r#r$rrr r9)r ry)r^r|r&r"r r$)FF)r#r$rrrrr r9)F)r#r$rrr r9)Nr)r#r$rrrrr r9)r#r$rryr r9)r#r$r r)rryr r9)r#r$r r9)r#r$rrvr r9)r#r$rryr r9)r)r#r$rrrr$r r9)r#r$r rr()r#r$rrr r)r&r"r r$)Grrr__doc__r rtagrr rr~r r7r r<rr;r r>r:rr@r?rKrrHpropertyrIrBrrbrirPrWrurxr{rrrrrrrrrrr0rrrrr shutdown_allrrrrrrr)rrrrrrrrrn __classcell__rrrRrr13s         3    ! -          r1c@seZdZUeddddZedddjddZed Z e d dd d Z e j Zded<e jZded<e jZded<e jZded<dS)AsyncMultiKernelManagerz.jupyter_client.ioloop.AsyncIOLoopKernelManagerTzThe kernel manager class. This is configurable to allow subclassing of the AsyncKernelManager for customized behavior. )r5r3FzWhether to make kernels available before the process has started. The kernel has a `.ready` future which can be awaited before connectingr2r4zmq.asyncio.Contextr?r cCsd|_tjSrJ)rKrLrrMr=rrrrBhs z(AsyncMultiKernelManager._context_defaultzt.Callable[..., t.Awaitable]rrrrN)r r)rrrr r7rrrr r?r rBr1rr__annotations__rrrrrrrrrrrWs&  r)rrr r),r __future__rrrjrsockettypingrr functoolsrpathlibrrLZ traitletsrrrr r r r r Ztraitlets.config.configurablerZtraitlets.utils.importstringrconnectrZ kernelspecrrmanagerrutilsrrrrmrr0r1rrrrrs4   (     (