File: //home/arjun/projects/env/lib/python3.10/site-packages/honcho/__pycache__/manager.cpython-310.pyc
o
weu � @ s� d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
dZejdd d
�ejd d d
�iZd
ZG dd� de�ZdS )� N� )� get_colours)�Env)�Process)�Printer�Message� �SIGINT� )�name�rc�SIGTERM� �systemc @ sr e Zd ZdZdZddd�Zddd�Zdd � Zd
d � Zd d
� Z d dd�Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �Managera�
Manager is responsible for running multiple external processes in parallel
managing the events that result (starting, stopping, printing). By default
it relays printed lines to a printer that prints to STDOUT.
Example::
import sys
from honcho.manager import Manager
m = Manager()
m.add_process('server', 'ruby server.rb')
m.add_process('worker', 'python worker.py')
m.loop()
sys.exit(m.returncode)
Nc C sZ t �� | _d | _t� | _t� | _|d ur|ntt j
�| _ t t
�| j _t| _i | _d| _d S )NF)�multiprocessing�Queue�events�
returncoder �_coloursr �_envr �sys�stdout�_printer�len�SYSTEM_PRINTER_NAME�widthr �
_process_ctor�
_processes� _terminating)�self�printer� r" �G/home/arjun/projects/env/lib/python3.10/site-packages/honcho/manager.py�__init__1 s
zManager.__init__Fc C s` || j vs J d��| j|||t| j�||d�}i | j |< || j | d< t| jjt|��| j_|S )z�
Add a process to this manager instance. The process will not be started
until :func:`~honcho.manager.Manager.loop` is called.
z process names must be unique)r �quiet�colour�env�cwd�obj)r r �nextr �maxr r r )r r �cmdr% r' r( �procr" r" r# � add_process@ s �
zManager.add_processc s~ � fdd�}t � t j|� t � t j|� � �� d}d} z � jjdd�}W n tjy5 |r3Y dS Y nSw |jdkrB� j �
|� nF|jd kr`|j d
� j |j
d
< � �d |j
|j d
f � n(|jd kr�|j d
� j |j
d
< � �d|j
|j d
f � � jdu r�|j d
� _� �� r�� �� r�d}|du r�� �� r�� �� r�� j�� }� �� |dur�� j�� | }|tjtd�kr�� �� q )aj
Start all the added processes and multiplex their output onto the bound
printer (which by default will print to STDOUT).
If one process terminates, all the others will be terminated by
Honcho, and :func:`~honcho.manager.Manager.loop` will return.
This method will block until all the processes have terminated.
c s0 � � dt| d � t| d � _� �� d S )Nz %s received
r r )�
_system_print�SIGNALSr � terminate)�signum�frame�r r" r# �
_terminate^ s z Manager.loop.<locals>._terminateFNr g�������?)�timeout�line�start�pidz%s started (pid=%s)
�stopr z%s stopped (rc=%s)
T)�seconds)�signalr
r �_startr �get�queue�Empty�typer �write�datar r r/ r � _all_started� _all_stopped� _any_stoppedr �nowr1 �datetime� timedelta� KILL_WAIT�kill)r r5 �exit�
exit_start�msg�waitingr" r4 r# �loopT sL
��
�
�
�z Manager.loopc C s | j rdS d| _ | �� dS )zI
Terminate all processes managed by this ProcessManager.
NT)r �_killallr4 r" r" r# r1 � s zManager.terminatec C s | j dd� dS )zD
Kill all processes managed by this ProcessManager.
T)�forceN)rQ r4 r" r" r# rK � s z Manager.killc C s� g }| j �� D ]
\}}d|vr|�|� q|D ],}| j | }|r"dnd}| �d|||d f � |r;| j�|d � q| j�|d � qdS )z6Kill all remaining processes, forcefully if requested.r �SIGKILLr
zsending %s to %s (pid %s)
r9 N)r �items�appendr/ r rK r1 )r rR �for_termination�n�p�signamer" r" r# rQ � s
�
��zManager._killallc C sD | j �� D ]\}}tj||d j| jdfd�|d< |d �� qd S )Nr) T)r �target�args�process)r rT r r �runr r8 )r r rX r" r" r# r= � s
��zManager._startc C � t dd� | j�� D ��S )Nc s �" � | ] \}}|� d �duV qdS )r9 N�r> ��.0�_rX r" r" r# � <genexpr>� � � z'Manager._all_started.<locals>.<genexpr>��allr rT r4 r" r" r# rD � � zManager._all_startedc C r^ )Nc s r_ �r Nr` ra r" r" r# rd � re z'Manager._all_stopped.<locals>.<genexpr>rf r4 r" r" r# rE � rh zManager._all_stoppedc C r^ )Nc s r_ ri r` ra r" r" r# rd � re z'Manager._any_stopped.<locals>.<genexpr>)�anyr rT r4 r" r" r# rF � rh zManager._any_stoppedc C s$ | j �td|| j�� td d�� d S )Nr7 )rA rC �timer r& )r rB r r rG r )r rC r" r" r# r/ � s
�zManager._system_print)N)FNN)F)�__name__�
__module__� __qualname__�__doc__r r$ r. rP r1 rK rQ r= rD rE rF r/ r" r" r" r# r s
9
r )rH r? r r< r r& r �environr r\ r r! r r rJ r r
r0 r �objectr r" r" r" r# �<module> s* ���