j,w dZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m ZddlmZddlmZddlZddlmZddlmZmZmZmZmZmZmZmZmZm Z m!Z!ddl"m#Z#d d l$m%Z%d d l&m'Z'd d l(m)Z)d d l*m+Z+d dl,m-Z-d dl.m/Z/d dl0m1Z1d dl0m2Z3GddeZ4ej5dej6dejfZ7dej8e effdZ9de7de7fdZ:Gdde-Z;Gdde;Z<e/j=e; d'd!e>d"e?d#ejdej@e;e)ffd$ZA d'd!e>d"e?d#ejdej@e>methodc tjtjdtjdtjdtjdtjffd }tjt |S)zSets the kernel to a pending state by creating a fresh Future for the KernelManager's `ready` attribute. Once the method is finished, set the Future's results. selfargskwargsr+cK|js|jst|_ |g|Ri|d{V}tjdd{V|jr|jd|S#t$rR}|j||j |j |d}~wwxYw)z)Create a future for the decorated method.Ng{Gz?) _attempted_start_readyr1r-sleep owns_kernel set_result Exception set_exceptionlog exception)r4r5r6outer2s r(wrapperz!in_pending_state..wrapperPs   (  (%--DK t5d555f55555555C-%% % % % % % % % - &&t,,,J    K % %a ( ( ( H  t{4466 7 7 7G sAA55 C?A C  C)t no_type_check functoolswrapsrcastr))r2rCs` r(in_pending_staterIJs| __VAE!%15QU_" 6!W  r'c Z eZdZUdZejeefdzed<dej dej ddffd Z e dZ e ed <e ejZe ed <ed dejfd Zed d Zeed<eed Zeed<eddefdZeddejeefddfdZedd Zejeefed<dZe dzed<e e!j"Z#e ed<edde!j"fdZ$ede%dejee fddfdZ&e'dd dZ(e'ed<ee!j)Z*ejeefed<eddejeefddfd Z+dZ,e!j-dzed!<e.de!j-dzfd"Z/e dd d#Z0e ed$<ed$de1fd%Z2e.dejeeffd&Z3e.de1fd'Z4ed Z5ej6d(ed)<e Z7e ed*<e Z8e ed+<e d d d,Z9e ed-<dZ:e1ed.<dYd/Z;dYd0Zd4eddfd5Z?dZd3ej>d4eddfd6Z@dej deAfd7ZBd8ededzfd9ZCd:ejeefddfd;ZDd[dejEed?ej ddfd@ZGeHeGZIdYdAZJdYdBZKd?ej dejLejEeejeej fffdCZMeHeMZNd?ej ddfdDZOeHeOZPeQd?ej ddfdEZReHeRZSd\d2e1ddfdFZTeHeTZU d]dHeVdzdIeVd2e1ddfdJZWeHeWZXd\d2e1ddfdKZYeHeYZZeQd^dLe1d2e1ddfdMZ[eHe[Z\ d^dLe1dNe1d?ej ddfdOZ]eHe]Z^e.de1fdPZ_e.de1fdQZ`d\d2e1ddfdRZaeHeaZbd\d2e1ddfdSZceHecZddYdTZeeHeeZfdUegddfdVZheHehZide1fdWZjeHejZkd_dIeVddfdXZlxZmS)` KernelManagerzgManages a single kernel in a subprocess on this host. This version starts kernels with Popen. Nr9r5r6r+c|rtjdtd|dd|_t jdi|tj|_ d|_ d|_ dS) zInitialize a kernel manager.zPassing positional only arguments to `KernelManager.__init__` is deprecated since jupyter_client 8.6, and will become an error on future versions. Positional arguments have been ignored since jupyter_client 7.0) stacklevelr;TFNr&) warningswarnDeprecationWarningpop _owns_kernelsuper__init__rr%_shutdown_statusr8r9)r4r5r6 __class__s r(rUzKernelManager.__init__ns   MH#     #JJ}d;;""6""" / 5 % r'F_created_contextcontextc6d|_tjSNT)rXzmqContextr4s r(_context_defaultzKernelManager._context_defaults ${}}r'z,jupyter_client.blocking.BlockingKernelClientTconfig client_classklassraclient_factoryc*t|jSN)rrbr^s r(_client_factory_defaultz%KernelManager._client_factory_defaults4,---r'changecTtt|d|_dS)Nnew)rstrrer4ris r(_client_class_changedz#KernelManager._client_class_changeds$)#fUm*<*<==r') allow_none kernel_id provisionerkernel_spec_managerc6tj|jS)N)data_dir)rKernelSpecManagerrtr^s r(_kernel_spec_manager_defaultz*KernelManager._kernel_spec_manager_defaults+T]CCCCr'cd|_dSrg) _kernel_specrms r(_kernel_spec_manager_changedz*KernelManager._kernel_spec_manager_changeds!r'@aTime to wait for a kernel to terminate before killing it, in seconds. When a shutdown request is initiated, the kernel will be immediately sent an interrupt (SIGINT), followedby a shutdown_request message, after 1/2 of `shutdown_wait_time`it will be sent a terminate (SIGTERM) request, and finally at the end of `shutdown_wait_time` will be killed (SIGKILL). terminate and kill may be equivalent on windows. Note that this value can beoverridden by the in-use kernel provisioner since shutdown times mayvary by provisioned environment.)rahelpshutdown_wait_time kernel_namecRd|_|ddkrtj|_dSdS)Nrkpython)rxrNATIVE_KERNEL_NAMEr}rms r(_kernel_name_changedz"KernelManager._kernel_name_changeds2  %=H $ $)rpython2python3r}r^s r( ipykernelzKernelManager.ipykernels#CCCr'zDict[str, Any] _launch_args_control_socket _restarterz,Should we autorestart the kernel if it dies. autorestart shutting_downcV||dSrg)_close_control_socketcleanup_connection_filer^s r(__del__zKernelManager.__del__s, ""$$$ $$&&&&&r'cdS)zStart the kernel restarter.Nr&r^s r(start_restarterzKernelManager.start_restarter r'cdS)zStop the kernel restarter.Nr&r^s r(stop_restarterzKernelManager.stop_restarterrr'restartcallbackeventcN|jdS|j||dS)z;Register a callback to be called when a kernel is restartedN)r add_callbackr4rrs r(add_restart_callbackz"KernelManager.add_restart_callbacks- ? " F $$Xu55555r'cN|jdS|j||dS)z=Unregister a callback to be called when a kernel is restartedN)rremove_callbackrs r(remove_restart_callbackz%KernelManager.remove_restart_callbacks- ? " F ''%88888r'c i}||d||j|d|||jdi|S)z3Create a client configured to connect to our kernelT)session)connection_fileparentr&)updateget_connection_inforre)r4r6kws r(clientzKernelManager.clients $**4*88999 #'#7      &"t"((R(((r'pathcH|jJ|j|S)zResolve path to given file.)rq resolve_path)r4rs r(rzKernelManager.resolve_paths(+++,,T222r'envct|jtrKd|jvrDt|jdtr&|jd|dSdSdSdS)a Allow to update the environment of a kernel manager. This will take effect only after kernel restart when the new env is passed to the new kernel. This is useful as some of the information of the current kernel reflect the state of the session that started it, and those session information (like the attach file path, or name), are mutable. .. version-added: 8.5 rN) isinstancerdictr)r4rs r( update_envzKernelManager.update_env s{ t($ / / 1***4,U3T::+  e $ + +C 0 0 0 0 0  1 1****r'extra_argumentscH|pg}|jJ|jj|z}|rD|dddtjdzdtjddzhvrtj|d<t j|jtj d|jr|jj d<t|j tsJ|j tjd d t"jd t"jffd fd |DS)z/Replace templated args (e.g. {connection_file})Nrrzpython%iz python%i.%irM)rprefix resource_dirz\{([A-Za-z0-9_]+)\}matchr+cz|d|S)z9Get the key out of ns if it's there, otherwise no change.r)getgroup)rnss r(from_nsz0KernelManager.format_kernel_cmd..from_nsZs)66%++a..%++--88 8r'c<g|]}|Sr&)sub).0argrpats r( z3KernelManager.format_kernel_cmd..^s'555#%%555r')rargvsys version_info executableosrrealpathrrrrrrrrecompilerDr)r4rcmdrrrs @@@r(format_kernel_cmdzKernelManager.format_kernel_cmd5sR)/R+++#o5  $3q6  )!, , C,RaR0 0   ^CF "w//0DEEj" "    ?!%!1!>B~ $+T22222 $#$$$j/00 915 9QU 9 9 9 9 9 9655555555r' kernel_cmdrcK|jJ|jj|fi|d{V}|jjsJ||dS)zactually launch the kernel override in a subclass to launch kernel subprocesses differently Note that provisioners can now be used to customize kernel environments and N)rq launch_kernel has_process_reconcile_connection_info)r4rrconnection_infos r(_async_launch_kernelz"KernelManager._async_launch_kernel`st+++ > 0 >z P PR P PPPPPPP++++ ''88888r'cd|j(|d|_d|j_dSdS)Ncontrold)r_create_connected_socketlingerr^s r(_connect_control_socketz%KernelManager._connect_control_socketrs:   '#'#@#@#K#KD *-D ' ' ' ( 'r'cX|jdS|jd|_dSrg)rcloser^s r(rz#KernelManager._close_control_socketws3   ' F ""$$$#r'cKd|_|jp3|dtt j|_||_|j?tj |j  |j|j ||_|jjdi|d{V}|d}||fS)a{Prepares a kernel for startup in a separate process. If random ports (port=0) are being used, this method must be called before the channels are created. Parameters ---------- `**kw` : optional keyword arguments that are passed down to build the kernel_cmd and launching the kernel (e.g. Popen kwargs). FrpN)rrr&)rrprRrluuiduuid4copyrrqKPFinstancercreate_provisioner_instancer pre_launchr4rrs r(_async_pre_start_kernelz%KernelManager._async_pre_start_kernel}s#Q266+s4:<>8>TTH 4"  l ;;XPQ\        # 4 4 4 HNNM N N N$3$BD !1133 3 3 3 3 3 3 3 3 3 4 ."  l ;;XPQ\        .'333&++----------- . . # ; ; ; HNNI J J J$3$BD !))'):: : : : : : : : : : : ;s%3A11ACC 3D22AF  F c:K|s|||d|j_|jr|s|jd|jr#|j |d{VdSdS)z/Clean up resources when the kernel is shut downNr)rr) rcleanup_ipc_filesrrrrXrYdestroyrqcleanupr4rs r(_async_cleanup_resourcesz&KernelManager._async_cleanup_resourcess +  ( ( * * *     ""$$$"    - - L  , , ,   <"**7*;; ; ; ; ; ; ; ; ; ; < > > > > > > > >++G+<<<<<<<<<<Cannot restart the kernel. No previous call to 'start_kernel'.T)rrr&)rr/rcleanup_random_portsrr)r4rrrrs r(_async_restart_kernelz#KernelManager._async_restart_kernel3s6   $RCs## #))c4)@@@@@@@@@  (  % % ' ' '   $$$&d&;;):;;;;;;;;;;;r'c|jSrg)rSr^s r(r;zKernelManager.owns_kernel^s   r'c,|jduo |jjS)z@Has a kernel process been started that we are actively managing.N)rqrr^s r(r zKernelManager.has_kernelbst+L0@0LLr'cpK|jr,|jJ|j|d{VdSdS)zIsimilar to _kill_kernel, but with sigterm (not sigkill), but do not blockNr)r rq terminaters r(rz(KernelManager._async_send_kernel_sigtermgs[ ? >#///",,W,== = = = = = = = = = > >r'c~K|jr|jJ|j|d{V tj|dd{V|jr!|jd{VdSdS#tj$r|j dYdSwxYwdS)zrKill the running kernel. This is a private method, callers should use shutdown_kernel(now=True). Nrrzrz>Wait for final termination of kernel timed out - continuing...) r rqkillr-rrr rr?warningrs r(r z KernelManager._async_kill_kernelos ? 2#///"'''88 8 8 8 8 8 8 8 2&t'7'7'9'93GGGGGGGGGG?2*//1111111111122 '     !abbb  2 2s.B )B:9B:c~K|js|jzt|jtrBt jt jtt j |j}n|j}|j r|d{V|d{V|jr|j J|j j }|dkr'| tjd{VdS|dkrR|jdi}||j|j|dSdSd}t+|)zInterrupts the kernel by sending it a signal. Unlike ``signal_kernel``, this operation is well supported on all platforms. Nsignalmessageinterrupt_requestrz.Cannot interrupt kernel. No kernel is running!)r r9rr0r- ensure_futurerDrHrrrrinterrupt_mode_async_signal_kernelr#SIGINTrrrrrr/)r4rr'rs r(rz%KernelManager._async_interrupt_kernels[  4;#:$+w// $-afVAE]DK.P.PQQ !  KKKKKKK ? $#///!->>>>>>>>>>9,,l&&':B&GG,,... !!$"6<<<<<-, CCs## #r'signumcK|jr+|jJ|j|d{VdSd}t|)aSends a signal to the process group of the kernel (this usually includes the kernel and any subprocesses spawned by the kernel). Note that since only SIGTERM is supported on Windows, this function is only useful on Unix systems. Nz+Cannot signal kernel. No kernel is running!)r rq send_signalr/)r4r*rs r(r(z"KernelManager._async_signal_kernelsc ? $#///"..v66 6 6 6 6 6 6 6 6 6?Cs## #r'cK|jsdS|jr,|jJ|jd{V}|dSdS)z$Is the kernel process still running?TNF)r;r rqpoll)r4rets r(_async_is_alivezKernelManager._async_is_alivesd 4 ? #///(--////////C{tur'cK|d{Vr6tj|d{V|d{V4dSdSrg)r0r-r:)r4rs r(rzKernelManager._async_waits ((******** .- -- - - - - - - -((******** . . . . .r')r+Nrrg)F)NrF)FF)r)nr!r"r#r$rDUnionrr0__annotations__rrUrrXr r\r]rYrr_r rbr rrerhrr rlrnrrprqrrrurrrvrryr r|rr}rrx KernelSpecpropertyrrboolrrrrOptionalrrrrrrrCallablerrrrrrListrrr_launch_kernelrrTuplerpre_start_kernelrpost_start_kernelrIr start_kernelrrequest_shutdownfloatr finish_shutdownrcleanup_resourcesrshutdown_kernelrrestart_kernelr;r r_send_kernel_sigtermr  _kill_kernelrinterrupt_kernelintr( signal_kernelr0is_aliver __classcell__rWs@r(rKrKfsU  GFGO $t ++++aequ""T%[[d(((!--GX--- WY#+ &6%56t&&&L" 4l4@@@ND@@@ W ..... W^>AF38H3H,I>d>>>>(/wt'E'E'EIqwsG|$EEE 15K&-444$,HZ-I$J$JJJJ W "##Dj.JDDD$#D W "##!16#x-3H!T!!!^$#!!& + ! ! !   *11N)O)OKg&OOO W]=16#s(+;===== 26L*'$.555 !Z2T9!!!X!   `K  W]'d''''qww/X D4DDDXD261F1F1FL!*-.FFF355OS   ceeJ T RK M4''''        66QZ66TX6666 99 939W[9999 )qu ))= ) ) ) )&33t3333 1S!11d1111*)6)6t1C)6qvVY{)6)6)6)6V 9QVC[ 9 9RV 9 9 9 9X233N.... $$$$ E afS!%Z00 1: x 788 115 1T 1 1 1 1!!9::2ae22222(8/00L @ @T @d @ @ @ @ x 788"&! %.%.$,%.%. %.  %.%.%.%.Nh566O < >>>>>> $8$>??2222222*8.//L$$$$> x 788 $ $ $ $ $ $H122M t    x((H..e.d........r'rKceZdZUdZeddZeed<eddZeed<e e j j Z e ed<edd e j j fd Zd ejd effd ZejZejZejd ejfed<ejZejd ejfed<ejZejd ejfed<ejZ ejd ejfed<ej!Z"ejd ejfed<ej#Z$ejd ejfed<ej%Z&ejd ejfed<ej'Z(ejd ejfed<ej)Z*ej+Z,ej-Z.ejd ejfed<ej/Z0ejd ejfed<ej1Z2ejd ejfed<xZ3S)AsyncKernelManagerzAn async kernel manager.z-jupyter_client.asynchronous.AsyncKernelClientTr`rbrcrerYr+cLd|_tjSr[)rXr\r-r]r^s r(r_z#AsyncKernelManager._context_defaults ${""$$$r'r6c 6tjdi|S)zGet a client for the manager.r&)rTr)r4r6rWs r(rzAsyncKernelManager.clients!uww~'''''r'.r>r<r=r?rArBrCrDrGrIrJ)4r!r"r#r$r rbr3r rer r\r-r]rYrr_rDrrrrKrr:rr>r8 Awaitablerr<rr=rr?r rArrBrrCrrDrrEr rFrrGr(rIr0rJrKrLs@r(rNrNs""&6%57&&&L" 4&U^bcccNDccc!!455GX555 WY%#+"5%%%%(( (((((( #7N1>1RL!*S!+-.RRR5B5Zajak!12ZZZ6C6\qz#q{"23\\\5B5Zajak!12ZZZ4A4XOQZQ[ 01XXX6C6\qz#q{"23\\\4A4XOQZQ[ 01XXX3@3VNAJsAK/0VVV(C 3L5B5Zajak!12ZZZ2?2TM1:c1;./TTT-:-JHajak)*JJJJJJJr'rN<rstartup_timeoutr}r6c 2t|}|jdi||}| ||n7#t $r*||wxYw||fS)5Start a new kernel, and return its Manager and Clientrrr&)rKr>rstart_channelswait_for_readyr/ stop_channelsrCrSr}r6kmkcs r(start_new_kernelr\s ; / / /BBOf B /2222     r6Ms A4BcZKt|}|jdi|d{V|}| ||d{Vn=#t $r0||d{VwxYw||fS)rUrNrr&)rNr>rrVrWr/rXrCrYs r(start_new_async_kernelr^ s  4 4 4B "/ # #F # ######## B8888888888    """""""""  8Os A,,:B&c+Ktdi|\}} |V||ddS#||dwxYw)zContext manager to create a kernel in a subprocess. The kernel is shut down when the context exits. Returns ------- kernel_client: connected KernelClient instance T)rNr&)r\rXrC)r6rZr[s r( run_kernelr`s ' ' ' 'FB%  t$$$$$  t$$$$s A,A/)rRr)Er$r-rFrrr#rtypingrDrrOasyncio.futuresrconcurrent.futuresr0 contextlibrenumrr\jupyter_core.utilsr traitletsrrr r r r r rrrrtraitlets.utils.importstringrrr asynchronousrblockingrrrconnectr managerabcr provisioningrrrrTypeVarr8r)r2r1rIrKrNregisterr@rlr;r\r^Iteratorr`r&r'r(rqs++  """"""000000%%%%%% ''''''                          544444++++++****** ((((((((((((//////999999 & & & & &d & & &AIcCJ/000QWVW_-Q18f .f .f .f .f .'f .f .f .R$K$K$K$K$K$K$K$KN-(((5=.1HIW]0 01&5=.1HIW !2 23$%%1:l#;%%%%%%r'