File: //usr/local/lib/python3.9/site-packages/celery/__pycache__/bootsteps.cpython-39.pyc
a
X>h�/ � @ s& d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
dd lmZmZ dd
lmZ zdd lmZ W n ey� d ZY n0 efZd
ZdZdZdZee�Z dd� Z dd� Z G dd� de
�Z G dd� d�Z G dd� de!�Z"G dd� de"d �Z#G d d � d e#�Z$G d d � d e$�Z%d!S )"z0A directed acyclic graph of reusable components.� )�deque)�Event)�
ignore_errors)� bytes_to_str��symbol_by_name� )�DependencyGraph�GraphFormatter)� instantiate�qualname)�
get_logger)� GreenletExit� )� Blueprint�Step�
StartStopStep� ConsumerStep� � c C s d| j � d|� �S )Nz| z: ��alias)�ns�fmtr r �:/usr/local/lib/python3.9/site-packages/celery/bootsteps.py�_pre s r c C s | j �dd�d S )N�.r ���)�name�rsplit)�sr r r �_label# s r! c @ sD e Zd ZdZdZdZdddd�Zdd � Zd
d � Zd d
� Z dd� Z
dS )�
StepFormatterz'Graph formatter for :class:`Blueprint`.u ⧉u ∘Z
parallelogramZ
slategray4Z
slategray3)�shape�colorZ fillcolorc C s, |o*d� | �|�t|jp t|��dd���S )Nz{}{}zutf-8�ignore)�format� _get_prefixr �labelr! �encode��self�stepr r r r( 2 s ��zStepFormatter.labelc C s |j r | jS |jr| jS dS )N� )�last�blueprint_prefix� conditional�conditional_prefixr* r r r r' 9 s
zStepFormatter._get_prefixc K s |j r | jn| j}| �|||�S �N)r. �blueprint_schemeZ node_schemeZ draw_node)r+ �obj�attrs�schemer r r �node@ s zStepFormatter.nodec K s& |j r|jddd� | �||| j|�S )N�noneZ
darkseagreen3)Z arrowheadr$ )r. �updateZ draw_edgeZ edge_scheme)r+ �a�br5 r r r �edgeD s zStepFormatter.edgeN) �__name__�
__module__� __qualname__�__doc__r/ r1 r3 r( r' r7 r< r r r r r" ' s �r" c @ s� e Zd ZdZeZdZdZdZe � Z
dde de de
diZd3dd �Zd
d � Zd d
� Zdd� Zdd� Zd4dd�Zd5dd�Zd6dd �Zd7d d �Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z e!d1d2� �Z"dS )8r a� Blueprint containing bootsteps that can be applied to objects.
Arguments:
steps Sequence[Union[str, Step]]: List of steps.
name (str): Set explicit name for this blueprint.
on_start (Callable): Optional callback applied after blueprint start.
on_close (Callable): Optional callback applied before blueprint close.
on_stopped (Callable): Optional callback applied after
blueprint stopped.
Nr Z initializing�running�closing� terminatingc C sT |p| j ptt| ��| _ t|p g �t| j�B | _|| _|| _|| _t � | _
i | _ d S r2 ) r r �type�set�
default_steps�types�on_start�on_close�
on_stoppedr �shutdown_complete�steps)r+ rL r rH rI rJ r r r �__init__c s zBlueprint.__init__c C sb t | _| jr| �� tdd� |jD ��D ]4\}}| �d|j� |d | _|�|� t �
d� q(d S )Nc s s | ]}|d ur|V qd S r2 r ��.0r r r r � <genexpr>q � z"Blueprint.start.<locals>.<genexpr>z Starting %sr z^-- substep ok) �RUN�staterH � enumeraterL �_debugr �started�start�logger�debug)r+ �parent�ir, r r r rW m s
zBlueprint.startc C s | j | jp d S �Nr )�
state_to_namerS �r+ r r r � human_statew s zBlueprint.human_statec C s( i }|j D ]}|�|�|�p i � q
|S r2 )rL r9 �info)r+ rZ r` r, r r r r` z s
zBlueprint.infoc C s$ | j r| � � | j|dddd� d S )N�closerB F)�reverse)rI �send_all�r+ rZ r r r ra � s zBlueprint.close�stop�
restartingFc C s | j ||||d� d S )N)� propagate)rc )r+ rZ �method� descriptionrg r r r �restart� s zBlueprint.restartTr c C s� |p|� dd�}|r t|j�n|j}|D ]�}|r(t||d �} | d ur(| �d|�� |j� z| |g|�R � W q( ty� }
z$|r�� t� d||j|
� W Y d }
~
q(d }
~
0 0 q(d S )N�_� z%s %s...zError on %s %s: %r)
�replace�reversedrL �getattrrU �
capitalizer � ExceptionrX � exception) r+ rZ rh ri rb rg �argsrL r, Zfun�excr r r rc � s"
�
�zBlueprint.send_allc C s� |rdnd}| j ttfv r d S | j tks8| jt|j�krLt| _ | j�� d S | � |� t| _ | j
||rjdnd|dd� | j r�| � � t| _ | j�� d S )NrC Zstopping� terminatere F)ri rg ) rS �CLOSE� TERMINATErR rV �lenrL rK rE ra rj rJ )r+ rZ ra ru �whatr r r re � s"
�zBlueprint.stopc C s* z| j j|d� W n ty$ Y n0 d S )N)�timeout)rK �wait�
IGNORE_ERRORS)r+ rz r r r �join� s zBlueprint.joinc K s� | � d� g }| _| �� }| _| � d� | �|�D ](}||fi |��}|||j< |�|� q6| � dd�dd� | jD ��� |D ]}|�|� q�| S )ar Apply the steps in this blueprint to an object.
This will apply the ``__init__`` and ``include`` methods
of each step, with the object as argument::
step = Step(obj)
...
step.include(obj)
For :class:`StartStopStep` the services created
will also be added to the objects ``steps`` attribute.
zPreparing bootsteps.zBuilding graph...zNew boot order: {%s}z, c s s | ] }|j V qd S r2 r rN r r r rP � rQ z"Blueprint.apply.<locals>.<genexpr>) rU �order� claim_stepsrL �_finalize_stepsr �appendr} �include)r+ rZ �kwargsr~ rL �Sr, r r r �apply� s
� zBlueprint.applyc C s8 | j j�|j j� | j �t|jd �t| jd �� d S )Nr r )�graph�adjacentr9 �add_edgerD r~ )r+ �otherr r r � connect_with� s zBlueprint.connect_withc C s
| j | S r2 )rL )r+ r r r r � __getitem__� s zBlueprint.__getitem__c C s t dd� | j�� D �d �S )Nc s s | ]}|j r|V qd S r2 )r. �rO �Cr r r rP � rQ z'Blueprint._find_last.<locals>.<genexpr>)�nextrL �valuesr^ r r r �
_find_last� s zBlueprint._find_lastc C sx |� � D ]}dd� |jD �|_qtdd� |� � D ��}|rt|�� D ].}t|�}|j| jvrd|||j< |�|j� qBq6d S )Nc S s g | ] }t |��qS r r )rO �depr r r �
<listcomp>� rQ z(Blueprint._firstpass.<locals>.<listcomp>c s s | ] }|j V qd S r2 ��requires�rO r, r r r rP � rQ z'Blueprint._firstpass.<locals>.<genexpr>)r� r� r �popleftr r rL r� )r+ rL r, �streamr7 r r r �
_firstpass� s
zBlueprint._firstpassc
C s� | � � }| �|� dd� |�� D �}t|| j|d�d� }| _|r`|D ]}||krF|�||� qFz
|�� W S ty� } ztd| ��W Y d }~n
d }~0 0 d S )Nc s s | ]}||j fV qd S r2 r� r� r r r rP � rQ z,Blueprint._finalize_steps.<locals>.<genexpr>)�root)� formatterzunknown bootstep: %s) r� r� r� r r
r� r� Ztopsort�KeyError)r+ rL r. �it�Gr4 rt r r r r� � s
�
zBlueprint._finalize_stepsc s t � fdd�� jD ��S )Nc 3 s | ]}� � |�V qd S r2 )� load_stepr� r^ r r rP � rQ z(Blueprint.claim_steps.<locals>.<genexpr>)�dictrG r^ r r^ r r � s zBlueprint.claim_stepsc C s t |�}|j|fS r2 )r r r* r r r r� � s zBlueprint.load_stepc G s t jt| |�g|�R � S r2 )rX rY r )r+ �msgrs r r r rU s zBlueprint._debugc C s t | �S r2 )r! r^ r r r r s zBlueprint.alias)NNNNN)re rf F)NTTr )TF)N)#r= r> r? r@ r" r
r rS rV rE rF rR rv rw r] rM rW r_ r` ra rj rc re r} r� r� r� r� r� r� r r� rU �propertyr r r r r r J sH � �
�
�
r c s8 e Zd ZdZdZdZ� fdd�Zdd� Zdd� Z� Z S ) �StepTypezMeta-class for steps.Nc sJ |� d�}|r |� d|� �n|}|j||� d�p2|d� t� �| |||�S )Nr> r r )r? r )�getr9 �super�__new__)�clsr �basesr5 �module�qname�� __class__r r r� s
�zStepType.__new__c C s | j S r2 )r �r� r r r �__str__ s zStepType.__str__c C s
d� | �S )Nz step:{0.name}{{{0.requires!r}}})r& r� r r r �__repr__ s zStepType.__repr__)
r= r> r? r@ r r� r� r� r� �
__classcell__r r r� r r�
s r� c @ st e Zd ZdZdZdZdZdZdZdZ dd� Z
dd � Z d
d � Z d d
� Z
dd� Zdd� Zdd� Zedd� �Zdd� ZdS )r z�A Bootstep.
The :meth:`__init__` method is called when the step
is bound to a parent object, and can as such be used
to initialize attributes in the parent object at
parent instantiation-time.
NFr Tc K s d S r2 r )r+ rZ r� r r r rM ? s z
Step.__init__c C s | j S )z�Return true if bootstep should be included.
You can define this as an optional predicate that decides whether
this step should be created.
)�enabledrd r r r �
include_ifB s zStep.include_ifc O s t |g|�R i |��S r2 )r )r+ r rs r� r r r r J s zStep.instantiatec C s | � |�rd| �|�fS dS )NT)FN)r� �createrd r r r �_should_includeM s
zStep._should_includec C s | � |�d S r\ )r� rd r r r r� R s z Step.includec C s dS )zCreate the step.Nr rd r r r r� U s z Step.createc C s d| j � d�S )Nz<step: �>r r^ r r r r� X s z
Step.__repr__c C s | j p t| �S r2 )r( r! r^ r r r r [ s z
Step.aliasc C s d S r2 r )r+ r4 r r r r` _ s z Step.info)r= r> r? r@ r r( r0 r� r. r� rM r� r r� r� r� r� r� r r` r r r r r s"
r )� metaclassc @ s< e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z dS )
r z3Bootstep that must be started and stopped in order.Nc C s | j r| j �� S d S r2 )r4 rW rd r r r rW k s zStartStopStep.startc C s | j r| j �� S d S r2 )r4 re rd r r r re o s zStartStopStep.stopc C s d S r2 r rd r r r ra s s zStartStopStep.closec C s | j rt| j d| j j�� S d S )Nru )r4 ro re rd r r r ru v s zStartStopStep.terminatec C s( | � |�\}}|r$|| _|j�| � |S r2 )r� r4 rL r� )r+ rZ �inc�retr r r r� z s
zStartStopStep.include)
r= r> r? r@ r4 rW re ra ru r� r r r r r c s r c @ sB e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z dd
d�Z
dS )r z(Bootstep that starts a message consumer.)z!celery.worker.consumer:ConnectionNc C s t d��d S )Nzmissing get_consumers)�NotImplementedError)r+ �channelr r r �
get_consumers� s zConsumerStep.get_consumersc C s2 |j �� }| �|�| _| jp g D ] }|�� q d S r2 )�
connectionr� r� � consumersZconsume)r+ �cr� �consumerr r r rW � s
zConsumerStep.startc C s | � |d� d S )NT��_close�r+ r� r r r re � s zConsumerStep.stopc C s | � |d� d S )NFr� r� r r r �shutdown� s zConsumerStep.shutdownTc C sV t � }| jpg D ](}|r&t|j|j� |jr|�|j� q|D ]}t|j|j� q>d S r2 )rE r� r r� �cancelr� �addra )r+ r� Zcancel_consumersZchannelsr� r� r r r r� � s zConsumerStep._close)T) r= r> r? r@ r� r� r� rW re r� r� r r r r r � s r N)&r@ � collectionsr � threadingr Z kombu.commonr Zkombu.utils.encodingr Zkombu.utils.importsr Z utils.graphr r
Z
utils.importsr r Z utils.logr
Zgreenletr � ImportErrorr| �__all__rR rv rw r= rX r r! r" r rD r� r r r r r r r �<module> s6
# AC