Edit File: 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
Back to File Manager