MhIX>dZddlmZddlZddlZddlZddlZddlZddl 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&Gdde'Z(ddZ)GddeZ*Gdde*Z+dS)z%A kernel manager for multiple kernels) annotationsNwraps)Path)AnyBoolDictDottedObjectNameInstanceUnicodedefaultobserve)LoggingConfigurable) import_item)KernelConnectionInfo)NATIVE_KERNEL_NAMEKernelSpecManager) KernelManager) ensure_asyncrun_syncutcnowceZdZdS)DuplicateKernelErrorN)__name__ __module__ __qualname__a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_client/multikernelmanager.pyrrsDrrf t.Callablereturnc>td fd }|S) zDdecorator for proxying MKM.method(kernel_id) to individual KMs by IDselft.Any kernel_idstrargskwargsr#t.Callable | t.Awaitablec||}t|j}||i|}||g|Ri||SN) get_kernelgetattrr)r%r'r)r*kmmethodrr!s r wrappedzkernel_method..wrapped!sd __Y ' 'QZ(( FD #F # # $ +D+++F+++r) r%r&r'r(r)r&r*r&r#r+r)r!r3s` r kernel_methodr4s: 1XX     X  NrceZdZdZeeddZee dZ e dddZ e d d\dZedZedd]dZd]dZedddZedZedZeZed^dZedd_dZedZeddZeZd`fd Z dafd Z!dbd"Z"dcd$Z#ddd(Z$ded,Z%dfd/Z&dgd4Z'dhd5Z(did6Z)dd7djd8Z*e+e*Z, dkdld<Z-e+e-Z.e/dmdnd=Z0e/ dodpdBZ1e/dmdqdCZ2drdDZ3dmdsdEZ4e+e4Z5dtdFZ6e/dudHZ7dmdvdIZ8e+e8Z9e/dddJZ:dtdKZ;drdLZe/dydRZ?e/ dzd{dVZ@e/ dzd{dWZAe/ dzd{dXZBe/ dzd{dYZCe/ dzd{dZZDd|d[ZExZFS)}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#Nonec8||_dSr-)_create_kernel_manager_factorykernel_manager_factory)r%r=s r _kernel_manager_class_changedz0MultiKernelManager._kernel_manager_class_changedCs&*&I&I&K&K###rz)this is kernel_manager_class after importrAr"c*|Sr-)r@r%s r _kernel_manager_factory_defaultz2MultiKernelManager._kernel_manager_factory_defaultIs22444rc@tjdfd }|S)Nr)r&r*r#rcjr@jjr_|dj|i|}|S)Ncontext)shared_contextrHclosed_context_default setdefault)r)r*r0kernel_manager_ctorr%s r create_kernel_managerzPMultiKernelManager._create_kernel_manager_factory..create_kernel_managerPs`" ;<&;#'#8#8#:#:DL!!)T\:::$$d5f55BIr)r)r&r*r&r#r)rr<)r%rNrMs` @r r@z1MultiKernelManager._create_kernel_manager_factoryMs@)$*CDD       %$rz4Share a single zmq.Context to talk to all my kernels zmq.ContextFdictc|jS)z#A shim for backwards compatibility.)_pending_kernelsrDs r _starting_kernelsz$MultiKernelManager._starting_kernelsfs $$rrHc6d|_tjSNT)_created_contextzmqContextrDs r rKz#MultiKernelManager._context_defaultks ${}}rNr)r*cHtj|i|i|_dSr-)super__init__kernel_id_to_connection_file)r%r)r* __class__s r r\zMultiKernelManager.__init__us-$)&)))=?)))rc|jrN|jrG|jjs;|jr|jd||j t j}|dS#t$rYdSwxYw)z:Handle garbage collection. Destroy context if applicable.zDestroying zmq context for %sN) rVrHrJlogdebugdestroyr[__del__AttributeError)r% super_delr^s r rczMultiKernelManager.__del__ys   #T\ #$,:M #x F>EEE L " " " I IKKKKK    DD sA77 BB list[str]c|jt|j}|rd|D}t |j}t |j}t |jD]1}||vr+|||}|j|=|j |=2|D]$}||jvr tj | }n#t$rYSwxYw|jd|d|vrd|vs{|}||j|<|||jd}||t'|_d|_d |_||_|d|_|jd||j |<&t |j S) z6Return a list of the kernel ids of the active kernels.Nc:g|]}||Sr)is_file).0ps r z6MultiKernelManager.list_kernel_ids..s*#`#`#`!TUT]T]T_T_#`A#`#`#`rzLoading connection file %s kernel_namekeyF)parentr` owns_kernelidler)external_connection_dirris_diriterdirlistr]keysvaluesindex_kernelsjsonloads read_text Exceptionr`ra new_kernel_idrAload_connection_infor last_activityexecution_state connectionsr'rmready set_result) r%rrconnection_fileskvconnection_filer'connection_infor0s r list_kernel_idsz"MultiKernelManager.list_kernel_idss[  ' 3&*4+G&H&H #&--//) 2#`#`/F/N/N/P/P#`#`#` :??AABB:AACCDD'+D,M,T,T,V,V'W'W55O&.>>>$%aggo&>&>$?  =iH M)4(822O&$*K*R*R*T*TTT !@D +5577AA%!!! !HNN#?QQQ)_<</AYAY $ 2 2 4 4ICRD5i@44# H$)5B ++O<<<'-xxB$)/B&%&BN#,BL%4]%CBNH''---/1DM),,DM&&(()))s&D== E  E intcDt|S)z%Return the number of running kernels.)lenrrDs r __len__zMultiKernelManager.__len__s4''))***rr'r(boolc||jvSr-)ryr%r's r __contains__zMultiKernelManager.__contains__sDM))rrm str | Nonetuple[KernelManager, str, str]c,|d|jdi|}||vrtd|z||j}i}|jr |j|d<|jdt j|j d|z||j |d|}|||fS)Nr'zKernel already exists: %skernel_spec_managerzkernel-%s.json)rror`rmr) popr~rdefault_kernel_namerrAospathjoinconnection_dirr`)r%rmr*r'constructor_kwargsr0s r pre_start_kernelz#MultiKernelManager.pre_start_kernelsJJ{,>D,>,H,H,H,HII   &'BY'NOO O  2K   # Q8<8P 4 5 (T ( GLL)<>NQZ>Z[[#   !   ; ))renvt.Dict[str, str]cT||vr#|j||dSdS)z Allow to update the environment of the given kernel. Forward the update env request to the corresponding kernel. .. version-added: 8.5 )rN)ry update_env)r%r'rs r rzMultiKernelManager.update_envs;    M) $ / /C / 8 8 8 8 8  rr0rkernel_awaitable t.AwaitablecK |d{V||j|<|j|ddS#t$r%}|j|Yd}~dSd}~wwxYwr-)ryrRrr}r` exception)r%r'r0res r _add_kernel_when_readyz)MultiKernelManager._add_kernel_when_readys "" " " " " " " "')DM) $  ! % %i 6 6 6 6 6 " " " H  q ! ! ! ! ! ! ! ! ! "s-3 A"AA"cK |d{V|||j|ddS#t$r%}|j|Yd}~dSd}~wwxYwr-) remove_kernelrRrr}r`r)r%r'rrs r _remove_kernel_when_readyz,MultiKernelManager._remove_kernel_when_readys "" " " " " " " "   y ) ) )  ! % %i 6 6 6 6 6 " " " H  q ! ! ! ! ! ! ! ! ! "s8> A-A((A-c$t|ddS)zReturns a boolean; a clearer method for determining if this multikernelmanager is using pending kernels or not use_pending_kernelsF)r/rDs r _using_pending_kernelsz)MultiKernelManager._using_pending_kernelsst2E:::r)rmc.K|||\}}}t|ts8|jd|jj||d<t|j di|}tj | |||}||j |<|r ||j|<n:|d{V|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'!)km_classr'Nr)r isinstancerr`warningformatr<r^r start_kernelasyncio create_taskrrRrryrr)r%rmr*r0r'startertasks r _async_start_kernelz&MultiKernelManager._async_start_kernels2&*%:%:;%O%O"K"m,,  H  Zaa!6@b    ({r888899"4#>#>y"g#V#VWW+/i(  & & ( ( +(*DM) $ $JJJJJJJx!!## +h((***rnow bool | NonerestartcpK|jd|||jvr|j|} |d{V||}t jt j|jd{Vn6#t j $rYn%t$r| |YdSwxYw||}|j s0|j r| |dSt|||}t j|||}||j|<|s:|d{V|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)r`inforRr.tcastrFuturerCancelledErrorr}r cancelledrrshutdown_kernel ensure_futurerr)r%r'rrrr0stopperfuts r _async_shutdown_kernelz)MultiKernelManager._async_shutdown_kernel!s"  +Y777 - - -(3D  __Y//fW^RX6666666666)      ""9--- __Y ' 'x!!## (:(:(<(<    y ) ) ) Fr11#w??@@#D$B$B9g$V$VWW+.i(**,, +IIIIIIIx!!## +h((***  + + + +sAA==B0B0/B0cdS)z,Ask a kernel to shut down by its kernel uuidNrr%r'rs r request_shutdownz#MultiKernelManager.request_shutdownPr皙?waittime float | None pollintervalc<|jd|dS)zDWait for a kernel to finish shutting down, and kill it if it doesn'trNr`r)r%r'rrs r finish_shutdownz"MultiKernelManager.finish_shutdownTs!  +Y77777rcdS)zClean up a kernel's resourcesNrrs r cleanup_resourcesz$MultiKernelManager.cleanup_resources^rrc8|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)ryrrs r rz MultiKernelManager.remove_kernelbs}  D111rcK}|tjz }tj}fdt |D}t j|d{VrW|D]V} |j d{V#t j $r'j|j  YFt$rYQwxYwdSdS)zShutdown all kernels.c>g|]}|S)r)r)rjkidrr%s r rlz:MultiKernelManager._async_shutdown_all..qs,OOOc++CS+99OOOrN)rrurRryrwsetrgatherrrrr'cancelr})r%rkidskmsfutsr0s`` r _async_shutdown_allz&MultiKernelManager._async_shutdown_allls5##%% T*+++4=''))**OOOOOSYYOOOnd########  & & ( (   (NNNNNNNN-AAA)",7>>@@@@@ D     s B--3C." C.-C.c||}|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.rdone RuntimeErrorinterrupt_kernelr`r)r%r'kernelmsgouts r rz#MultiKernelManager.interrupt_kernelsk++|  "" $CCs## #%%''  . ::: rsignumc>|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'rs r signal_kernelz MultiKernelManager.signal_kernels#  2IvFFFFFrc8K||}|r*|jsd}t |t ||d{V|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.rrrrrrestart_kernelr`r)r%r'rrrs r _async_restart_kernelz(MultiKernelManager._async_restart_kernels++  & & ( ( $1B1B1D1D $ACs## #600S099:::::::::  ,i88888rcdS)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. Nrrs r is_alivezMultiKernelManager.is_aliverrc2||vrtd|zdS)zcheck that a kernel id is validzKernel with id not found: %sN)KeyErrorrs r _check_kernel_idz#MultiKernelManager._check_kernel_ids) D 9IEFF F ! rcF|||j|S)zGet the single KernelManager object for a kernel by its uuid. Parameters ========== kernel_id : uuid The id of the kernel. )rryrs r r.zMultiKernelManager.get_kernels% i(((}Y''rcallbackeventcdS)z&add a callback for the KernelRestarterNrr%r'rrs r add_restart_callbackz'MultiKernelManager.add_restart_callbackrrcdS)z)remove a callback for the KernelRestarterNrrs r remove_restart_callbackz*MultiKernelManager.remove_restart_callbackrrdict[str, t.Any]cdS)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). Nrrs r get_connection_infoz&MultiKernelManager.get_connection_inforridentity bytes | None socket.socketcdS)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'rs r connect_iopubz MultiKernelManager.connect_iopubrrcdS)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 Nrrs r connect_shellz MultiKernelManager.connect_shellrrcdS)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 Nrrs r connect_controlz"MultiKernelManager.connect_controlrrcdS)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 Nrrs r connect_stdinz MultiKernelManager.connect_stdin)rrcdS)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 Nrrs r connect_hbzMultiKernelManager.connect_hb;rrc BttjS)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*s r r~z MultiKernelManager.new_kernel_idMs4:<<   r)r=r&r#r>)r#r")r#rP)r#rO)r)r&r*r&r#r>)r#r>)r#rf)r#r)r'r(r#r)rmrr*r&r#r)r'r(rrr#r>)r'r(r0rrrr#r>)r'r(rrr#r>)r#r)rmrr*r&r#r()FF)r'r(rrrrr#r>)F)r'r(rrr#r>)Nr)r'r(rrrrr#r>)r'r(rrr#r>)r'r(r#r)rrr#r>)r'r(r#r>)r'r(rrr#r>)r'r(rrr#r>)r)r'r(rr"rr(r#r>)r'r(r#rr-)r'r(rrr#r)r*r&r#r()Grrr__doc__r rtagrr rrr r<rrBrrAr rEr@rrIrHrVr rRpropertyrSrKrrrryr\rcrrrrrrrrrrrrrr4rrrrr shutdown_allrrrrrrr.rrrrrr r r r~ __classcell__)r^s@r r6r63s00!'!J cc#(#4FFF++3   cc  W #$$LLL%$L!S&QRRR W %&&555'&5 % % % %T C cc h}%%GtE{{tvv %%%X% WYWR[[N%gdt<<<tvvH@@@@@@      1*1*1*1*f++++********2 9 9 9 9"""""""";;;; FJB8/00L !$ ++++++++++Zh566O;;;;];"&%( 8888]8,,,,],2222$8/00L  G G G] G99999,X344N    ]  GGGG ( ( ( (AJ5555]5 AJ8888]8    ] "7;    ] "7;    ] "7;    ] "7;    ] "7;    ] "!!!!!!!!rr6ceZdZUedddZedddZed Z e d dd Z e j Zd ed<e jZd ed<e jZd ed<e jZd ed<dS)AsyncMultiKernelManagerz.jupyter_client.ioloop.AsyncIOLoopKernelManagerTzThe kernel manager class. This is configurable to allow subclassing of the AsyncKernelManager for customized behavior. )r:r8FzWhether to make kernels available before the process has started. The kernel has a `.ready` future which can be awaited before connectingr7r9zmq.asyncio.ContextrHr#cLd|_tjSrU)rVrWrrXrDs r rKz(AsyncMultiKernelManager._context_defaulths ${""$$$rzt.Callable[..., t.Awaitable]rrrrN)r#r)rrrr r<rrrr rHr rKr6rr__annotations__rrrrrrrrr rrWs++8 $ O cc  h,--G WY%%%%2D1WLWWWW3E3[N[[[[4F4]O]]]]1C1WLWWWWWWrr)r!r"r#r"),r __future__rrrzrsockettypingrr functoolsrpathlibrrW traitletsrrr r r r r rtraitlets.config.configurablertraitlets.utils.importstringrconnectr kernelspecrrmanagerrutilsrrrr}rr4r6rrrr r's++#"""""   \\\\\\\\\\\\\\\\\\\\======444444))))))========""""""1111111111     9   *a!a!a!a!a!,a!a!a!HXXXXX0XXXXXr