:vjd2dZddlZddlZddlZddlZddlZddlmZmZddl Z ddl m Z m Z ddl mZddlmZmZdd lmZGd d eZy) zKernel Provisioner ClassesN) TYPE_CHECKINGAny)KernelConnectionInfoLocalPortCache) launch_kernel) is_local_ip local_ips)KernelProvisionerBasecneZdZdZdZdZdZdZdZdZ dZ e de fdZ dedzfdZdedzfdZdeddfd Zdd e ddfd Zdd e ddfd Zed eddfdZdd e ddfdZdedeeefffd ZdeededefdZdededzfdZedeeefdeeeffdZ deffd Z!deddffd Z"xZ#S)LocalProvisionera0 :class:`LocalProvisioner` is a concrete class of ABC :py:class:`KernelProvisionerBase` and is the out-of-box default implementation used when no kernel provisioner is specified in the kernel specification (``kernel.json``). It provides functional parity to existing applications by launching the kernel locally and using :class:`subprocess.Popen` to manage its lifecycle. This class is intended to be subclassed for customizing local kernel environments and serve as a reference implementation for other custom provisioners. NFreturnc|jduS)N)process)selfs ~/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/jupyter_client/provisioning/local_provisioner.py has_processzLocalProvisioner.has_process(s||4''c^Kd}|jr|jj}|Sw)zPoll the provisioner.r)rpoll)rrets rrzLocalProvisioner.poll,s) <<,,##%C s+-ctKd}|jr|jd{6tjdd{|jd{6|jj }dD]+}t |j|}|s|j -d|_|S77r7\w)z!Wait for the provisioner process.rNg?)stdoutstderrstdin)rrasynciosleepwaitgetattrclose)rrattrfids rrzLocalProvisioner.wait3s <<  #+mmC((( #+,,##%C5dllD1IIK6 DL $($s?"B8B2B8B4B8B6B87B8B84B86B8signumcK|jr|tjk(r6tjdk(r#ddlm}||jjy|jr2ttdr" tj|j|y|jj|yy#t$rY(wxYww)aTSends 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, we will check if the desired signal is for interrupt and apply the applicable code on Windows in that case. win32r)send_interruptNkillpg)rsignalSIGINTsysplatform win_interruptr'win32_interrupt_eventpgidhasattrosr(OSError send_signal)rr$r's rr3zLocalProvisioner.send_signalHs <<&3<<7+B:t||AAByyWR2IIdii0 LL $ $V , # s*A2C5 B3C3 B?<C>B??CrestartcFK|jrUttdr) |jtjd{y |jj yy7"#t $rY*wxYw#t $r}tj|Yd}~yd}~wwxYww)z,Kill the provisioner and optionally restart.SIGKILLN) rr0r)r3r6r2killr_tolerate_no_processrr4es rr7zLocalProvisioner.killds <<vy)**6>>::: 9 !!# ; 9 55a88 9dB!"A'A%A'B! A6#B!%A'' A30B!2A33B!6 B?BB!BB!cFK|jrUttdr) |jtjd{y |jj yy7"#t $rY*wxYw#t $r}tj|Yd}~yd}~wwxYww)z1Terminate the provisioner and optionally restart.SIGTERMN) rr0r)r3r=r2 terminaterr8r9s rr>zLocalProvisioner.terminatess <<vy)**6>>::: 9 &&( ; 9 55a88 9r;os_errorctjdk(r |jdk7rd|}t|yddlm}t |tr|j|k7rd|}t|y)Nr&z3Invalid Error, expecting error number to be 5, got r)ESRCHz:Invalid Error, expecting ProcessLookupError or ESRCH, got )r+r,winerror ValueErrorerrnorB isinstanceProcessLookupError)r? err_messagerBs rr8z%LocalProvisioner._tolerate_no_processs} <<7 "  A% ST\S]^  --& $h(:;x~~QV?VPQYPZ[!-- @WrcDK|jr|stj}|jd|jd|jd|jd|jdf}|D]+}trt |t sJ|j|-yyyw)zFClean up the resources used by the provisioner and optionally restart. shell_port iopub_port stdin_porthb_port control_portN) ports_cachedrinstanceconnection_inforrFint return_port)rr4lpcportsports rcleanupzLocalProvisioner.cleanups   W ))+C$$\2$$\2$$\2$$Y/$$^4 E %dC000%&- sBB kwargscK|j}|ry|jdt|dd}t|dr|j |nt |rdnd}|dk(}|dv}|j dv}d}d} |r|s d } t| |j d k(r:t|js%d |jd t} t| |jd g} |jr|jstj} | j|j|_| j|j|_| j|j|_| j|j|_| j|j|_d|_ |rp|rn|xst|dxr|j+} | rJ|j,&t/j0\}} ||_| |_n|j,}|j2} d|vr(|dj5dd}|j7|n|j7|j9|_|j=| }n+|jd g} |j>j@| z}tC|dd|i|d{S7w)aPerform any steps in preparation for kernel process launch. This includes applying additional substitutions to the kernel launch command and env. It also includes preparation of launch parameters. Returns the updated kwargs. transport_encryptiondisabled_transport_encryption_policyautorequired>r]r^>ipctcpNzStransport_encryption='required' is only supported when transport is 'tcp' or 'ipc'.r`z@Can only launch a kernel on a local interface. This one is not: zc.Make sure that the '*_address' attributes are configured properly. Currently valid addresses are: extra_argumentsT!_kernel_supports_curve_encryptionenvJPY_SESSION_NAME)jupyter_session)racmd)#parentpopr r0r\bool transport RuntimeErrorr ipr cache_portsrOrrPfind_available_portrJrKrLrMrNrbcurve_publickeyzmq curve_keypaircurve_secretkeygetwrite_connection_fileget_connection_inforQformat_kernel_cmd kernel_specargvsuper pre_launch)rrXkmrZtransport_encryption_policyencryption_requiredencryption_enabledencryption_supported_transportrqrtmsgrarTkernel_curve_okrf kernel_cmd __class__s rr|zLocalProvisioner.pre_launchs[[ #)::&4JJ(W$ 2=>//0DE $%9 :f  ( #>"K !<@T!T .0\\^-K *,0O,0O"+I3#3''||u$[-?((*w/67@k] D#3''$jj):B?O~~d&7&7$--/ # 7 7 > # 7 7 > # 7 7 >  44RUU; "%"9"9"%%"@$(!!&D"5#B CD?<<> #))1;>;L;L;N8-<*-<* +-*<*<*,*<*<"(-"3"34F"K(((I((*#%#9#9#;D -- /.J%jj):B?O))..@JW'AJA&AAAAsKK K KrgcKtj|}t|fi||_d}t t dr* t j |jj}|jj|_||_ |jdtjj|_ |jS#t$rYlwxYww)zLaunch a kernel with a command.Ngetpgidcwd)r _scrub_kwargsrrr0r1rpidr2r/rupathlibPathrrQ)rrgrXscrubbed_kwargsr/s rrzLocalProvisioner.launch_kernels+88@$Ss <:$<)F)$__name__ __module__ __qualname____doc__r _exit_futurerr/rnrOrpropertyrkrrRrrr3r7r> staticmethodr2r8rWrdictstrr|listrrrrrr __classcell__)rs@rrrs GL C D BL C (T((C$JC$J*8 9$ 94 9 9t 9 9.w.4..&&T&d&"VBVBc3hVBp$tCy$C$DX$"SS4Zd38nc3h D )D)T))rr)rrr1rr)r+typingrrrrconnectrrlauncherrlocalinterfacesr r provisioner_baser rrhrrrs:   % :$43Y),Y)r