File: //home/arjun/projects/env/lib/python3.10/site-packages/celery/bin/__pycache__/base.cpython-310.pyc
o
-weM! � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl Z ddl m
Z
ddl m Z ddl
mZ dd lmZ dd
lmZ dd lmZ dd lmZ zdd
lmZ ddlmZ ddlm Z W n e yo dd� ZdZ dZ Y nw e � Z e� Z G dd� d�Z dd� Z!G dd� de j"�Z#G dd� de j$�Z%G dd� de%�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 j,�Z-e(� Z.e)� Z/e*� Z0e+� Z1e-� Z2e'� Z3dS )(z Click customizations for Celery.� N)� OrderedDict��update_wrapper)�pformat)� 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 �H/home/arjun/projects/env/lib/python3.10/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 � | j dddd�S )N�OK�greenT��fg�bold��style�r r r r r + � z
CLIContext.OKc C r )N�ERROR�redTr r" r$ r r r r& / r% zCLIContext.ERRORNc K s | j r|S tj|fi |��S �N)r �clickr# �r �messager r r r r# 3 s zCLIContext.stylec K s: | j rd|d< tj|fi |�� d S tj|fi |�� d S �NF�color�r r) �echo�sechor* r r r r0 9 � zCLIContext.sechoc K s: | j rd|d< tj|fi |�� d S tj|fi |�� d S r, )r r) r/ r* r r r r/ @ r1 zCLIContext.echoc K sB d|d< | j rd|d< tj|fi |�� d S tj|fi |�� d S )NT�errFr- r. r* r r r �errorG s
zCLIContext.errorc C s� t |t�r
| j| �|�fS t |t�r5d|v sd|v r | �|�S tj|ddd�}| js0t |t
t �}| j|fS t |t �r?| j|fS | jt
|�fS )N�okr3 T� )� sort_keys�indent)�
isinstance�listr � pretty_list�dict�pretty_dict_ok_error�json�dumpsr r
�LEXER� FORMATTER�strr )r �nr r r r �prettyO 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>a s �
�z)CLIContext.pretty_list.<locals>.<genexpr>)�join�r rB r r$ r r: ^ s
�zCLIContext.pretty_listc C sV z| j t�| �|d �d d�fW S ty Y nw | jt�| �|d �d d�fS )Nr4 � r5 r3 )r r
r7 rC �KeyErrorr& rL r r r r< e s � ��z CLIContext.pretty_dict_ok_error� c C sf |dkr | j r d S | j r| j|ddd�� d�pd}| �|� d|� �� |r/|r1| �|� d S d S d S )Nz<-rF Tr rG rO )r r# r/ )r � direction�title�body� show_body�dirstrr r r �say_chatn s "�zCLIContext.say_chat)Fr( )rO F)�__name__�
__module__� __qualname__�__doc__r r r r& r# r0 r/ r3 rC r: r< rU 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 �}|r#� fdd�|D �}tj�||d� �| g|�R i � ��S )Nc S s g | ]}|j �qS r )�name)rH �or r r �
<listcomp>| s z:handle_preload_options.<locals>.caller.<locals>.<listcomp>�preloadc s i | ]}|� | �qS r r )rH �preload_option�r r r �
<dictcomp> s ��z:handle_preload_options.<locals>.caller.<locals>.<dictcomp>)�senderr �options)�objr � user_options�getr �send)�ctxr r r �preload_optionsrd ��fr_ r �callery s
�z&handle_preload_options.<locals>.callerr )rj rk r ri r �handle_preload_optionsw s
rl c �, 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_contextrc �default�super� get_default)r rg r r �� __class__r r rr � s zCeleryOption.get_defaultc s2 |� dd�| _|� dd�| _t� j|i |�� dS )zInitialize a Celery option.�
help_groupNro )�popru ro rq r �r r r rs r r r � s zCeleryOption.__init__)rV rW rX rY rr r �
__classcell__r r rs r rn � s rn c @ s e Zd ZdZdd� ZdS )�
CeleryCommandz Customized command for Celery.c C s� t � }| �|�D ])}|�|�}|dur1t|d�r(|jr(|�t|j�g ��|� q|�dg ��|� q|�� D ] \}}|� |��
|�
|� W d � n1 sOw Y q6dS )z7Write all the options into the formatter if they exist.Nru �Options) r �
get_params�get_help_record�hasattrru �
setdefaultrA �append�items�section�write_dl)r rg � formatter�opts�param�rvrZ �
opts_groupr r r �format_options� s
� ���z CeleryCommand.format_optionsN)rV rW rX rY r� r r r r ry � s ry c s e Zd ZdZ� fdd�Z� ZS )�CeleryDaemonCommandzDaemon commands.c s� t � j|i |�� | j�tdddd�� | j�tddd�� | j�tddd�� | j�tddd�� | j�td dd�� | j�td
dd�� d S ) z7Initialize a Celery command with common daemon options.)z-fz --logfilezDaemonization Optionsz#Log destination; defaults to stderr)ru �help)z --pidfile)ru )z--uid)z--gid)z--umask)z --executableN)rq r �paramsr rn rw rs r r r � s �z CeleryDaemonCommand.__init__)rV rW rX rY r rx r r rs r r� � s r� c @ � e Zd ZdZdZdd� ZdS )�CommaSeparatedListz Comma separated list argument.zcomma separated listc C s
t �|�S r( )r
� str_to_list�r �valuer� rg r r r �convert� s
zCommaSeparatedList.convertN�rV rW rX rY rZ r� r r r r r� � � r� c @ r� )� JsonArrayz JSON formatted array argument.z
json arrayc
C �n t |t�r|S zt�|�}W n ty' } z
| �t|�� W Y d }~nd }~ww t |t�s5| �|� d�� |S )Nz was not an array)r8 r9 r= �loads�
ValueError�failrA �r r� r� rg �v�er r r r� � �
��
zJsonArray.convertNr� r r r r r� � r� r� c @ r� )�
JsonObjectz JSON formatted object argument.z json objectc
C r� )Nz was not an object)r8 r; r= r� r� r� rA r� r r r r� � r� zJsonObject.convertNr� r r r r r� � r� r� c @ r� )�ISO8601DateTimez ISO 8601 Date Time argument.z iso-86091c
C s@ zt |�W S ttfy } z | �|� W Y d }~d S d }~ww r( )r � TypeErrorr� r� �r r� r� rg r� r r r r� � s
��zISO8601DateTime.convertNr� r r r r r� � r� r� c @ r� )�ISO8601DateTimeOrFloatz%ISO 8601 Date Time or float argument.ziso-86091 or floatc
C sb zt |�W S ttfy Y nw zt|�W S ttfy0 } z | �|� W Y d }~d S d }~ww r( )�floatr� r� r r� r� r r r r� � s
�
��z ISO8601DateTimeOrFloat.convertNr� r r r r r� � r� r� c rm )�LogLevelzLog level option.c s t � �d� dS )z:Initialize the log level option with the relevant choices.)�DEBUG�INFO�WARNINGr& �CRITICAL�FATALN)rq r r$ rs r r r
r% zLogLevel.__init__c s0 t |tj�r|S |�� }t� �|||�}t|�S r( )r8 �numbers�Integral�upperrq r� r r� rs r r r� s
zLogLevel.convert)rV rW rX rY r r� rx r r rs r r�
s r� )4rY r= r� � collectionsr � functoolsr �pprintr r) r �kombu.utils.objectsr �
celery._stater �celery.signalsr � celery.utilsr
�celery.utils.logr �celery.utils.timer �pygmentsr
�pygments.formattersr �pygments.lexersr � ImportErrorr? r@ r rl �Optionrn �Commandry r� r� r� r� r� r� �Choicer� �
JSON_ARRAY� JSON_OBJECT�ISO8601�ISO8601_OR_FLOAT� LOG_LEVEL�COMMA_SEPARATED_LISTr r r r �<module> sT �V