File: //usr/local/lib/python3.9/site-packages/celery/app/__pycache__/routes.cpython-39.pyc
a
X>h� � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZmZ dd lmZ z
ejZW n ey� ejZY n0 d
ZG d d � d �ZG d
d� d�Zdd� Zdd� ZdS )z[Task Routing.
Contains utilities for working with task routers, (:setting:`task_routes`).
� N)� OrderedDict)�Mapping)�Queue)�
QueueNotFound)�lpmerge)�maybe_evaluate�mlazy)�symbol_by_name)�MapRoute�Router�expand_router_string�preparec @ s e Zd ZdZdd� Zdd� ZdS )r
z(Creates a router out of a :class:`dict`.c C sv t |t�r|�� n|}i | _t� | _|D ]H\}}t |t�rF|| j|< q(d|v rf|| jt�t �
|��< q(|| j|< q(d S )N�*) �
isinstancer �items�mapr �patterns�Pattern�re�compile�fnmatch� translate)�selfr �k�v� r �;/usr/local/lib/python3.9/site-packages/celery/app/routes.py�__init__ s
zMapRoute.__init__c O s� zt | j| �W S ty" Y n ty@ d| j| i Y S 0 | j�� D ]@\}}|�|�rLzt |�W S ty� d|i Y S 0 qLd S �N�queue)�dictr �KeyError�
ValueErrorr r �match)r �name�args�kwargs�regex�router r r �__call__) s
zMapRoute.__call__N)�__name__�
__module__� __qualname__�__doc__r r) r r r r r
s r
c @ s> e Zd ZdZddd�Zddd�Zd d
� Zdd d �Zd
d� ZdS )r z8Route tasks based on the :setting:`task_routes` setting.NFc C s4 || _ |d u ri n|| _|d u r$g n|| _|| _d S �N)�app�queues�routes�create_missing)r r1 r0 r2 r/ r r r r ; s zRouter.__init__r c C sd |si n|}| � |�}| jrB| �|||||�}|rBt| � |�|�S d|vr`t| � | jjj�|�}|S r )�expand_destinationr1 � lookup_router r/ �confZtask_default_queue)r �optionsr$ r% r& � task_typer( r r r r( B s
��z Router.routec C st t |t�r|i }}n |�dd �}|rpt |t�r:||d< n6z| j| |d< W n" tyn td|�d���Y n0 |S )Nr zQueue z missing from task_queues)r �str�popr r0 r! r )r r( r r r r r3 N s
�
zRouter.expand_destinationc C s8 | j }| jD ]&}|||||||�}|d ur | S q d S r. )� query_routerr1 ) r r$ r% r& r6 r7 �query�routerr( r r r r4 b s
zRouter.lookup_routec C s2 t |�}t|d�r |�|||�S ||||||d�S )N�route_for_task)�task)r �hasattrr= )r r< r> r% r& r6 r7 r r r r: j s
zRouter.query_router)NNFN)r NN)NNNN) r* r+ r, r- r r( r3 r4 r: r r r r r 8 s �
�
r c C s t | �} t| d�r| � } | S )Nr= )r r? )r< r r r r r s
r c s: dd� � | du rdS t | ttf�s(| f} � fdd�| D �S )z*Expand the :setting:`task_routes` setting.c S s0 t | tttf�rt| �S t | t�r,tt| �S | S r. )r r �list�tupler
r8 r r )r( r r r � expand_route} s
z prepare.<locals>.expand_routeNr c s g | ] }� |��qS r r )�.0r( �rB r r �
<listcomp>� � zprepare.<locals>.<listcomp>)r r@ rA )r1 r rD r r
z s r
)r- r r � collectionsr �collections.abcr Zkombur Zcelery.exceptionsr Zcelery.utils.collectionsr Zcelery.utils.functionalr r Zcelery.utils.importsr Z
_pattern_typer �AttributeError�__all__r
r r r
r r r r �<module> s$
: