File: //home/arjun/projects/env/lib/python3.10/site-packages/celery/__pycache__/bootsteps.cpython-310.pyc
o
-we�/ � @ s& d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
dd lmZmZ dd
lmZ zdd lmZ W n eyI d ZY nw 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 �I/home/arjun/projects/env/lib/python3.10/site-packages/celery/bootsteps.py�_pre � r c C s | j �dd�d S )N�.r ���)�name�rsplit)�sr r r �_label# r r" c @ sD e Zd ZdZdZdZdddd�Zdd � Zd
d � Zd d
� Z dd� Z
dS )�
StepFormatterz'Graph formatter for :class:`Blueprint`.u ⧉u ∘�
parallelogram�
slategray4�
slategray3)�shape�color� fillcolorc C s, |od� | �|�t|jpt|��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)r3 �blueprint_scheme� node_scheme� draw_node)r0 �obj�attrs�schemer r r �node@ s zStepFormatter.nodec K s& |j r
|jddd� | �||| j|�S )N�none�
darkseagreen3)� arrowheadr( )r3 �update� draw_edge� edge_scheme)r0 �a�br< r r r �edgeD s zStepFormatter.edgeN) �__name__�
__module__� __qualname__�__doc__r4 r6 r8 r- r, r> rG r r r r r# ' s � r# c @ s� e Zd ZdZeZdZdZdZe � Z
dde de de
diZ d3dd �Zd
d � Zd d
� Zdd� Zdd� Z d4dd�Z d5dd�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 � initializing�running�closing� terminatingc C sT |p
| j p
tt| ��| _ t|pg �t| j�B | _|| _|| _|| _t � | _
i | _ d S r7 ) r r �type�set�
default_steps�types�on_start�on_close�
on_stoppedr �shutdown_complete�steps)r0 rX r rT rU rV r r r �__init__c s
zBlueprint.__init__c C sb t | _| jr
| �� tdd� |jD ��D ]\}}| �d|j� |d | _|�|� t �
d� qd S )Nc s s � | ] }|d ur|V qd S r7 r ��.0r! r r r � <genexpr>q s � z"Blueprint.start.<locals>.<genexpr>z Starting %sr z^-- substep ok) �RUN�staterT � enumeraterX �_debugr �started�start�logger�debug)r0 �parent�ir1 r r r rb m s
�zBlueprint.startc C s | j | jpd S �Nr )�
state_to_namer^ �r0 r r r � human_statew � zBlueprint.human_statec C s( i }|j D ] }|�|�|�pi � q|S r7 )rX rB �info)r0 re rl r1 r r r rl z s
zBlueprint.infoc C s$ | j r| � � | j|dddd� d S )N�closerN F)�reverse)rU �send_all�r0 re r r r rm � s zBlueprint.close�stop�
restartingFc C s | j ||||d� d S )N)� propagate)ro )r0 re �method� descriptionrs r r r �restart� s zBlueprint.restartTr c C s� |p|� dd�}|rt|j�n|j}|D ]A}|rUt||d �} | d urU| �d|�� |j� z
| |g|�R � W q tyT }
z|rA� t� d||j|
� W Y d }
~
qd }
~
ww qd S )N�_� z%s %s...zError on %s %s: %r)
�replace�reversedrX �getattrr` �
capitalizer � Exceptionrc � exception) r0 re rt ru rn rs �argsrX r1 �fun�excr r r ro � s*
�
�����zBlueprint.send_allc C s� |rdnd}| j ttfv rd S | j tks | jt|j�kr&t| _ | j�� d S | � |� t| _ | j
||r5dnd|dd� | j rB| � � t| _ | j�� d S )NrO �stopping� terminaterq F)ru rs ) r^ �CLOSE� TERMINATEr] ra �lenrX rW rQ rm rv rV )r0 re rm r� �whatr r r rq � s"
�zBlueprint.stopc C s* z
| j j|d� W d S ty Y d S w )N)�timeout)rW �wait�
IGNORE_ERRORS)r0 r� r r r �join� s
�zBlueprint.joinc K s� | � d� g }| _| �� }| _| � d� | �|�D ]}||fi |��}|||j< |�|� q| � dd�dd� | jD ��� |D ]}|�|� qA| 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 � � | ]}|j V qd S r7 r rZ r r r r\ � � � z"Blueprint.apply.<locals>.<genexpr>) r` �order� claim_stepsrX �_finalize_stepsr �appendr� �include)r0 re �kwargsr� rX �Sr1 r r r �apply� s
� zBlueprint.applyc C s8 | j j�|j j� | j �t|jd �t| jd �� d S )Nr r )�graph�adjacentrB �add_edgerP r� )r0 �otherr r r � connect_with� s &zBlueprint.connect_withc C s
| j | S r7 )rX )r0 r r r r � __getitem__� �
zBlueprint.__getitem__c C s t dd� | j�� D �d �S )Nc s s � | ]}|j r|V qd S r7 )r3 �r[ �Cr r r r\ � � � z'Blueprint._find_last.<locals>.<genexpr>)�nextrX �valuesri r r r �
_find_last� s zBlueprint._find_lastc C s~ |� � D ] }dd� |jD �|_qtdd� |� � D ��}|r=|�� D ]}t|�}|j| jvr2|||j< |�|j� q!|s d S d S )Nc S s g | ]}t |��qS r r )r[ �depr r r �
<listcomp>� s z(Blueprint._firstpass.<locals>.<listcomp>c s r� r7 ��requires�r[ r1 r r r r\ � r� z'Blueprint._firstpass.<locals>.<genexpr>)r� r� r �popleftr r rX r� )r0 rX r1 �streamr> r r r �
_firstpass� s
�zBlueprint._firstpassc
C s� | � � }| �|� dd� |�� D �}t|| j|d�d� }| _|r0|D ] }||kr/|�||� q#z|�� W S tyG } ztd| ��d }~ww )Nc s s � | ]}||j fV qd S r7 r� r� r r r r\ � r� z,Blueprint._finalize_steps.<locals>.<genexpr>)�root)� formatterzunknown bootstep: %s) r� r� r� r r
r� r� �topsort�KeyError)r0 rX r3 �it�Gr; r� r r r r� � s"
� �
��zBlueprint._finalize_stepsc s t � fdd�� jD ��S )Nc 3 s � | ]}� � |�V qd S r7 )� load_stepr� ri r r r\ � r� z(Blueprint.claim_steps.<locals>.<genexpr>)�dictrS ri r ri r r� � � zBlueprint.claim_stepsc C s t |�}|j|fS r7 )r r r/ r r r r� � s
zBlueprint.load_stepc G s t jt| |�g|�R � S r7 )rc rd r )r0 �msgr r r r r` r� zBlueprint._debugc C s t | �S r7 )r" ri r r r r s zBlueprint.alias)NNNNN)rq rr F)NTTr )TFr7 )#rH rI rJ rK r# r
r r^ ra rQ rR r] r� r� rh rY rb rj rl rm rv ro rq r� r� r� r� r� r� r� r� r� r` �propertyr r r r r r J sJ �
�
�
�
r c s8 e Zd ZdZdZdZ� fdd�Zdd� Zdd� Z� Z S ) �StepTypezMeta-class for steps.Nc sJ |� d�}|r|� d|� �n|}|j||� d�p|d� t� �| |||�S )NrI r r )rJ r )�getrB �super�__new__)�clsr �basesr< �module�qname�� __class__r r r� s
�zStepType.__new__c C � | j S r7 )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__ r� zStepType.__repr__)
rH rI rJ rK 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 � d S r7 r )r0 re r� r r r rY ? � z
Step.__init__c C r� )z�Return true if bootstep should be included.
You can define this as an optional predicate that decides whether
this step should be created.
)�enabledrp r r r �
include_ifB s zStep.include_ifc O s t |g|�R i |��S r7 )r )r0 r r r� r r r r J s zStep.instantiatec C s | � |�r d| �|�fS dS )NT)FN)r� �createrp r r r �_should_includeM s
zStep._should_includec C s | � |�d S rg )r� rp r r r r� R � z Step.includec C s dS )zCreate the step.Nr rp r r r r� U s z Step.createc C s d| j � d�S )Nz<step: �>r ri r r r r� X r� z
Step.__repr__c C s | j pt| �S r7 )r- r" ri r r r r [ s z
Step.aliasc C r� r7 r )r0 r; r r r rl _ r� z Step.info)rH rI rJ rK r r- r5 r� r3 r� rY r� r r� r� r� r� r� r rl 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 � | j r| j �� S d S r7 )r; rb rp r r r rb k �
�zStartStopStep.startc C r� r7 )r; rq rp r r r rq o r� zStartStopStep.stopc C r� r7 r rp r r r rm s r� zStartStopStep.closec C s | j r
t| j d| j j�� S d S )Nr� )r; r{ rq rp r r r r� v s �zStartStopStep.terminatec C s( | � |�\}}|r|| _|j�| � |S r7 )r� r; rX r� )r0 re �inc�retr r r r� z s
zStartStopStep.include)
rH rI rJ rK r; rb rq rm r� 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��)Nzmissing get_consumers)�NotImplementedError)r0 �channelr r r �
get_consumers� s zConsumerStep.get_consumersc C s2 |j �� }| �|�| _| jpg D ]}|�� qd S r7 )�
connectionr� r� � consumers�consume)r0 �cr� �consumerr r r rb � s
�zConsumerStep.startc C � | � |d� d S )NT��_close�r0 r� r r r rq � rk zConsumerStep.stopc C r� )NFr� r� r r r �shutdown� rk zConsumerStep.shutdownTc C sV t � }| jpg D ]}|rt|j|j� |jr |�|j� q|D ] }t|j|j� q d S r7 )rQ r� r r� �cancelr� �addrm )r0 r� �cancel_consumers�channelsr� r� r r r r� � s ��zConsumerStep._close)T) rH rI rJ rK r� r� r� rb rq r� r� r r r r r � s r N)&rK � collectionsr � threadingr � kombu.commonr �kombu.utils.encodingr �kombu.utils.importsr � utils.graphr r
�
utils.importsr r � utils.logr
�greenletr � ImportErrorr� �__all__r] r� r� rH rc r r" r# r rP r� r r r r r r r �<module> s: �# AC