File: //proc/self/root/lib/python3.6/site-packages/supervisor/__pycache__/dispatchers.cpython-36.pyc
3
�k�`�K � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d d l
m Z d d
l
mZ d d � ZG d
d� d�ZG dd� de�ZG dd� de�ZG dd� de�ZdZd+Zd%d&� ZG d'd(� d(e�Zd)d*� ZdS ),� N)�compact_traceback)� as_string)�notify)�EventRejectedEvent)�ProcessLogStderrEvent)�ProcessLogStdoutEvent)�EventListenerStates)� getEventListenerStateDescription)�loggersc C s6 t |�d }x$|r0| j|d |� � r0|d8 }qW |S )N� )�len�endswith)ZhaystackZneedle�l� r �!/usr/lib/python3.6/dispatchers.py�find_prefix_at_end
s r c @ s\ e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dS )� PDispatcherz{ Asyncore dispatcher for mainloop, representing a process channel
(stdin, stdout, or stderr). This class is abstract. Fc C s || _ || _|| _d| _d S )NF)�process�channel�fd�closed)�selfr r r r r r �__init__ s zPDispatcher.__init__c C s d| j jt| �| j| jf S )Nz<%s at %s for %s (%s)>)� __class__�__name__�idr r )r r r r �__repr__ s zPDispatcher.__repr__c C s t �d S )N)�NotImplementedError)r r r r �readable% s zPDispatcher.readablec C s t �d S )N)r )r r r r �writable( s zPDispatcher.writablec C s t �d S )N)r )r r r r �handle_read_event+ s z PDispatcher.handle_read_eventc C s t �d S )N)r )r r r r �handle_write_event. s z PDispatcher.handle_write_eventc C s< t � \}}}}| jjjjjdt| �|||f � | j� d S )Nz:uncaptured python exception, closing channel %s (%s:%s %s))r r �config�options�loggerZcritical�repr�close)r Znil�t�vZtbinfor r r � handle_error1 s
zPDispatcher.handle_errorc C s, | j s(| jjjjjd| j| f � d| _ d S )Nz#fd %s closed, stopped monitoring %sT)r r r" r# r$ �debugr )r r r r r& > s zPDispatcher.closec C s d S )Nr )r r r r �flushD s zPDispatcher.flushN)r �
__module__� __qualname__�__doc__r r r r r r r! r) r& r+ r r r r r s
r c @ s| e Zd ZdZdZdZdZdZdZdd� Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )�POutputDispatcherae
Dispatcher for one channel (stdout or stderr) of one process.
Serves several purposes:
- capture output sent within <!--XSUPERVISOR:BEGIN--> and
<!--XSUPERVISOR:END--> tags and signal a ProcessCommunicationEvent
by calling notify(event).
- route the output to the appropriate log handlers as specified in the
config.
NF� c C s� || _ || _|| _| jj| _| j� | j� | j| _| jj}| jj }|t
|�f| _ |t
|�f| _ t
jj| _| j j}|jj| jk| _|j| _|j| _dS )z�
Initialize the dispatcher.
`event_type` should be one of ProcessLogStdoutEvent or
ProcessLogStderrEvent
N)r �
event_typer r �_init_normallog�_init_capturelog� normallog�childlogZ BEGIN_TOKENZ END_TOKENr �begintoken_data�
endtoken_datar
Z LevelsByNameZDEBG�
mainlog_levelr" r# Zloglevel�log_to_mainlog�stdout_events_enabled�stderr_events_enabled)r r r1 r Z
begintokenZendtokenr" r r r r Y s
zPOutputDispatcher.__init__c C s� | j j}| j}t|d| �}t|d| �}t|d| �}t|d| �}|sN|rZ|jj� | _|rztj| j|d| ||d� |r�tj | j|j
d d� d S )
z�
Configure the "normal" (non-capture) log for this channel of this
process. Sets self.normallog if logging is enabled.
z
%s_logfilez%s_logfile_maxbytesz%s_logfile_backupsz %s_syslogz %(message)s)�filename�fmt�rotating�maxbytes�backupsz %(message)s)r= N) r r" r �getattrr# � getLoggerr4 r
� handle_fileZ
handle_syslog�name)r r" r �logfiler? r@ Z to_syslogr r r r2 u s( z!POutputDispatcher._init_normallogc C s> t | jjd| j �}|r:| jjjj� | _tj| jd|d� dS )z�
Configure the capture log for this process. This log is used to
temporarily capture output when special output is detected.
Sets self.capturelog if capturing is enabled.
z%s_capture_maxbytesz %(message)s)r= r? N) rA r r" r r# rB �
capturelogr
Zhandle_boundIO)r Zcapture_maxbytesr r r r3 � s z"POutputDispatcher._init_capturelogc C sB x<| j | jfD ],}|d k rx |jD ]}|j� |j� q"W qW d S )N)r4 rF �handlers�remove�reopen)r �log�handlerr r r �
removelogs� s
z POutputDispatcher.removelogsc C s: x4| j | jfD ]$}|d k rx|jD ] }|j� q"W qW d S )N)r4 rF rG rI )r rJ rK r r r �
reopenlogs� s z POutputDispatcher.reopenlogsc C s� |r�| j j}|jjr t|�}| jr.| jj|� | jr�t|t �sD|}n,y|j
d�}W n t k
rn d| }Y nX d}|jj j
| j||j| j|d� | jdkr�| jr�tt| j | j j|�� n | jr�tt| j | j j|�� d S )Nzutf-8zUndecodable: %rz%%(name)r %(channel)s output:
%(data)s)rD r �data�stdout)r r" r# �
strip_ansi� stripEscapesr5 �infor9 �
isinstance�bytes�decode�UnicodeDecodeErrorr$ rJ r8 rD r r: r r �pidr; r )r rN r" �text�msgr r r �_log� s4
zPOutputDispatcher._logc C s� | j d kr$| j}d| _| j|� d S | jr6| j\}}n
| j\}}t| j�|krRd S | j}d| _y|j|d�\}}W nT tk
r� d }t ||�}|r�| j|| d � | _|d | � }| j|� Y nX | j|� | j
� || _|r�| j � d S )Nr0 r ) rF �
output_bufferrZ � capturemoder7 r6 r �split�
ValueErrorr �toggle_capturemode�
record_output)r rN �token�tokenlenZbefore�after�indexr r r r` � s4
z POutputDispatcher.record_outputc C s� | j | _ | jd k r�| j r$| j| _n�x| jjD ] }|j� q.W | jj� }| j}| jjj }| j
| j| jj |�}t |� d}| jjj
jj|||d� x | jjD ]}|j� |j� q�W | j| _d S )Nz-%(procname)r %(channel)s emitted a comm event)�procnamer )r\ rF r5 rG r+ �getvaluer r r" rD r1 rW r r# r$ r* rH rI r4 )r rK rN r re �eventrY r r r r_ � s&
z$POutputDispatcher.toggle_capturemodec C s dS )NFr )r r r r r s zPOutputDispatcher.writablec C s | j r
dS dS )NFT)r )r r r r r s zPOutputDispatcher.readablec C s8 | j jjj| j�}| j|7 _| j� |s4| j� d S )N)r r" r# �readfdr r[ r` r& )r rN r r r r s
z#POutputDispatcher.handle_read_event)r r, r- r. r5 r4 rF r\ r[ r r2 r3 rL rM rZ r` r_ r r r r r r r r/ G s"
$r/ c @ sx e Zd ZdZdZdZdZdZee�Z ee�Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )�PEventListenerDispatcherzN An output dispatcher that monitors and changes a process'
listener_state Nr0 s READY
s RESULT c C s� t j| |||� tj| j_d | j_d| _d | _t |j
d| �}|r�t |j
d| �}t |j
d| �}|j
j j � | _
tj| j
|d| ||d� d S )Nr0 z
%s_logfilez%s_logfile_maxbytesz%s_logfile_backupsz %(message)s)r> r? r@ )r r r � ACKNOWLEDGEDr �listener_staterg �result� resultlenrA r" r# rB r5 r
rC )r r r r rE r? r@ r r r r + s"
z!PEventListenerDispatcher.__init__c C s0 | j d k r,x | j jD ]}|j� |j� qW d S )N)r5 rG rH rI )r rK r r r rL C s
z#PEventListenerDispatcher.removelogsc C s( | j d k r$x| j jD ] }|j� qW d S )N)r5 rG rI )r rK r r r rM I s
z#PEventListenerDispatcher.reopenlogsc C s dS )NFr )r r r r r O s z!PEventListenerDispatcher.writablec C s | j r
dS dS )NFT)r )r r r r r R s z!PEventListenerDispatcher.readablec C s� | j jjj| j�}|rx| j|7 _| j jj}d|| j|f }| j jjjj |� | j
r�| j jjj rjt |�}| j
j
|� n| j� | j� d S )Nz%r %s output:
%s)r r" r# rh r � state_bufferrD r r$ r* r5 rP rQ rR r& � handle_listener_state_change)r rN re rY r r r r W s
z*PEventListenerDispatcher.handle_read_eventc
C sZ | j }|sd S | j}|jj}|j}|tjkr6d| _ d S |tjkr�t|�| j k rRd S |j
| j �r�| j tj
� | j }| j |d � | _ d |_n| j tj� d| _ d |_| j r�| j� nd S �n�|tj
kr�| j tj� d| _ d |_d S |tjk�rV| jd k�r�|jd�}|dk�rd S | j d |� }| j |d d � | _ || jd � }yt|�| _W n� tk
�r� y t|�}W n tk
�r� d| }Y nX |jjjjd||f � | j tj� d| _ tt ||j�� d |_d S X nt| jt| j � } | �r | j | j d | � 7 _ | j | d � | _ | jt| j � } | �sF| j | j � d | j_d| _ d | _| j �rV| j� d S )Nr0 �
r zUndecodable: %rz%s: bad result line: '%s'���) rn r r" rD rk r �UNKNOWNrj r �READY_FOR_EVENTS_LEN�
startswith�READY_FOR_EVENTS_TOKEN�_change_listener_stateZREADYrg ro ZBUSYrm �find�RESULT_TOKEN_START_LEN�intr^ r rV r# r$ �warnr r rl �
handle_result)
r rN r re �staterb �posZ result_linerm Zneededr r r ro k s|
z5PEventListenerDispatcher.handle_listener_state_changec C s� | j }|jj}|jjj}y2| j jjj|j|� |jd| � | j t
j � W nr t k
r� |j
d| � | j t
j � tt||j�� Y n6 |j
d| � | j t
j� tt||j�� Y nX d S )Nz%s: event was processedz%s: event was rejectedz%s: event caused an error)r r" rD r# r$ �groupZresult_handlerrg r* rv r rj � RejectEventrz r r rr )r rl r re r$ r r r r{ � s
z&PEventListenerDispatcher.handle_resultc C sf | j }|jj}|j}d|t|�t|�f }|jjjj|� ||_|tj krbd| }|jjjj
|� d S )Nz %s: %s -> %sz�%s: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol) r r" rD rk r r# r$ r* r rr rz )r Z new_stater re Z old_staterY r r r rv � s
z/PEventListenerDispatcher._change_listener_state)r r, r- r. r5 rn ru ZRESULT_TOKEN_STARTr rs rx r rL rM r r r ro r{ rv r r r r ri s Wri c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�PInputDispatcherz Input (stdin) dispatcher c C s t j| |||� d| _d S )Nr0 )r r � input_buffer)r r r r r r r r � s zPInputDispatcher.__init__c C s | j r| j rdS dS )NTF)r� r )r r r r r � s zPInputDispatcher.writablec C s dS )NFr )r r r r r � s zPInputDispatcher.readablec C s* | j jjj| j| j�}| j|d � | _d S )N)r r" r# �writer r� )r Zsentr r r r+ � s zPInputDispatcher.flushc C s\ | j rXy | j� W nD tk
rV } z(|jd tjkrDd| _ | j� n� W Y d d }~X nX d S )Nr r0 )r� r+ �OSError�args�errnoZEPIPEr& )r Zwhyr r r r! � s
z#PInputDispatcher.handle_write_eventN) r r, r- r. r r r r+ r! r r r r r� � s r� s [� H� f� A� B� C� D� R� s� u� J� K� h� l� p� mc C s� d}d}d}t | �}xv||k r�|dkr@| ||d � tkr@d}n@|r�| jt|�}|dkrh|| |d� S || ||� }|}d}|d7 }qW |S )z>
Remove all ANSI color escapes from the given string.
r0 r r Nrq )r �ANSI_TERMINATORSrw �ANSI_ESCAPE_BEGIN)�srl Zshow�i�L�nr r r rQ s
rQ c @ s e Zd ZdZdS )r zY The exception type expected by a dispatcher when a handler wants
to reject an event N)r r, r- r. r r r r r ! s r c C s |dkrt |��d S )Ns OK)r )rg Zresponser r r �default_handler% s r� )r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� )r� Z supervisor.medusa.asyncore_25r Zsupervisor.compatr Zsupervisor.eventsr r r r Zsupervisor.statesr r Z
supervisorr
r r r/ ri r� r� r� rQ � Exceptionr r� r r r r �<module> s, 4 Z H