U
{�1`�) � @ s d Z ddlmZmZ ddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
mZmZmZ ze W n ek
r� eZY nX eZe�de� G dd � d e�ZG d
d � d ee�ZG d d
� d
ee �Z G dd� dee �Z G dd� d�Z ddd�Z dd� Z!dd� Z"dS )z Daemon runner library.
� )�absolute_import�unicode_literalsN� ��pidfile)�0_chain_exception_from_existing_exception_context�
DaemonContext�
basestring�unicodeu@ The ‘runner’ module is not a supported API for this library.c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�DaemonRunnerErrorz3 Abstract base class for errors from DaemonRunner. c s | � � t� j||� d S )N)�_chain_from_context�super�__init__)�self�args�kwargs�� __class__� �>/opt/alt/python38/lib/python3.8/site-packages/daemon/runner.pyr 2 s zDaemonRunnerError.__init__c C s t | dd� d S )NT�Zas_cause�r �r r r r r 7 s z%DaemonRunnerError._chain_from_context)�__name__�
__module__� __qualname__�__doc__r r �
__classcell__r r r r r / s r c @ s e Zd ZdZdd� ZdS )� DaemonRunnerInvalidActionErrorz; Raised when specified action for DaemonRunner is invalid. c C s t | dd� d S )NFr r r r r r r > s z2DaemonRunnerInvalidActionError._chain_from_contextN)r r r r r r r r r r ; s r c @ s e Zd ZdZdS )� DaemonRunnerStartFailureErrorz, Raised when failure starting DaemonRunner. N�r r r r r r r r r C s r c @ s e Zd ZdZdS )� DaemonRunnerStopFailureErrorz, Raised when failure stopping DaemonRunner. Nr r r r r r! G s r! c @ sr e Zd ZdZdZdd� Zdd� Zdd� Zdd
d �Zd d
� Z dd� Z
dd� Z dd� Z e e e d�Z
dd� Zdd� Zd S )� DaemonRunnera4 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}c C sN | � � || _t� | _| �|� d| _|jdk r@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_args�appr �daemon_context�#_open_streams_from_app_stream_pathsr Z pidfile_path�make_pidlockfileZpidfile_timeout�r r$ r r r r X s
�zDaemonRunner.__init__c C s8 t |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`.
Zrtzw+tr )� bufferingN)�openZ
stdin_pathr% �stdinZ stdout_path�stdoutZ stderr_path�stderrr( r r r r& z s
�z0DaemonRunner._open_streams_from_app_stream_pathsc C sH t j�|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})�prognameZusageN)
�os�path�basename�join� action_funcs�keys�format� emit_message�sys�exit)r �argvr0 Zusage_exit_codeZ action_usage�messager r r � _usage_exit� s �zDaemonRunner._usage_exitNc C sP |dkrt j}d}t|�|k r(| �|� t|d �| _| j| jkrL| �|� dS )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 )r9 r; �lenr= r
�actionr5 )r r; Zmin_argsr r r r# � s
zDaemonRunner.parse_argsc C s| t | j�r| j�� z| j�� W n, tjk
rN tdj| jd��}|�Y nX t �
� }| 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 lockedr )�pidN)�is_pidfile_staler �
break_lockr% r* �lockfileZ
AlreadyLockedr r7 r1 �getpid�
start_messager8 r$ �run)r �errorr@ r< r r r �_start� s
��
zDaemonRunner._startc
C sZ | j �� }zt�|tj� W n8 tk
rT } ztdj||d��}|�W 5 d}~X Y nX dS )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})r@ �excN) r �read_pidr1 �kill�signal�SIGTERM�OSErrorr! r7 )r r@ rI rG r r r �_terminate_daemon_process� s
��z&DaemonRunner._terminate_daemon_processc C sB | j �� s tdj| j d��}|�t| j �r6| j �� n| �� 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 lockedr N)r Z is_lockedr! r7 rA rB rO )r rG r r r �_stop� s
��
zDaemonRunner._stopc C s | � � | �� dS )z Stop, then start.
N)rP rH r r r r �_restart� s zDaemonRunner._restart)�start�stopZrestartc C s@ z| j | j }W n* tk
r: tdj| jd��}|�Y nX |S )ae 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})r? )r5 r? �KeyErrorr r7 )r �funcrG r r r �_get_action_func� s ��
z DaemonRunner._get_action_funcc C s | � � }|| � dS )z� Perform the requested action.
:return: ``None``.
The action is specified by the `action` attribute, which is set
during `parse_args`.
N)rV )r rU r r r � do_action s zDaemonRunner.do_action)N)r r r r rE r r& r= r# rH rO rP rQ r5 rV rW r r r r r" K s
"
�r" c C s, |dkrt j}|�dj| d�� |�� dS )z@ Emit a message to the specified stream (default `sys.stderr`). Nz
{message}
)r< )r9 r- �writer7 �flush)r< �streamr r r r8 s r8 c C sN t | t�s tdj| d��}|�tj�| �s>tdj| d��}|�t�| |�}|S )z= Make a PIDLockFile instance with the given filesystem path. z Not a filesystem path: {path!r})r2 z Not an absolute path: {path!r}) �
isinstancer �
ValueErrorr7 r1 r2 �isabsr ZTimeoutPIDLockFile)r2 Zacquire_timeoutrG rC r r r r' & s
� � r' c
C st d}| � � }|dk rpzt�|tj� W nH tk
r> d}Y n2 tk
rn } z|jtjkr^d}W 5 d}~X Y nX |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) rJ r1 rK rL �SIG_DFL�ProcessLookupErrorrN �errnoZESRCH)r �resultZ pidfile_pidrI r r r rA 5 s rA )N)#r Z
__future__r r r` r1 rL r9 �warningsrC � r Zdaemonr r r r
r_ � NameError�NotImplemented�typeZ
__metaclass__�warn�DeprecationWarning� Exceptionr r\ r � RuntimeErrorr r! r" r8 r' rA r r r r �<module>
s6
� T