Mhrv 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..wrapperOs   (  (%--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_staterIIs| __VAE!%15QU_" 6!W  r'c eZdZUdZejejeefe d<dej dej ddffd Z e dZ e e d <eejZee d <ed dejfd Zed Zee d <eeZee d<eddefdZed dejeefddfdZeddZejeefe d<dZ eje!e d<ee"j#Z$ee d<edde"j#fdZ%ede&dejeefddfdZ'e(dddZ)e(e d<ee"j*Z+ejeefe d<eddejeefddfdZ,dZ-eje"j.e d <e/deje"j.fd!Z0e ddd"Z1e e d#<ed#de2fd$Z3e/dejeeffd%Z4e/de2fd&Z5edZ6ejd'e d(<e Z7e e d)<e Z8e e d*<e ddd+Z9e e d,<dZ:e2e d-<dVd.Z;dVd/Zd3eddfd4Z?dWd2ej>d3eddfd5Z@dej deAfd6ZBd7ejeefddfd8ZCdXd9ejejDedejDefd:ZEd;ejDedZIdVd?ZJdr~python2python3r|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||dSre)_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_inforrc)r4r6kws r(clientzKernelManager.clients $**4*88999 #'#7      &"t"((R(((r'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_envs{ 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})Nrr~zpython%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_nsTs)66%++a..%++--88 8r'c<g|]}|Sr&)sub).0argrpats r( z3KernelManager.format_kernel_cmd..Xs'555#%%555r')rargvsys version_info executableospathrealpathrrrrrrrrecompilerDr)r4rcmdrrrs @@@r(format_kernel_cmdzKernelManager.format_kernel_cmd/sR)/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)ro launch_kernel has_process_reconcile_connection_info)r4rrconnection_infos r(_async_launch_kernelz"KernelManager._async_launch_kernelZst+++ > 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_socketls:   '#'#@#@#K#KD *-D ' ' ' ( 'r'cX|jdS|jd|_dSre)rcloser^s r(rz#KernelManager._close_control_socketqs3   ' 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). FrnN)rrr&)rrnrRrjuuiduuid4copyrroKPFinstancercreate_provisioner_instancer pre_launchr4rrs r(_async_pre_start_kernelz%KernelManager._async_pre_start_kernelws#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_filesrrrrXrYdestroyrocleanupr4rs 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_kernel-s6   $RCs## #))c4)@@@@@@@@@  (  % % ' ' '   $$$&d&;;):;;;;;;;;;;;r'c|jSre)rSr^s r(r;zKernelManager.owns_kernelXs   r'c,|jduo |jjS)z@Has a kernel process been started that we are actively managing.N)rorr^s r(rzKernelManager.has_kernel\st+L0@0LLr'cpK|jr,|jJ|j|d{VdSdS)zIsimilar to _kill_kernel, but with sigterm (not sigkill), but do not blockNr)rro terminaters r(rz(KernelManager._async_send_kernel_sigtermas[ ? >#///",,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). Nrrxrz>Wait for final termination of kernel timed out - continuing...) rrokillr-rrrrr?warningrs r(rz KernelManager._async_kill_kernelis ? 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!)rr9rr0r- 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!)rro 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;rropoll)r4rets r(_async_is_alivezKernelManager._async_is_alivesd 4 ? #///(--////////C{tur'cK|d{Vr6tj|d{V|d{V4dSdSre)r-r-r:)r4rs r(rzKernelManager._async_waits ((******** .- -- - - - - - - -((******** . . . . .r')r+Nrre)F)NrF)FF)r)mr!r"r#r$rDOptionalUnionrr0__annotations__rrUrrXr r\r]rYrr_r r`r rrcrfrr rjrlrrnrorrrsrprtrrwr r{rr|rrv KernelSpecpropertyrrboolrrrrrrrrrrrCallablerrrrrListrrr_launch_kernelrrTuplerpre_start_kernelrpost_start_kernelrIr start_kernelrrequest_shutdownfloatr finish_shutdownrcleanup_resourcesrshutdown_kernelrrestart_kernelr;rr_send_kernel_sigtermr _kill_kernelrinterrupt_kernelintr% signal_kernelr-is_aliver __classcell__rWs@r(rKrKes6  Jqwvw/ 0000aequ""T%[[d(((!--GX--- WY#+ &6%56&&L" 4l333ND333 W ..... W^>AF38H3H,I>d>>>>(/wt'E'E'EIqwsG|$EEE 6:K12999$,HZ-I$J$JJJJ W "##Dj.JDDD$#D W "##!16#x-3H!T!!!^$#!!& + ! ! !   *11N)O)OKg&OOO W]=16#s(+;===== 7;L!*Z23::: !QZ (=>!!!X!   `K  W]'d''''qww/X D4DDDXD261F1F1FL!*-.FFF355OS   ceeJ T RK M4''''        66QZ66TX6666 99 939W[9999 )qu ))= ) ) ) )&1S!11d1111*)6)6AF3K1H)6TUTZ[^T_)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'+! %.%.*U#%.%. %.  %.%.%.%.Nh566O < >>>>>> $8$>??2222222*8.//L$$$$> x 788 $ $ $ $ $ $H122M t    x((H..e.d........r'rKceZdZUdZedZeed<edZeed<e e j j Z e ed<edde j j fdZd ejdeffd 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.AsyncKernelClientr`rarcrYr+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;r9r:r<r>r?r@rArDrFrG)4r!r"r#r$r r`r1r rcr r\r-r]rYrr_rDrrrrKrr7rr;r5 Awaitablerr9rr:rr<r r>rr?rr@rrArrBrrCrrDr%rFr-rGrHrIs@r(rKrKsy""&6%57&&L" 4&UVVVNDVVV!!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'rK<r~startup_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_channelsr@rPr|r6kmkcs r(start_new_kernelrYs ; / / /BBOf B /2222     r6Ms A4BcZKt|}|jdi|d{V|}| ||d{Vn=#t $r0||d{VwxYw||fS)rRrNrr&)rKr;rrSrTr/rUr@rVs 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&)rYrUr@)r6rWrXs r( run_kernelr]s ' ' ' 'FB%  t$$$$$  t$$$$s A,A/)rOr~)Er$r-rFrrr rtypingrDrrOasyncio.futuresrconcurrent.futuresr0 contextlibrenumrr\jupyter_core.utilsr traitletsrrr r r r r rrrrtraitlets.utils.importstringrrr asynchronousrblockingrrrconnectr managerabcr provisioningrrrrTypeVarr5r)r0r1rIrKrKregisterr=rjr8rYr[Iteratorr]r&r'r(rns++  """"""000000%%%%%% ''''''                          544444++++++****** ((((((((((((//////999999 & & & & &d & & &AIcCJ/000QWVW_-Q18a .a .a .a .a .'a .a .a .H$K$K$K$K$K$K$K$KN-(((5=.1HIW]0 01&5=.1HIW !2 23$%%1:l#;%%%%%%r'