Ne;JfdZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZd dlmZd dlmZd dlmZd dlmZd dlmZGddeZdej>dej>fdZ GddeZ!Gdde!Z"y)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 eZdZy)DuplicateKernelErrorN)__name__ __module__ __qualname__C/usr/lib/python3/dist-packages/jupyter_client/multikernelmanager.pyrrsrrfreturnc dtjdtdtjdtjdtjtjtj ff fd }|S)zDdecorator for proxying MKM.method(kernel_id) to individual KMs by IDself kernel_idargskwargsrc|j|}t|j}||i|}||g|i||SN) get_kernelgetattrr)rrr r!kmmethodrrs rwrappedzkernel_method..wrapped$sM__Y 'QZZ( D #F # $ +D+F+r)trstrUnionCallable Awaitable)rr)s` r kernel_methodr/!sY ee  # ,-EE =>UU  Q[[( )  Nrc deZdZdZeedj dZee dZ e ddj dZ e d d Zed Zed d Zdej(fdZeddj dZedZedZeZedZeddej>fdZ edZ!eZ"fdZ#dejHe%fdZ&de'fdZ(de%de)fdZ*dejVe%dejdejXe-e%e%ffdZ.de%de-d ej^dd!fd"Z0de%d ej^dd!fd#Z1d$Z2 dCdejVe%dejde%fd%Z3e4e3Z5 dDde%d&ejVe)d'ejVe)dd!fd(Z6e4e6Z7e8dEde%d'ejVe)dd!fd)Z9e8 dFde%d*ejVe:d+ejVe:dd!fd,Z;e8dEde%d'e)dd!fd-Ze4e>Z?de%dd!fd0Z@e8de%d1e'dd!fd2ZAdEde%d&e)dd!fd3ZBe4eBZCe8de%de)fd4ZDde%dd!fd5ZEde%de-fd6ZFe8 dGde%d7ej(d8e%dd!fd9ZGe8 dGde%d7ej(d8e%dd!fd:ZHe8de%dej4e%ejffd;ZIe8dCde%dZMe8dCde%d.create_kernel_managerRsQ""<<&&#'#8#8#:DL!!)T\\:$d5f5BIr)r r7r*rr)rrHrGs` @rr9z1MultiKernelManager._create_kernel_manager_factoryOs>)$*C*CD  !%% M %$rz4Share a single zmq.Context to talk to all my kernelsz zmq.ContextFc|jS)z#A shim for backwards compatibility.)_pending_kernelsr>s r_starting_kernelsz$MultiKernelManager._starting_kernelshs$$$rrBc8d|_tjS)NT)_created_contextzmqContextr>s rrEz#MultiKernelManager._context_defaultms ${{}rc(|jrd|jrX|jjsB|jr|jj d||jj  t |}|y#t$rYywxYw)z:Handle garbage collection. Destroy context if applicable.zDestroying zmq context for %sN) rMrBrDlogdebugdestroysuper__del__AttributeError)r super_del __class__s rrVzMultiKernelManager.__del__vsq  T\\$,,:M:Mxx>E LL " I K   s3 B BBcHt|jjS)z6Return a list of the kernel ids of the active kernels.)list_kernelskeysr>s rlist_kernel_idsz"MultiKernelManager.list_kernel_idssDMM&&())rc4t|jS)z%Return the number of running kernels.)lenr^r>s r__len__zMultiKernelManager.__len__s4'')**rrc||jvSr#)r\rrs r __contains__zMultiKernelManager.__contains__sDMM))r kernel_namer!cb|jd|jdi|}||vrtd|z| |j}i}|jr|j|d<|j dt jj|jd|z||j|d|}|||fS)NrzKernel already exists: %skernel_spec_managerzkernel-%s.json)connection_fileparentrRrer) pop new_kernel_idrdefault_kernel_namergr:ospathjoinconnection_dirrR)rrer!rconstructor_kwargsr&s rpre_start_kernelz#MultiKernelManager.pre_start_kernelsJJ{,>D,>,>,H,HI  &'BY'NO O  22K   # #8<8P8P 4 5 (T ( ( GGLL)<)<>NQZ>Z[#  !  ; ))rr&kernel_awaitableNcK |d{||j|<|jj|dy70#t$r%}|jj |Yd}~yd}~wwxYwwr#)r\rJrj ExceptionrR exception)rrr&rses r_add_kernel_when_readyz)MultiKernelManager._add_kernel_when_readysc "" " "')DMM) $  ! ! % %i 6 # " HH  q ! ! "s7A-<:/<A-< A*A% A-%A**A-cK |d{|j||jj|dy72#t$r%}|jj |Yd}~yd}~wwxYwwr#) remove_kernelrJrjrurRrv)rrrsrws r_remove_kernel_when_readyz,MultiKernelManager._remove_kernel_when_readysc "" " "   y )  ! ! % %i 6 # " HH  q ! ! "s7A/><1>A/> A,A'"A/'A,,A/ct|ddS)zReturns a boolean; a clearer method for determining if this multikernelmanager is using pending kernels or not use_pending_kernelsF)r%r>s r_using_pending_kernelsz)MultiKernelManager._using_pending_kernelsst2E::rcDK|j||\}}}t|ts?|jj dj |j j||d<t|jdi|}tj|j|||}||j|<|jr||j|<|S|d{|j j#r|j j#|S7:w)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)rr isinstancerrRwarningformatr7rYr start_kernelasyncio create_taskrxrJr~r\readyrv)rrer!r&rstartertasks r_async_start_kernelz&MultiKernelManager._async_start_kernels&*%:%:;%O"K"m, HH  Zaa!66@@b  ({r889""4#>#>y"g#VW+/i(  & & ((*DMM) $ JJxx!!#hh((** sC!D #D$;D nowrestartcK|jjd|z||jvra|j|} |d{|j|}t j t j|jd{|j|}|jjs,|jjr|j|yt|j||}t j |j#||}||j|<|j%s?|d{|jjr|jjyy7B7#t j$rYt$r|j|YywxYw7tw)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)rRinforJr$r*castrFuturerCancelledErrorrurz cancelledrvrshutdown_kernel ensure_futurer{r~)rrrrrr&stopperfuts r_async_shutdown_kernelz)MultiKernelManager._async_shutdown_kernelsu"  +i78 -- -((3D  __Y/ffW^^RXX666 __Y 'xx!!#(:(:(<   y ) r11#w?@##D$B$B9g$VW+.i(**,IIxx!!#hh((**$-#6))  ""9-  sf>@ s`AD#C>DC D0C?CC DC:D D DDDDc|j|}|jjs td|j }|j j d|z|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_kernelrRr)rrkernelouts rrz#MultiKernelManager.interrupt_kernelGsW+||  "PQ Q%%'  .:; rsignumcF|jjd|d|y)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 z with Nr)rrrs r signal_kernelz MultiKernelManager.signal_kernelVs  iHIrc K|j|}|jr%|jjs t dt |j |d{}|jjd|z|S7$w)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$r~rrrrrestart_kernelrRr)rrrrrs r_async_restart_kernelz(MultiKernelManager._async_restart_kernelfsz+  & & (<<$$&"#RSS !6!63!6!?@@  ,y89 AsA%B'B (%Bcy)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. Nrrcs ris_alivezMultiKernelManager.is_aliverrc(||vrtd|zy)zcheck that a kernel id is validzKernel with id not found: %sN)KeyErrorrcs r_check_kernel_idz#MultiKernelManager._check_kernel_ids D 9IEF F !rcB|j||j|S)zGet the single KernelManager object for a kernel by its uuid. Parameters ========== kernel_id : uuid The id of the kernel. )rr\rcs rr$zMultiKernelManager.get_kernels! i(}}Y''rcallbackeventcy)z&add a callback for the KernelRestarterNrrrrrs radd_restart_callbackz'MultiKernelManager.add_restart_callbackrrcy)z)remove a callback for the KernelRestarterNrrs rremove_restart_callbackz*MultiKernelManager.remove_restart_callbackrrcy)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). Nrrcs rget_connection_infoz&MultiKernelManager.get_connection_inforridentitycy)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 Nrrrrs r connect_iopubz MultiKernelManager.connect_iopubrrcy)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_shellrrcy)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 rconnect_controlz"MultiKernelManager.connect_controlrrcy)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_stdinrrcy)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_hbrrc <ttjS)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)rr!s rrkz MultiKernelManager.new_kernel_id s4::<  rr#)FF)F)Ng?)r)Srrr__doc__r rtagrlrrrgrr7r r<rr:rr?r*r-r9rrCrBrMrrJpropertyrKrNrOrErpr\rVListr+r^intraboolrdOptionalTuplerrrr.rxr{r~rrrrrr/rfloatrrrzr shutdown_allrrrrrrr$rrrbytessocketrrrrrrk __classcell__)rYs@rr1r15s0!!J cc##4F+3   cc  #$L%L!&QR %&5'5 % % C cc }%GE{v %% Y#++R[NvH ** ++*c*d**::c?*45EE* S( )*2"""/"CD;;" """01 " ";.2!::c?!=>UU! !F/0L !&$) ++++ZZ ++D! ++  ++Z56O;#; 48H;UY;;'+*- 99**U#9jj' 9  99,3,,$,,2s2}2Td$/0L # $  Js JC JD J JStPT.34N  #  $    G#G$G (C (M (AJ55() 5;>5 55 AJ88() 8;>8 88  S QVVCJ5G  "  s  ajj6G  SYS`S`      s  ajj6G  SYS`S`         58I  U[UbUb      s  ajj6G  SYS`S`      C  1::e3D  PVP]P]    !aee!!rr1ceZdZedddZeddj dZejZ ejZ ejZejZy ) 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 connectingr2r4N)rrrrr7rrr}r1rrrrrrrrrrrrrsl+8  O cc  &99L'==N(??O%99Lrr)#rrrmrtypingr*rrN traitletsrrrrrrr r traitlets.config.configurabler traitlets.utils.importstringr kernelspecrrmanagerrutilsrrrurr-r/r1rrrrrs+  &=4)*" 9 QZZAJJ(\!,\!~:0:r