File: //usr/local/lib/python3.9/site-packages/celery/events/__pycache__/cursesmon.cpython-39.pyc
a
X>h)F � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl m
Z
mZ dd lmZ dd
lmZmZ d Zd Zd
ZdZdZdZdZdZdZ G dd� d�Z G dd� dej �Z dd� Z d dd�Z!e"dkr�e!� dS ) z0Graphical monitor of Celery events using curses.� N)�datetime)�count)�ceil)�wrap)�time)�VERSION_BANNER�states)�app_or_default)�abbr�abbrtask)�
CursesMonitor�evtop� � �$ � � � zGevents: {s.event_count} tasks:{s.task_count} workers:{w_alive}/{w_all}
c @ sf e Zd ZdZi ZdZdZdZdZdZ e
j Z e
j
ZdZdZdZd e� �Zd
ZdGd d �Zd
d� Zedd� �Zedd� �Zedd� �Zedd� �Zedd� �Z dd� Z dd � Z d d � Z dHd d!�Z e
j!d"e
j"d#e
j#d$iZ$d%d&� Z%dId'd(�Z&d)d*� Z'd+d,� Z(d-d.� Z)d/d0� Z*d1d2� Z+d3d4� Z,d5d6� Z-d7d8� Z.d9d:� Z/d;d<� Z0d=d>� Z1d?d@� Z2dAdB� Z3edCdD� �Z4edEdF� �Z5dS )Jr z#A curses based Celery task monitor.N�
r z
Selected: zWorkers online: zKeys: z9j:down k:up i:info t:traceback r:result c:revoke ^c: quitzcelery events zInfo: c C s\ || _ |p| j| _|| _| j| j| j| j| j| j| j d�}t
|fi | j��| _t � � | _
d S )N)�J�K�C�T�R�I�L)�app�keymap�state�move_selection_down�move_selection_up�revoke_selection�selection_traceback�selection_result�selection_info�selection_rate_limit�dict� threading�RLock�lock)�selfr r r Zdefault_keymap� r+ �A/usr/local/lib/python3.9/site-packages/celery/events/cursesmon.py�__init__3 s � zCursesMonitor.__init__c
C s� | j }|d t d t }|d t d t }|tk r<|} nt} || d }tt|d ��}
||
d } t|| �� | �}t|| �� | �}t
||
�� |
�}t|t�� t�}|� t�}|� d|� d|� d|� d|� d�
} | j d u r�t | d |� �| _ | d |� S )N� g @� )
�
display_width� STATE_WIDTH�TIMESTAMP_WIDTH�MIN_TASK_WIDTH�MIN_WORKER_WIDTH�
UUID_WIDTH�intr r
�ljustr � screen_width�len)
r* �uuid�task�worker� timestampr �mxZ detail_widthZ
uuid_spaceZ
uuid_widthZ
task_widthZ worker_width�rowr+ r+ r, �
format_rowC s$
"
zCursesMonitor.format_rowc C s | j �� \}}|S �N��win�getmaxyx�r* �_r> r+ r+ r, r8 ^ s zCursesMonitor.screen_widthc C s | j �� \}}|S rA rB �r* �myrF r+ r+ r, �
screen_heightc s zCursesMonitor.screen_heightc C s | j �� \}}|t S rA )rC rD �BORDER_SPACINGrE r+ r+ r, r0 h s zCursesMonitor.display_widthc C s | j �� \}}|d S )Nr rB rG r+ r+ r, �display_heightm s z CursesMonitor.display_heightc C s | j S rA )rK �r* r+ r+ r, �limitr s zCursesMonitor.limitc C s8 | j s
dS t| j �D ] \}}| j|d kr| S qdS �Nr )�tasks� enumerate�
selected_task)r* �i�er+ r+ r, �
find_positionv s
zCursesMonitor.find_positionc C s | � d� d S )N�����move_selectionrL r+ r+ r, r ~ s z CursesMonitor.move_selection_upc C s | � d� d S )Nr. rV rL r+ r+ r, r � s z!CursesMonitor.move_selection_downr. c C sR | j s
d S | �� }z| j || d | _W n" tyL | j d d | _Y n0 d S rN )rO rT rQ �
IndexError)r* � direction�posr+ r+ r, rW � s z CursesMonitor.move_selectionr r r c C sV z| j �� �� }W n ty& Y d S 0 | j�|�p6|}| j�|�}|d urR|� d S rA )rC �getkey�upper� Exception�keyalias�getr )r* �key�handlerr+ r+ r, �handle_keypress� s z CursesMonitor.handle_keypressc C s� | j �� | j �� \}}td� }}|rN| j �t|�d|tjtjB � t|� |||t|�� | j �|d ddtj� | j � � z| j �
� � � W S t y� Y q�0 q�d S )N� r r. r z Press any key to continue...)
rC �eraserD r �addstr�next�curses�A_BOLD� A_UNDERLINE�refreshr[ r\ r] )r* �callback�titlerH r> �y�
blank_liner+ r+ r, �alert� s"
��
zCursesMonitor.alertc C s� | j st�� S | jj| j }|js*t�� S | j�� \}}d}| j�|d d|tj tj
B � | j�|d t |�d d|t |� � | � |d dt |� �}|r�| j
jj|j|�� dd�}| �|� d S )NzNew rate limit: rc r r/ T��reply)rQ rg �beepr rO �namerC rD re rh ri r9 �readliner �controlZ
rate_limit�strip�alert_remote_control_reply)r* r; rH r> �rZrlimitrq r+ r+ r, r% � s ( �z"CursesMonitor.selection_rate_limitc s � �fdd�}�� |d�S )Nc
s� t |�}� s2�j�t|�ddtjt�d� � d S � D ]�}t|�}t|�� �\}}|� d�}�j�|d|tj� tj}d} d|v r�|d } |t�d�O }n d|v r�|d } |t�d�O }�j�|dt |� | |� q6d S )Nr z#No replies received in 1s deadline.rc �: � �error�ok)
r rC re rf rg rh �
color_pair�items�A_NORMALr9 )
rH r> �xsrm Zsubreply�curline�host�response�attr�text�rq r* r+ r, rk � s*
�
z:CursesMonitor.alert_remote_control_reply.<locals>.callbackz Remote Control Command Replies)ro )r* rq rk r+ r� r, rw � s z(CursesMonitor.alert_remote_control_replyc C sz d}t �� z^d}| j�||| �}|dkr|dt jfv r<q`|dv rJd}q`|t|�7 }|d7 }qW t �� n
t �� 0 |S )Nrz r rU r )� r. )rg �echorC Zgetch� KEY_ENTER�chrZnoecho)r* �xrm �bufferrR �chr+ r+ r, rt � s zCursesMonitor.readlinec C s0 | j st�� S | jjj| j dd�}| �|� d S )NTrp )rQ rg rr r ru Zrevokerw )r* rq r+ r+ r, r! � s z CursesMonitor.revoke_selectionc s* � j s
d S � fdd�}� �|d� j � ��S )Nc
s: � j �� \}} t|�}� jj� j }|jdgd�}d|�dd �fd|�dd �fgt|� � � }|D ]�\}}|d u rtqbt
|�}t |�} |d }
� j � | d|
t
j� t|| d �} t| �dkr� j � | t|
�d t| d � jt|
�d �� qb| D ]B} t |�}
|
|d k�rd
} � j � |
dtd | � jd �t
j� q�qbd S )
Nr )�extra�args�kwargsry r rc r. r z [...]z r )rC rD r r rO rQ �info�pop�listr~ �strrf re rg rh r r9 r
r8 r )r> rH r� rm r; r� Z infoitemsr` �valuer� �keys�wrappedZsublineZnextyrL r+ r, �alert_callback� sB �
� ���z4CursesMonitor.selection_info.<locals>.alert_callbackzTask details for �rQ ro �r* r� r+ rL r, r$ � s ! �z CursesMonitor.selection_infoc sR � j st�� S � jj� j ��jtjvr0t�� S � �fdd�}� �|d� j � ��S )Nc s2 t |�}�j�d�D ]}� j�t|�d|� qd S )N�
r )r � traceback�splitrC re rf )rH r> r� rm �line�r* r; r+ r, r� % s z9CursesMonitor.selection_traceback.<locals>.alert_callbackz Task Exception Traceback for )rQ rg rr r rO r �EXCEPTION_STATESro r� r+ r� r, r" s
�z!CursesMonitor.selection_tracebackc s* � j s
d S � fdd�}� �|d� j � ��S )Nc s^ t |�}� jj� j }t|dd �p,t|dd �}t|p6d|d �D ]}� j�t|�d|� q@d S )N�result� exceptionrz rc r ) r r rO rQ �getattrr rC re rf )rH r> r� rm r; r� r� rL r+ r, r� 3 s
�z6CursesMonitor.selection_result.<locals>.alert_callbackzTask Result for r� r� r+ rL r, r# / s
�z CursesMonitor.selection_resultc C s� | j �|j�}tj}|j| jkr&tj}t� |j
p4t � �}|� d�}|j
rP|j
jnd}| �|j|j|||j�}| j�|t||� |r�| j�|t|�t t d |j||B � d S )Nz%H:%M:%Sz*NONE*r. )� state_colorsr_ r rg r r: rQ Z
A_STANDOUTr �utcfromtimestampr= r �strftimer< �hostnamer@ rs rC re �LEFT_BORDER_OFFSETr9 r1 rJ ) r* �linenor; Z state_colorr� r= Ztimefr� r� r+ r+ r, �display_task_row@ s&
�
�
�z CursesMonitor.display_task_rowc C sv | j ��X | j}| �� t}td� }}|�� \}}|�� |�dt� d�� |�
� |� d|| j tj
t� d�B � t|� |� t|�|| �ddddd �tjtjB � | j}|r�t|�D ]4\}\}} || jkr� q�| jr�t|�}
| �|
| � q�t|� |�|d
|tj| jd � | j�r6|� |d || jtj� d } d
} z| j j| j }
W n t �y` Y n^0 |
� � } d| v �r�d� | d �| d< d| v �r�t | d d�| d< d�!dd� | �"� D ��} d} t | | jt#| j� d | �}|� |d |t#| j� |� | |v �rLt#|�t#| � }|� |d |t#| j� | | tj� n|� |d |dtj$� | j%�r�|� |d || j&tj� |� |d |t#| j&� d�!t'| j%��tj$� n|� |d |d� |� |d || j(tj� |� |d |t#| j(� t)j | j t#dd� | j j%�*� D ��t#| j j%�d�tj
� | �+|d || j,tj� | �+|d |t#| j,� | j-tj
� |�.� W d � n1 �sh0 Y d S ) Nrc r/ r. � �UUIDZTASKZWORKERZTIMEZSTATE� r zMissing extended inforz Zruntimez{:.2f}r� r c s s | ]\}}|� d |� �V qdS )�=Nr+ )�.0r` r� r+ r+ r, � <genexpr>� s �z%CursesMonitor.draw.<locals>.<genexpr>z ... -> key izNo task selectedz, zNo workers discovered.r c S s g | ]}|j r|�qS r+ ��alive)r� �wr+ r+ r, �
<listcomp>� s �z&CursesMonitor.draw.<locals>.<listcomp>)�sZw_aliveZw_all)/r) rC rb r� r rD rd Zbkgdrg r} Zborderre �greetZA_DIMrf r@ rh ri rO rP rK r: r� ZhlineZ ACS_HLINEr8 rQ � selected_strr �KeyErrorr� �formatr
�joinr~ r9 r �workers�
online_str�sorted�info_str�
STATUS_SCREEN�values� safe_add_str�
help_title�helprj )r* rC r� rm rn rH rF rO r? r; r� r� ZdetailZ selectionZinfowinZ detailposr+ r+ r, �drawT s�
�
�
��
��
�� �zCursesMonitor.drawc O sF |t |� | jkr$|d | j| � }| jj|||g|�R i |�� d S rA )r9 r8 rC re )r* rm r� �stringr� r� r+ r+ r, r� � s zCursesMonitor.safe_add_strc C s | j �� t�� | _| j�d� | j�d� t�� t�d| j| j � t�dtj
| j � t�dtj | j � t�dtj | j � t�dtj
| j � t�dtj| j� tjt�d�tjt�d�tjt�d�i| _tjD ]}t�d�| j|< q�t�� W d � n1 �s0 Y d S )NTr. rc r r r� r� )r) rg ZinitscrrC Znodelay�keypadZ start_colorZ init_pair�
foreground�
backgroundZ COLOR_REDZ COLOR_GREENZ
COLOR_MAGENTAZ
COLOR_BLUEZ COLOR_YELLOWr �SUCCESSr} ZREVOKEDZSTARTEDr� r� Zcbreak)r* r r+ r+ r, � init_screen� s$
�
zCursesMonitor.init_screenc C sN | j �4 t�� | j�d� t�� t�� W d � n1 s@0 Y d S �NF)r) rg ZnocbreakrC r� r� ZendwinrL r+ r+ r, � resetscreen� s
zCursesMonitor.resetscreenc C s t �| j� d S rA )rg Znapms� screen_delayrL r+ r+ r, �nap� s zCursesMonitor.napc C s t | jj| jd��S )N)rM )r� r Z
tasks_by_timerM rL r+ r+ r, rO � s zCursesMonitor.tasksc C s dd� | j j�� D �S )Nc S s g | ]\}}|j r|�qS r+ r� )r� r� r� r+ r+ r, r� � s �z)CursesMonitor.workers.<locals>.<listcomp>)r r� r~ rL r+ r+ r, r� � s zCursesMonitor.workers)N)r. )N)6�__name__�
__module__� __qualname__�__doc__r rC r� rQ Zselected_positionr� rg Z COLOR_BLACKr� Z COLOR_WHITEr� r� r� r� r r� r� r- r@ �propertyr8 rI r0 rK rM rT r r rW ZKEY_DOWNZKEY_UPr� r^ rb ro r% rw rt r! r$ r" r# r� r� r� r� r� r� rO r� r+ r+ r+ r, r " sj
�
)X
r c s$ e Zd Z� fdd�Zdd� Z� ZS )�
DisplayThreadc s || _ d| _t� �� d S r� )�display�shutdown�superr- )r* r� �� __class__r+ r, r- � s zDisplayThread.__init__c C s | j s | j�� | j�� q d S rA )r� r� r� r� rL r+ r+ r, �run� s
zDisplayThread.run)r� r� r� r- r� �
__classcell__r+ r+ r� r, r� � s r� c C s� dd� }t dtjd� | �� ��}zB|�|| jj� | jj|d|j id�}|�
� |� � |� � W n@ |j
|j y� } z t d|��tjd� W Y d }~n
d }~0 0 W d � q1 s�0 Y qd S )Nc S s t d�| |�tjd� d S )Nz&Connection Error: {!r}. Retry in {}s.��file)�printr� �sys�stderr)�exc�intervalr+ r+ r, �on_connection_error� s
��z+capture_events.<locals>.on_connection_errorz -> evtop: starting capture...r� �*)�handlerszConnection lost: )r� r� r� Zconnection_for_readZensure_connection�confZ broker_connection_max_retries�eventsZReceiver�eventr� r� �captureZconnection_errorsZchannel_errors)r r r� r� �conn�recvr� r+ r+ r, �capture_events� s
� r� c C s� t | �} | j�� }t|| �}|�� t|�}|�� zt| ||� W nT tyn d|_ |�
� |� � � Y n, t t
fy� d|_ |�
� |� � Y n0 dS )zStart curses monitor.TN)r r� �Stater r� r� �startr� r] r� r� r� �KeyboardInterrupt�
SystemExit)r r r� Z refresherr+ r+ r, r
s"
r
�__main__)N)#r� rg r� r' r � itertoolsr �mathr �textwrapr r Zceleryr r Z
celery.appr Zcelery.utils.textr
r �__all__rJ r� r5 r1 r2 r4 r3 r� r �Threadr� r� r
r� r+ r+ r+ r, �<module> s: @