File: //usr/local/lib/python3.9/site-packages/celery/bin/__pycache__/logtool.cpython-39.pyc
a
X>h� � @ s� d Z ddlZddlmZ ddlmZ ddlZddlmZm Z dZ
e� d�Z e� d�Z
e� d �Ze� d
�Ze� d �Zd ZG d
d� de�Zdd� ZG dd� d�Ze�� eje dd� ���Zejed�ejddd�ejdd� ���Zejed�ejddd�ejdd � ���Z ejed�ejddd�ejd d � ���Z ejed�ejddd�ejd d � ���Z ejed�ejddd�ejd!d"� ���Z dS )#� The ``celery logtool`` command.� N)�Counter)� FileInput)�
CeleryCommand�handle_preload_options)�logtoolz^\[\d\d\d\d\-\d\d-\d\d z.+?\] Receivedz
.+?\] Taskz.+?([\w\.]+)\[(.+?)\].+z.+?[\w\.]+\[.+?\] (.+)z�
Report
======
Task total: {task[total]}
Task errors: {task[errors]}
Task success: {task[succeeded]}
Task completed: {task[completed]}
Tasks
=====
{task[types].format}
c @ s e Zd Zedd� �ZdS )� _task_countsc C s d� dd� | D ��S )N�
c s s | ]}d j |� V qdS )z{}: {}N)�format)�.0�i� r
�</usr/local/lib/python3.9/site-packages/celery/bin/logtool.py� <genexpr># � z&_task_counts.format.<locals>.<genexpr>)�join��selfr
r
r r
! s z_task_counts.formatN)�__name__�
__module__� __qualname__�propertyr
r
r
r
r r s r c C s t �| �}|�� S �N)� RE_TASK_INFO�match�groups)�line�mr
r
r � task_info& s
r c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�AuditNc C sF t � | _i | _i | _t � | _t� | _d| _|| _|| _ || _
d | _ d S )Nr ) �set�ids�names�results�readyr �
task_types� task_errors�
on_task_error�on_trace�on_debug� prev_line)r r' r( r) r
r
r �__init__- s zAudit.__init__c C s t |�D ]}| �|� q| S r )r �feed)r �filesr r
r
r �run9 s z Audit.runc C s, || j |< | j�|� | j| d7 < d S �N� )r" r! �addr% )r r � task_name�task_idr
r
r �
task_received> s
zAudit.task_receivedc C s2 | j �|� || j|< d|vr.| �||||� d S )N� succeeded)r$ r1 r# �
task_error�r r r2 r3 �resultr
r
r �
task_readyC s
zAudit.task_readyc C s( | j d7 _ | jr$| �||||� d S r/ )r& r' r7 r
r
r r6 I s zAudit.task_errorc C s� t �|�r�t�|�r0t|�\}}| �|||� nPt�|�rpt|�\}}t�|�}|r^|�� \}| �||||� n| j r�| � |� || _
n(| j r�| � d� t
d | j
|g��� d | _
d S )Nr )� RE_LOG_STARTr �RE_TASK_RECEIVEDr r4 �
RE_TASK_READY�RE_TASK_RESULTr r9 r) r* r( r �filterr7 r
r
r r, N s
z
Audit.feedc C s | j | jA S r )r! r$ r r
r
r �incomplete_tasksb s zAudit.incomplete_tasksc C s8 dt | j�� �t| j�| jt| j�t| j�| j d�iS )N�task)�types�total�errors� completedr5 )r r% � most_common�lenr! r& r$ r r
r
r �reporte s ��z Audit.report)NNN) r r r r+ r. r4 r9 r6 r, r? rG r
r
r
r r + s
r c C s dS )r Nr
��ctxr
r
r r q s r )�clsr- ���)�nargsc C s( | j �tjf i t� �|��� ��� d S r )�obj�echo�
REPORT_FORMATr
r r. rG �rI r- r
r
r �statsx s �rQ c C s t | jjd��|� d S )N)r( �r rM rN r. rP r
r
r �traces� s rS c s t � fdd�d��|� d S )Nc s � j �| �S r )rM rN )r �_rH r
r �<lambda>� r zerrors.<locals>.<lambda>)r' )r r. rP r
rH r rC � s rC c C s4 t � }|�|� |�� D ]}| j�d|� �� qd S )NzDid not complete: )r r. r? rM rN )rI r- �auditr3 r
r
r �
incomplete� s
rW c C s t | jjd��|� d S )N)r) rR rP r
r
r �debug� s rX ) �__doc__�re� collectionsr Z fileinputr ZclickZcelery.bin.baser r �__all__�compiler: r; r< r r= rO �listr r r �groupZ pass_contextr �command�argumentrQ rS rC rW rX r
r
r
r �<module> sN
F