File: //home/arjun/projects/env/lib/python3.10/site-packages/celery/bin/__pycache__/graph.cpython-310.pyc
o
-we� � @ 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 �I/home/arjun/projects/env/lib/python3.10/site-packages/celery/bin/graph.py�graph s r �allow_extra_argsT)�cls�context_settingsc C sh | j j�� }dd� | jpddgD �}d|v r'|jj}d|v r&|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�app�WorkController�args� blueprintr � connect_withr �to_dot�sys�stdout)r r �include� 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 �pnd�}t��d d �pxd�}|f��fd d!� } | jj}
z
�d" } ��d#�p�g } W n. t y� |
j
� � � � p�i }
g g } } |
�
� D ]\}}| �|� | �|d$ d% � q�Y nw t| �}��d&|
jj�}i }| | d�} |r�||kr�| dd � | d' g } t| �D ]\}} | ttt| ���d(|�|| | < q�|��d)|
�� �� ��}|�r||�nd*}t|� d+�}|�|� |�r(|�|� dg}t| �D ]E\}}|||d,�}|�|� |�||� |�rM|�||� |�|j�} | �rk| D ]}� |�}|�|� |�||� �qX|d d7 < �q/|�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 � |�fS | 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>+ s � 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- )r3 �pos)�self�labelr6 r r r
�__init__6 s
z workers.<locals>.Node.__init__c S s | j S r- �r3 �r7 r r r
r8 : s zworkers.<locals>.Node.labelc S s | � � S r- �r8 r; r r r
�__str__= s z workers.<locals>.Node.__str__r- )r2 �
__module__� __qualname__� force_label�schemer9 r8 r= r r r r
�Node2 s
rB c s0 e Zd Zddddddd�Z� �fdd �Z� ZS )
zworkers.<locals>.Thread�
lightcyan4�yellow�oval�
g333333�?�black)� fillcolor� fontcolor�shape�fontsize�width�colorc s$ || _ t� jdt��� �dd� d S )Nzthr-r )r8 r6 )�
real_label�superr9 �next)r7 r8 �kwargs)� __class__�tidsr r
r9 J s
�z workers.<locals>.Thread.__init__)r2 r> r? rA r9 �
__classcell__r )rS )rR r
�Thread@ s � rU c s4 e Zd Zdd� Z� fdd�Zdd� Z� fdd�Zd S )
zworkers.<locals>.Formatterc S s |o|� � S r- r< �r7 r r r r
r8 S s z workers.<locals>.Formatter.labelc sD |j rt|j�n|j}t|� �r|j|d< | �|t| jfi |���S )Nr8 )r6 �dictrA �
isinstancerN � draw_node� node_scheme)r7 r rA �rU r r
r4 V s
�z workers.<locals>.Formatter.nodec S s | � |t| jfi |j���S r- )rY rW � term_schemerA rV r r r
�
terminal_node^ s �z(workers.<locals>.Formatter.terminal_nodec s* t |� �r |jddd� | �||| j|�S )N�none�tee)� arrowhead� arrowtail)rX �update� draw_edge� edge_scheme)r7 �a�b�attrsr[ r r
�edgec s
z workers.<locals>.Formatter.edgeN)r2 r> r? r8 r4 r] rh r r[ r r
� FormatterQ s
ri 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 rv r
� subscripth s
� zworkers.<locals>.subscriptc @ s e Zd ZdS )zworkers.<locals>.WorkerN)r2 r> r? r r r r
�Workerm s r} c s* e Zd Zdddddd�Z� �fdd�Zd S )
zworkers.<locals>.Backend�folderr# r) rG �
peachpuff3)rJ rL �heightrM rH c � � r�| �S | j S r- r: r; �r/ r5 r r
r8 y � z workers.<locals>.Backend.labelN�r2 r> r? rA r8 r r� r r
�Backendp s �r� c s( e Zd Zddddd�Z� �fdd�ZdS ) zworkers.<locals>.Broker�circle�
cadetblue3�
cadetblue4r) )rJ rH rM r� c r� r- r: r; r� r r
r8 � r� z workers.<locals>.Broker.labelNr� r r� r r
�Broker| s �r� r )�countr) �wmax� �tmax� c s~ t | �}|o ||k}d�v r� �fdd�t| �D �} |r=| d|d � | |d g } d�� d �||d ��| |d < | S )N� enumeratec s$ g | ]\}}� � �|d �� ��qS )r) r )r ru r )�namer| r r
rx � s �z/workers.<locals>.maybe_abbr.<locals>.<listcomp>r r) u
{}⎨…{}⎬r# )�lenr� r0 )r+ r� �max�size�abbr)r r| )r� r
�
maybe_abbr� s � �zworkers.<locals>.maybe_abbr�nodes�threads�poolzmax-concurrency�backend����P�brokerN)� formatter)r6 )r( ) rW r r � itertoolsr� �int�getr r �KeyError�control�inspect�stats�items�appendr� �conf�result_backendr� �list�range�connection_for_read�as_urir �add_arc�add_edger3 r r r )r rB ri r} r� r� r� �Wmax�Tmaxr� r �workersr� �repliesr �reply�wlenr� � threads_forru r� �deps� curworker�threadr )rU r r/ r5 r& r' r| rS r
r� ! s|
��
� �
r� )�__doc__r �operatorr �click�celery.bin.baser r �celery.utils.graphr r �group� pass_contextr �commandr" r� r r r r
�<module> s