o tf1@sdZddlZddlZddlZddlZddlZddlZddlmZm Z Gddde Z GdddZ Gd d d e Z Gd d d e ZGd ddZdS)z( Simple asynchronous process supervisor N) add_handlerremove_handlerc@seZdZdZdS)KilledProcessErrorz Raised when a process that has been explicitly killed is started again. Each SupervisedProcess can be killed only once. N)__name__ __module__ __qualname____doc__r r \/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/simpervisor/process.pyrsrc@s\eZdZdZdZdZdZddZddZddZ d d Z d d Z e d dZ e ddZdS)Processzb Abstract class to start, wait and send signals to running processes in a OS agnostic way NcOs||_||_dSN) _proc_cmd _proc_kwargs)selfcmdkwargsr r r __init__ s zProcess.__init__ct)z# Start the process NotImplementedErrorrr r r start$z Process.startcr)zU Wait for the process to terminate and return the process exit code. rrr r r wait*rz Process.waitcCst)zF Returns the preferred OS signal to kill the process. rrr r r get_kill_signal0szProcess.get_kill_signalcCs|jr |j|dSdS)z4 Send the OS signal to the process. N)_proc send_signalrsignumr r r r6szProcess.send_signalcC|jr|jjSdSr )rpidrr r r r!=z Process.pidcCr r )r returncoderr r r r#Br"zProcess.returncode)rrrr rrrrrrrrpropertyr!r#r r r r r s r c@(eZdZdZddZddZddZdS) POSIXProcesszG A process that uses asyncio-subprocess API to start and wait. cs"tj|ji|jIdH|_dS)zL Start the process using asyncio.create_subprocess_exec API N)asynciocreate_subprocess_execrrrrr r r rMs zPOSIXProcess.startcs|jIdHS)zP Wait for the process to stop and return the process exit code. N)rrrr r r rUszPOSIXProcess.waitcCtjS)zH Returns the OS signal used for kill the child process. )signalSIGKILLrr r r rzPOSIXProcess.get_kill_signalNrrrr rrrr r r r r&Hs  r&c@r%) WindowsProcesszS A process that uses subprocess API to start and wait (uses busy polling). cs"tjt|jfi|j|_dS)z? Starts the process using subprocess.Popen API N) subprocessPopenlistrrrrr r r rgs zWindowsProcess.startcs8|jdurtdIdH|jdus|jS)a3 Wait for the process to stop and return the process exit code. subprocess.Popen.wait() is a blocking call which can cause the asyncio event loop to remain blocked until the child process is terminated. To circumvent this behavior, we use busy polling with asyncio.sleep to check whether the child process is alive or not and keeping the asyncio event loop running. See https://github.com/jupyter/jupyter_client/blob/main/jupyter_client/provisioning/local_provisioner.py#L54_L55 for similar use. Ng?)rpollr'sleeprrr r r rms   zWindowsProcess.waitcCr))a Returns the OS signal used for kill the child process. Windows doesn't support SIGKILL. subprocess.Popen.kill() is an alias of subprocess.Popen.terminate(), so we can use SIGTERM instead of SIGKILL on windows platform. )r*SIGTERMrr r r r~szWindowsProcess.get_kill_signalNr,r r r r r-bs  r-c@szeZdZdddddddZdddZd d Zd d Zd dZddZddZ ddZ ddZ e ddZ e ddZdS)SupervisedProcessFN)always_restart ready_func ready_timeoutlogcOs`||_||_||_||_||_||_d|_|dur td|_ n||_ d|_ d|_ t |_dS)NZ simpervisorF)r6name _proc_argsrr7r8proclogging getLoggerr9running_killedr'Lock _proc_lock)rr:r6r7r8r9argsrr r r rs zSupervisedProcess.__init__cGs<||j|j|jd}|r|||jj|j||ddS)zn Log debug message with some added meta information. Makes structured logging easier )actionz proccess-namez process-argszprocess-kwargs)extraN)r:r;rupdater9debugformat)rrDmessageextrasrCZ base_extrasr r r _debug_logs zSupervisedProcess._debug_logcCs*|j|d|_|ddi||jdS)NTr*zPropagated signal {} to {})r<rr@rKr:)rr*r r r _handle_signals z SupervisedProcess._handle_signalc s|j4IdHn|jr WdIdHdS|jr$td|jd|ddi|jtjdkr=t|j i|j |_ n t |j i|j |_ |j IdH|ddi|jd |_d |_t||_t|jWdIdHdS1IdHs|wYdS) z Start the process if it isn't already running. If the process is already running, this is a noop. If the process has already been killed, this raises an exception NProcess # has already been explicitly killedz try-startzTrying to start {}win32startedz Started {}FT)rBr?r@rr:rKsysplatformr-r;rr<r&rr' ensure_future_restart_process_if_needed_restart_process_futurerrLrr r r rs,    .zSupervisedProcess.startcsh|jIdH}t|j|ddd|i|j|d|_|js0|js'|dkr2| IdHdSdSdS)z Watch for process to exit & restart it if needed. This is a long running task that keeps running until the process exits. If we restart the process, `start()` sets this up again. NZexitedz{} exited with code {}codeFr) r<rrrLrKr:r?r@r6r)rretcoder r r rTs z,SupervisedProcess._restart_process_if_neededc s~|j4IdH*|j|d|_|j|jIdHd|_t|j WdIdHdS1IdHs8wYdS)a Send a SIGTERM or SIGKILL to the child process & reap it. - Send the signal to the process - Make sure we don't restart it when it stops - Wait for it to stop - Remove signal handler for it after we are done. NTF) rBr<rr@rUcancelrr?rrLrr r r _signal_and_waits   .z"SupervisedProcess._signal_and_waitcs,|jr td|jd|tjIdHS)zz Send SIGTERM to process & reap it. Might take a while if the process catches & ignores SIGTERM. rMrNN)r@rr:rYr*r3rr r r terminates  zSupervisedProcess.terminatecs4|jr td|jd|j}||IdHS)z3 Send SIGKILL to process & reap it rMrNN)r@rr:r<rrYrr r r kill*s  zSupervisedProcess.killcst}d} t||jkrdS|js|jsdSzt||dIdH}Wn tjy5d}Ynwt|}|dd|||d||||rNdSt |IdHd |}t|||jkrp||jtd}q) z4 Wait for process to become 'ready' g{Gz?TFr5Nz ready-waitz1Readyness: {} after {} seconds, next check in {}s) wait_timereadyZ elapsed_time) timer8r@r<r'wait_forr7 TimeoutErrorrKr2)r start_timer\Zis_readyZcur_timer r r r]5s:   zSupervisedProcess.readycC|jjSr )r<r!rr r r r!ozSupervisedProcess.pidcCrcr )r<r#rr r r r#srdzSupervisedProcess.returncoder )rrrrrKrLrrTrYrZr[r]r$r!r#r r r r r4s$ $ +  : r4)r r'r=r*r.rQr_Z atexitasyncrr Exceptionrr r&r-r4r r r r s2'