File: //usr/local/lib/python3.9/site-packages/celery/bin/__pycache__/graph.cpython-39.pyc
a
X>h� � @ s� d Z ddlZddlmZ ddlZddlmZmZ ddlm Z m
Z
e� � ej edd� ���Z
e
jedd id
�ej d d � ��Ze
jedd id
�ej d
d� ��ZdS )� The ``celery graph`` command.� N)�
itemgetter)�
CeleryCommand�handle_preload_options)�DependencyGraph�GraphFormatterc C s dS )r N� )�ctxr r �:/usr/local/lib/python3.9/site-packages/celery/bin/graph.py�graph s r Zallow_extra_argsT)�clsZcontext_settingsc C sh | j j�� }dd� | jp ddgD �}d|v rN|jj}d|v rX|j�|jj� n
|jjj}|�t j
� dS )zDisplay bootsteps graph.c S s h | ] }|� � �qS r )�lower��.0�argr r r
� <setcomp> � z bootsteps.<locals>.<setcomp>�worker�consumerN) �obj�appZWorkController�argsZ blueprintr Z connect_withr �to_dot�sys�stdout)r r �includeZ worker_graphr r r
� bootsteps s
r c s �fdd��d-dd��t �fdd�| jD ���d�v �d d
� �G d d � d �}G �fd
d�d|�� G � fdd�dt�}dd� �G dd� d|�}G ��fdd�d|�}G ��fdd�d|�}ddlm} |d��t��d d �p�d�}t��d d �p�d�}|f��fd d!� } | jj}
z �d" } ��d#��p&g } W nb t �y� |
j
� � � � �pLi }
g g } } |
�
� D ]&\}}| �|� | �|d$ d% � �q`Y n0 t| �}��d&|
jj�}i }| | d�} |�r�||k�r�| dd � | d' g } t| �D ]*\}} | ttt| ���d(|�|| | < �q�|��d)|
�� �� ��}|�r4||�nd*}t|� d+�}|�|� |�r^|�|� dg}t| �D ]�\}}|||d,�}|�|� |�||� |�r�|�||� |�|j�} | �r�| D ]$}� |�}|�|� |�||� �q�|d d7 < �ql|�t j � d*S ).zDisplay workers graph.c s � t dd�| �d���S )Nr � �:)r � partition)r )�
maybe_listr r
� simplearg% s zworkers.<locals>.simplearg�,c S s* | d || d v r | d � |�n| d fS )Nr � )�split)�l�sepr r r
r ( s zworkers.<locals>.maybe_listc 3 s | ]}� |�V qd S �Nr r )r! r r
� <genexpr>+ r zworkers.<locals>.<genexpr>�genericc S s d� t| �j| j�d�d �S )Nz
{} ({}://)z://r )�format�type�__name__�_labelr$ )�noder r r
�
generic_label. s �z workers.<locals>.generic_labelc @ s. e Zd ZdZi Zddd�Zdd� Zdd� ZdS ) zworkers.<locals>.NodeNc S s || _ || _d S r' )r- �pos)�self�labelr0 r r r
�__init__6 s z workers.<locals>.Node.__init__c S s | j S r' �r- �r1 r r r
r2 : s zworkers.<locals>.Node.labelc S s | � � S r' �r2 r5 r r r
�__str__= s z workers.<locals>.Node.__str__)N)r, �
__module__� __qualname__Z force_label�schemer3 r2 r7 r r r r
�Node2 s
r; c s0 e Zd Zddddddd�Z� �fdd �Z� ZS )
zworkers.<locals>.ThreadZ
lightcyan4�yellowZoval�
g333333�?�black)� fillcolorZ fontcolor�shapeZfontsize�width�colorc s$ || _ t� jdt��� �dd� d S )Nzthr-r )r2 r0 )�
real_label�superr3 �next)r1 r2 �kwargs)� __class__�tidsr r
r3 J s
�z workers.<locals>.Thread.__init__)r, r8 r9 r: r3 �
__classcell__r )rH )rG r
�Thread@ s � rJ c s4 e Zd Zdd� Z� fdd�Zdd� Z� fdd�Zd S )
zworkers.<locals>.Formatterc S s |o
|� � S r' r6 �r1 r r r r
r2 S s z workers.<locals>.Formatter.labelc sD |j rt|j�n|j}t|� �r*|j|d< | �|t| jfi |���S )Nr2 )r0 �dictr: �
isinstancerC � draw_nodeZ node_scheme)r1 r r: �rJ r r
r. V s
�z workers.<locals>.Formatter.nodec S s | � |t| jfi |j���S r' )rN rL Z term_schemer: rK r r r
�
terminal_node^ s �z(workers.<locals>.Formatter.terminal_nodec s* t |� �r|jddd� | �||| j|�S )N�none�tee)Z arrowheadZ arrowtail)rM �updateZ draw_edgeZ edge_scheme)r1 �a�b�attrsrO r r
�edgec s
z workers.<locals>.Formatter.edgeN)r, r8 r9 r2 r. rP rW r rO r r
� FormatterQ s rX c s6 ddddddddd d
d �
� d � � fd
d�t| �D ��S )Nu ₀u ₁u ₂u ₃u ₄u ₅u ₆u ₇u ₈u ₉)
�0�1�2�3�4�5�6�7�8�9� c s g | ] }� | �qS r r )r �i��Sr r
�
<listcomp>k r z.workers.<locals>.subscript.<locals>.<listcomp>)�join�str)�nr re r
� subscripth s
�zworkers.<locals>.subscriptc @ s e Zd ZdS )zworkers.<locals>.WorkerN)r, r8 r9 r r r r
�Workerm s rl c s* e Zd Zdddddd�Z� �fdd�Zd S )
zworkers.<locals>.Backend�folderr r# r> Z
peachpuff3)r@ rA �heightrB r? c s � r �| �S | j S r' r4 r5 �r) r/ r r
r2 y s z workers.<locals>.Backend.labelN�r, r8 r9 r: r2 r ro r r
�Backendp s �rq c s( e Zd Zddddd�Z� �fdd�ZdS ) zworkers.<locals>.BrokerZcircleZ
cadetblue3Z
cadetblue4r# )r@ r? rB rn c s � r �| �S | j S r' r4 r5 ro r r
r2 � s z workers.<locals>.Broker.labelNrp r ro r r
�Broker| s �rr r )�countr# Zwmax� Ztmax� c s~ t | �}|o||k}d�v r4� �fdd�t| �D �} |rz| d|d � | |d g } d�� d �||d ��| |d < | S )N� enumeratec s$ g | ] \}}� � �|d �� ��qS )r# r )r rd r )�namerk r r
rg � s �z/workers.<locals>.maybe_abbr.<locals>.<listcomp>r r# u
{}⎨…{}⎬r )�lenrv r* )r% rw �max�sizeZabbr)r rk )rw r
�
maybe_abbr� s � � zworkers.<locals>.maybe_abbr�nodes�threads�poolzmax-concurrency�backend����P�brokerN)� formatter)r0 )r" ) rL r r � itertoolsrs �int�getr r �KeyError�control�inspect�stats�items�appendrx �confZresult_backendrv �list�rangeZconnection_for_read�as_urir Zadd_arcZadd_edger- r r r )r r; rX rl rq rr rs ZWmaxZTmaxr{ r �workersr} Zrepliesr ZreplyZwlenr Z threads_forrd r� �depsZ curworker�threadr )rJ r r) r/ r r! rk rH r
r� ! sx
� �
r� )�__doc__r �operatorr ZclickZcelery.bin.baser r Zcelery.utils.graphr r �groupZ pass_contextr �commandr r� r r r r
�<module> s