File: //usr/local/lib/python3.9/site-packages/celery/bin/__pycache__/control.cpython-39.pyc
a
X>h�! �
@ s$ d Z ddlmZ ddlmZ ddlZddlmZ ddlm Z m
Z
m Z m Z ddl
mZ ddlmZ dd lmZ dd
lmZ d5d d
�Zdd� Zdd� Zed Zeedd�dd�Zed�dd�Zee d�dd�Z ej e
d �ej d d e e d d d!d"�ej d#d$e e d d%d&�ej d'd(e d)d d*d+�ej!e d,d-� ������Z"ej e
d.d)id/�e�#d0�ed1�ej d d e e d d d!d"�ej d#d$e e d d%d&�ej d'd(e d)d d*d+�ej!e d2d1� ��������Z$ej e
d.d)id/�e�#d0�ed3�ej d d e e d d d!d"�ej d#d$e e d d%d&�ej d'd(e d)d d*d+�ej!e d4d3� ��������Z%dS )6z@The ``celery control``, ``. inspect`` and ``. status`` programs.� )�partial)�LiteralN)�dumps)�COMMA_SEPARATED_LIST�
CeleryCommand� CeleryOption�handle_preload_options)�CeleryCommandException)�EX_UNAVAILABLE)�text)�PanelFc C sl t t|��}|| }| jj|� d�ddd�}| j�|�\}}| jjd|� |� �|r^t�|d�nd|d� d S ) Nz: �cyanT)�fg�boldz->� � )Z show_body)�next�iter�obj�style�prettyZsay_chatr �indent)�ctx�replies�
show_reply�node�reply�statusZpreply� r �</usr/local/lib/python3.9/site-packages/celery/bin/control.py�_say_remote_command_reply s �r c
c s� d}z�t |�D ]h\}}z| j| \}}W n2 tyZ | jrBY qxt�d�|| j���Y q0 ||d urn||�n|fV qW ||d � |d d �< n||d � |d d �< 0 d S )Nr z Command {!r} takes arguments: {})� enumerate�args�
IndexError�variadic�click�
UsageError�format� signature)�meta�methodr" �i�arg�name�typr r r �_consume_arguments s ��
r/ c C sJ t j| }i }|jr0|�dd� t|| |�D �� |jrF|�|j|i� |S )Nc S s i | ]\}}||�qS r r )�.0�k�vr r r �
<dictcomp>/ s z&_compile_arguments.<locals>.<dictcomp>)r r) r" �updater/ r$ )�commandr" r) � argumentsr r r �_compile_arguments+ s
�r7 )�inspect�control)�type_r5 �returnc C s< t | �}||vr8d�|�}tjd|� d| � d|� �d��d S )Nz, zCommand z not recognized. Available z commands: )�message)�_get_commands_of_type�joinr% r& )r: r5 �choicesZcommand_listingr r r �_verify_command_name: s
�r@ �r: c s4 t jd d�� fdd� }t jddd� � d�dd|d �S )
N)r r; c s� |sd S t � �}t�� }|�� �� � d���\ g }|�� D ]4\}}|jrZ|� d|j� �}n|}|�||jf� q:|� |� W d � n1 s�0 Y | j
j |� � dd� | �
� d S )Nz Commands� F)�nl)r= r% �
HelpFormatter�section�
capitalize�itemsr( �append�helpZwrite_dlr �echo�getvalue�exit) r �param�valuer? � formatterZ command_list� command_name�infoZcommand_previewrA r r �callbackE s (z _list_option.<locals>.callbackz--listTzList available z commands and exit.F)�is_flagrI Z expose_valueZis_eagerrR )r% �Context�option)r: rR r rA r � _list_optionD s
�rV )r: r; c s$ � fdd�t j�� D �}tt|��S )Nc s( g | ] \}}|j � kr|jr||f�qS r )�typeZvisible)r0 r- rQ rA r r �
<listcomp>c s �z)_get_commands_of_type.<locals>.<listcomp>)r r) rG �dict�sorted)r: Zcommand_name_info_pairsr rA r r= b s
�r= )�clsz-tz --timeoutg �?zRemote Control Optionsz%Timeout in seconds waiting for reply.)r[ rW �default�
help_grouprI z-dz
--destinationz/Comma separated list of destination node names.)r[ rW r] rI z-jz--jsonTzUse json as output format.)r[ rS r] rI c K s� |rdnt t| �}| jjjj|||d��� }|s<tdtd��|rP| j� t
|�� t |�}|� dd�s�| j� d�
|t�|d��� dS ) z%Show list of workers that are online.N��timeout� destinationrR �'No nodes replied within time constraint�r< Z exit_code�quietF�
{} {} online.r )r r r �appr9 r8 Zpingr r
rJ r �len�getr' r � pluralize)r r_ r` �json�kwargsrR r � nodecountr r r r j s" �
�
�r Zallow_extra_args)r[ Zcontext_settingsr5 r8 c K s� t d|� |rdn tt| dd�}t|| j�}| jjjj|||d�}|j |fi |��} | sdt
dt d��|r|| j� t
| �� dS t| �}
| jjs�| j� d�|
t�|
d ��� dS )
z~Inspect the workers by sending them the COMMAND inspect command.
Availability: RabbitMQ (AMQP) and Redis transports.
r8 NT�r r^ ra rb rd r )r@ r r r7 r" r re r9 r8 �_requestr r
rJ r rf rc r' r rh ) r r5 r_ r` ri rj rR r6 r8 r rk r r r r8 � s.
� ��
�r9 c C st t d|� |rdn tt| dd�}| j}t||�}| jjjj||||d|d�}|s\t dt
d��|rp| j� t |�� dS )zxSend the COMMAND control command to the workers.
Availability: RabbitMQ (AMQP), Redis, and MongoDB transports.
r9 NTrl )r_ r` rR r r6 ra rb )
r@ r r r" r7 r re r9 � broadcastr r
rJ r ) r r5 r_ r` ri rR r" r6 r r r r r9 � s&
�
��)F)&�__doc__� functoolsr �typingr r% Zkombu.utils.jsonr Zcelery.bin.baser r r r Zcelery.exceptionsr Zcelery.platformsr
Z celery.utilsr Zcelery.worker.controlr r r/ r7 Z_RemoteControlType�strr@ rV rY r= r5 rU �floatZ pass_contextr �argumentr8 r9 r r r r �<module> s�
������� ����