File: //usr/local/lib/python3.9/site-packages/celery/bin/__pycache__/base.cpython-39.pyc
a
X>h�# � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl Z ddl m Z m
Z
ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ z(ddlmZ ddl m Z ddl m Z W n" e y� dd� ZdZ!dZ"Y n0 e � Z!e � Z"G dd� d�Z#dd� Z$G dd� de j%�Z&G dd� de j'�Z(G dd � d e&�Z)G d d � d e(�Z*G d d � d e
�Z+G d!d"� d"e
�Z,G d#d$� d$e
�Z-G d%d&� d&e
�Z.G d'd(� d(e
�Z/G d)d*� d*e j0�Z1e,� Z2e-� Z3e.� Z4e/� Z5e1� Z6e+� Z7dS )+z Click customizations for Celery.� N)� OrderedDict��update_wrapper)�pformat)�Any)�Context� ParamType)�cached_property)�get_current_app)�user_preload_options)�text)�mlevel)�
maybe_iso8601)� highlight)�Terminal256Formatter)� PythonLexerc O s | S )z2Place holder function in case pygments is missing.� )�s�args�kwargsr r �9/usr/local/lib/python3.9/site-packages/celery/bin/base.pyr s r c @ s| e Zd ZdZddd�Zedd� �Zedd� �Zd d
d �Zd d d
�Z d dd�Z
d dd�Z dd� Z dd� Z
dd� Zd dd�Zd S )!�
CLIContextzContext Object for the CLI.Fc C s" |pt � | _|| _|| _|| _dS )zInitialize the CLI context.N)r
�app�no_color�quiet�workdir)�selfr r r r r r r �__init__% s zCLIContext.__init__c C s | j dddd�S )N�OK�greenT��fg�bold��style�r r r r r , s z
CLIContext.OKc C s | j dddd�S )N�ERROR�redTr r# r% r r r r& 0 s zCLIContext.ERRORNc K s | j r
|S tj|fi |��S d S �N)r �clickr$ �r �messager r r r r$ 4 s zCLIContext.stylec K s8 | j r"d|d< tj|fi |�� ntj|fi |�� d S �NF�color�r r) �echo�sechor* r r r r0 : s zCLIContext.sechoc K s8 | j r"d|d< tj|fi |�� ntj|fi |�� d S r, )r r) r/ r* r r r r/ A s zCLIContext.echoc K s@ d|d< | j r*d|d< tj|fi |�� ntj|fi |�� d S )NT�errFr- r. r* r r r �errorH s
zCLIContext.errorc C s� t |t�r| j| �|�fS t |t�rjd|v s4d|v r>| �|�S tj|ddd�}| js`t |t
t �}| j|fS t |t �r~| j|fS | jt
|�fS )N�okr2 T� )� sort_keys�indent)�
isinstance�listr � pretty_list�dict�pretty_dict_ok_error�json�dumpsr r �LEXER� FORMATTER�strr )r �nr r r r �prettyP s
zCLIContext.prettyc s |sdS d� � fdd�|D ��S )Nz - empty -�
c 3 s& | ] }� j d dd�� d|� �V qdS )�*�white)r! � Nr# )�.0�itemr% r r � <genexpr>b s z)CLIContext.pretty_list.<locals>.<genexpr>)�join�r rA r r% r r9 _ s
�zCLIContext.pretty_listc C sV z"| j t�| �|d �d d�fW S ty4 Y n0 | jt�| �|d �d d�fS )Nr3 � r4 r2 )r r r6 rB �KeyErrorr& rK r r r r; f s � �z CLIContext.pretty_dict_ok_error� c C s^ |dkr| j rd S | j r0| j|ddd�� d�p2d}| �|� d|� �� |rZ|rZ| �|� d S )Nz<-rE Tr rF rN )r r$ r/ )r � direction�title�bodyZ show_bodyZdirstrr r r �say_chato s "zCLIContext.say_chat)F)N)N)N)N)rN F)�__name__�
__module__� __qualname__�__doc__r r r r& r$ r0 r/ r2 rB r9 r; rR r r r r r " s
r c s � fdd�}t |� �S )z6Extract preload options and return a wrapped callable.c s\ | j j}dd� |j�dg �D �}|rF� fdd�|D �}tj�||d� �| g|�R i � ��S )Nc S s g | ]
}|j �qS r )�name)rG �or r r �
<listcomp>} � z:handle_preload_options.<locals>.caller.<locals>.<listcomp>Zpreloadc s i | ]}|� | �qS r r )rG Zpreload_option�r r r �
<dictcomp>� s �z:handle_preload_options.<locals>.caller.<locals>.<dictcomp>)Zsenderr �options)�objr � user_options�getr �send)�ctxr r r Zpreload_optionsr_ ��fr[ r �callerz s
�z&handle_preload_options.<locals>.callerr )rd re r rc r �handle_preload_optionsx s rf c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )� CeleryOptionz Customized option for Celery.c s. | j r|j| j | _t� j|g|�R i |��S r( )�default_value_from_contextr^ �default�super� get_default)r rb r r �� __class__r r rk � s zCeleryOption.get_defaultc s2 |� dd�| _|� dd�| _t� j|i |�� dS )zInitialize a Celery option.�
help_groupNrh )�poprn rh rj r �r r r rl r r r � s zCeleryOption.__init__)rS rT rU rV rk r �
__classcell__r r rl r rg � s rg c @ s e Zd ZdZdd� ZdS )�
CeleryCommandz Customized command for Celery.c C s� t � }| �|�D ]R}|�|�}|durt|d�rP|jrP|�t|j�g ��|� q|�dg ��|� q|�� D ]<\}}|� |�� |�
|� W d � ql1 s�0 Y qldS )z7Write all the options into the formatter if they exist.Nrn �Options) r �
get_paramsZget_help_record�hasattrrn �
setdefaultr@ �append�items�sectionZwrite_dl)r rb � formatter�opts�param�rvrW Z
opts_groupr r r �format_options� s
z CeleryCommand.format_optionsN)rS rT rU rV r~ r r r r rr � s rr c s4 e Zd ZdZ� fdd�Zeeeed�dd�Z� Z S )� DaemonOptionzCommon daemonization optionc s0 t � j|f|�dd�|�d| j�d�|�� d S )Nrn zDaemonization Options�callback)rn r� )rj r ro �daemon_settingrp rl r r r � s
��zDaemonOption.__init__)rb �opt�value�returnc C s& |p$t |jjj|jj� d| j� �d�S )z�
Try to fetch daemonization option from applications settings.
Use the daemon command name as prefix (eg. `worker` -> `worker_pidfile`)
�_N)�getattrr^ r �conf�commandrW )r rb r� r� r r r r� � s zDaemonOption.daemon_setting)
rS rT rU rV r r rg r r� rq r r rl r r � s r c s e Zd ZdZ� fdd�Z� ZS )�CeleryDaemonCommandzDaemon commands.c s` t � j|i |�� | j�tdddd�tddd�tddd�td d
d�td d d�td
dd�f� dS )z7Initialize a Celery command with common daemon options.z --logfilez-fz#Log destination; defaults to stderr)�helpz --pidfilez&PID file path; defaults to no PID filez--uidz Drops privileges to this user IDz--gidz!Drops privileges to this group IDz--umaskz,Create files and directories with this umaskz --executablez&Override path to the Python executableN)rj r �params�extendr rp rl r r r � s
�z CeleryDaemonCommand.__init__)rS rT rU rV r rq r r rl r r� � s r� c @ s e Zd ZdZdZdd� ZdS )�CommaSeparatedListz Comma separated list argument.zcomma separated listc C s
t �|�S r( )r Z str_to_list�r r� r| rb r r r �convert� s zCommaSeparatedList.convertN�rS rT rU rV rW r� r r r r r� � s r� c @ s e Zd ZdZdZdd� ZdS )� JsonArrayz JSON formatted array argument.z
json arrayc
C sn t |t�r|S zt�|�}W n2 tyN } z| �t|�� W Y d }~n
d }~0 0 t |t�sj| �|� d�� |S )Nz was not an array)r7 r8 r< �loads�
ValueError�failr@ �r r� r| rb �v�er r r r� � s
$
zJsonArray.convertNr� r r r r r� � s r� c @ s e Zd ZdZdZdd� ZdS )�
JsonObjectz JSON formatted object argument.z json objectc
C sn t |t�r|S zt�|�}W n2 tyN } z| �t|�� W Y d }~n
d }~0 0 t |t�sj| �|� d�� |S )Nz was not an object)r7 r: r< r� r� r� r@ r� r r r r� � s
$
zJsonObject.convertNr� r r r r r� � s r� c @ s e Zd ZdZdZdd� ZdS )�ISO8601DateTimez ISO 8601 Date Time argument.z iso-86091c
C sB z
t |�W S ttfy< } z| �|� W Y d }~n
d }~0 0 d S r( )r � TypeErrorr� r� �r r� r| rb r� r r r r� s
zISO8601DateTime.convertNr� r r r r r� s r� c @ s e Zd ZdZdZdd� ZdS )�ISO8601DateTimeOrFloatz%ISO 8601 Date Time or float argument.ziso-86091 or floatc
C sd z
t |�W S ttfy Y n0 z
t|�W S ttfy^ } z| �|� W Y d }~n
d }~0 0 d S r( )�floatr� r� r r� r� r r r r� s
z ISO8601DateTimeOrFloat.convertNr� r r r r r� s r� c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�LogLevelzLog level option.c s t � �d� dS )z:Initialize the log level option with the relevant choices.)�DEBUG�INFO�WARNINGr& �CRITICAL�FATALN)rj r r% rl r r r s zLogLevel.__init__c s0 t |tj�r|S |�� }t� �|||�}t|�S r( )r7 �numbers�Integral�upperrj r� r
r� rl r r r� $ s
zLogLevel.convert)rS rT rU rV r r� rq r r rl r r� s r� )8rV r< r� � collectionsr � functoolsr �pprintr �typingr r) r r Zkombu.utils.objectsr Z
celery._stater
Zcelery.signalsr Z celery.utilsr Zcelery.utils.logr
Zcelery.utils.timer Zpygmentsr Zpygments.formattersr Zpygments.lexersr � ImportErrorr> r? r rf �Optionrg �Commandrr r r� r� r� r� r� r� ZChoicer� Z
JSON_ARRAYZ JSON_OBJECTZISO8601ZISO8601_OR_FLOATZ LOG_LEVELZCOMMA_SEPARATED_LISTr r r r �<module> sT
V