o #b&@sdZddlZddlZddlZddlZddlZddlZddlmZddl m Z e de Gddde ZGd d d eeZGd d d eeZGd ddeeZGdddZdddZddZddZdS)z Daemon runner library. Npidfile) DaemonContextu@The ‘runner’ module is not a supported API for this library.c@eZdZdZdS)DaemonRunnerErrorz3 Abstract base class for errors from DaemonRunner. N__name__ __module__ __qualname____doc__r r 1usr/lib/python3.10/site-packages/daemon/runner.pyrrc@r)DaemonRunnerInvalidActionErrorz; Raised when specified action for DaemonRunner is invalid. Nrr r r rrrrc@r)DaemonRunnerStartFailureErrorz, Raised when failure starting DaemonRunner. Nrr r r rr#rrc@r)DaemonRunnerStopFailureErrorz, Raised when failure stopping DaemonRunner. Nrr r r rr'rrc@sreZdZdZdZddZddZddZdd d Zd d Z ddZ ddZ ddZ e e e dZ ddZddZd S) DaemonRunnera3 Controller for a callable running in a separate background process. The first command-line argument is the action to take: * 'start': Become a daemon and call `app.run()`. * 'stop': Exit the daemon process specified in the PID file. * 'restart': Stop, then start. zstarted with pid {pid:d}cCsN|||_t|_||d|_|jdur t|j|j|_|j|j_dS)a" Set up the parameters of a new runner. :param app: The application instance; see below. :return: ``None``. The `app` argument must have the following attributes: * `stdin_path`, `stdout_path`, `stderr_path`: Filesystem paths to open and replace the existing `sys.stdin`, `sys.stdout`, `sys.stderr`. * `pidfile_path`: Absolute filesystem path to a file that will be used as the PID file for the daemon. If ``None``, no PID file will be used. * `pidfile_timeout`: Used as the default acquisition timeout value supplied to the runner's PID lock file. * `run`: Callable that will be invoked when the daemon is started. N) parse_argsapprdaemon_context#_open_streams_from_app_stream_pathsrZ pidfile_pathmake_pidlockfileZpidfile_timeoutselfrr r r__init__7s  zDaemonRunner.__init__cCs8t|jd|j_t|jd|j_t|jddd|j_dS)ab Open the `daemon_context` streams from the paths specified. :param app: The application instance. Open the `daemon_context` standard streams (`stdin`, `stdout`, `stderr`) as stream objects of the appropriate types, from each of the corresponding filesystem paths from the `app`. rtzw+tr) bufferingN)openZ stdin_pathrstdinZ stdout_pathstdoutZ stderr_pathstderrrr r rrXs  z0DaemonRunner._open_streams_from_app_stream_pathscCsHtj|d}d}d|j}dj||d}t|t |dS)z Emit a usage message, then exit. :param argv: The command-line arguments used to invoke the program, as a sequence of strings. :return: ``None``. r|zusage: {progname} {usage})prognameusageN) ospathbasenamejoin action_funcskeysformat emit_messagesysexit)rargvr$Zusage_exit_code action_usagemessager r r _usage_exitgszDaemonRunner._usage_exitNcCsT|durtj}d}t||kr||t|d|_|j|jvr(||dSdS)a Parse command-line arguments. :param argv: The command-line arguments used to invoke the program, as a sequence of strings. :return: ``None``. The parser expects the first argument as the program name, the second argument as the action to perform. If the parser fails to parse the arguments, emit a usage message and exit the program. Nr"r)r.r0lenr3stractionr*)rr0Zmin_argsr r rrvs   zDaemonRunner.parse_argsc Cst|jr |jz|jWntjy*}z tdj|jd}||d}~wwt }|j j|d}t ||j dS)z Open the daemon context and run the application. :return: ``None``. :raises DaemonRunnerStartFailureError: If the PID file cannot be locked by this process. z(PID file {pidfile.path!r} already lockedrN)pid)is_pidfile_staler break_lockrrlockfileZ AlreadyLockedrr,r&getpid start_messager-rrun)rexcerrorr7r2r r r_starts"  zDaemonRunner._startc CsP|j}z t|tjWdSty'}z tdj||d}||d}~ww)z Terminate the daemon process specified in the current PID file. :return: ``None``. :raises DaemonRunnerStopFailureError: If terminating the daemon fails with an OS error. z"Failed to terminate {pid:d}: {exc})r7r>N) rread_pidr&killsignalSIGTERMOSErrorrr,)rr7r>r?r r r_terminate_daemon_processs z&DaemonRunner._terminate_daemon_processcCsD|jstdj|jd}|t|jr|jdS|dS)z Exit the daemon process specified in the current PID file. :return: ``None``. :raises DaemonRunnerStopFailureError: If the PID file is not already locked. z$PID file {pidfile.path!r} not lockedrN)rZ is_lockedrr,r8r9rF)rr?r r r_stops   zDaemonRunner._stopcCs||dS)z Stop, then start. N)rGr@)rr r r_restarts zDaemonRunner._restart)startstopZrestartcCs8z |j|j}W|Stytdj|jd}|w)ad Get the function for the specified action. :return: The function object corresponding to the specified action. :raises DaemonRunnerInvalidActionError: if the action is unknown. The action is specified by the `action` attribute, which is set during `parse_args`. zUnknown action: {action!r})r6)r*r6KeyErrorrr,)rfuncr?r r r_get_action_funcs  zDaemonRunner._get_action_funccCs|}||dS)z Perform the requested action. :return: ``None``. The action is specified by the `action` attribute, which is set during `parse_args`. N)rM)rrLr r r do_actions zDaemonRunner.do_actionN)r r r r r<rrr3rr@rFrGrHr*rMrNr r r rr+s" !  rcCs,|durtj}|dj|d|dS)z@ Emit a message to the specified stream (default `sys.stderr`). Nz {message} )r2)r.r!writer,flush)r2streamr r rr-s r-cCsNt|tstdj|d}|tj|stdj|d}|t||}|S)z= Make a PIDLockFile instance with the given filesystem path. zNot a filesystem path: {path!r})r'zNot an absolute path: {path!r}) isinstancer5 ValueErrorr,r&r'isabsrZTimeoutPIDLockFile)r'Zacquire_timeoutr?r:r r rrs   rcCsFd}|}|dur!z t|tjW|Sty d}Y|Sw|S)u Determine whether a PID file is stale. :return: ``True`` iff the PID file is stale; otherwise ``False``. The PID file is “stale” if its contents are valid but do not match the PID of a currently-running process. FNT)rAr&rBrCSIG_DFLProcessLookupError)rresultZ pidfile_pidr r rr8 s r8rO)r errnor&rCr.warningsr:rdaemonrwarnDeprecationWarning ExceptionrrTr RuntimeErrorrrrr-rr8r r r rs,   J