File: //usr/local/lib/python3.9/site-packages/celery/app/__pycache__/registry.cpython-39.pyc
a
X>h� � @ sj d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z m
Z
dZ G dd � d e �Z
d
d � Zdd d
�ZdS )z Registry of available tasks.� N)�
import_module)�get_current_app)�add_autoretry_behaviour)�InvalidTaskError�
NotRegistered)� TaskRegistryc @ sD e Zd ZdZeZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dS )r zMap of registered tasks.c C s | � |��d S �N)r )�self�key� r �=/usr/local/lib/python3.9/site-packages/celery/app/registry.py� __missing__ s zTaskRegistry.__missing__c C sH |j du r td�t|�j���t�|�r.|� p0|}t|� || |j < dS )z�Register a task in the task registry.
The task will be automatically instantiated if not already an
instance. Name must be configured prior to registration.
Nz,Task class {!r} must specify .name attribute)�namer �format�type�__name__�inspect�isclassr )r �taskr r r �register s
��zTaskRegistry.registerc C s8 z| � t|d|�� W n ty2 | �|��Y n0 dS )a Unregister task by name.
Arguments:
name (str): name of the task to unregister, or a
:class:`celery.app.task.Task` with a valid `name` attribute.
Raises:
celery.exceptions.NotRegistered: if the task is not registered.
r N)�pop�getattr�KeyErrorr )r r r r r �
unregister" s
zTaskRegistry.unregisterc C s
| � d�S )N�regular�� filter_types�r r r r r 2 s zTaskRegistry.regularc C s
| � d�S )N�periodicr r r r r r 5 s zTaskRegistry.periodicc s � fdd�| � � D �S )Nc s& i | ] \}}t |d d�� kr||�qS )r r )r )�.0r r �r r r �
<dictcomp>9 s �z-TaskRegistry.filter_types.<locals>.<dictcomp>)�items)r r r r r r 8 s zTaskRegistry.filter_typesN) r �
__module__� __qualname__�__doc__r r
r r r r r r r r r r s r c C s t � j| S r )r �tasks)r r r r �_unpickle_task= s r' c C s |r t |� t� j| S r )r r r& )r �moduler r r �_unpickle_task_v2A s r) )N)r% r � importlibr Z
celery._stater Zcelery.app.autoretryr Zcelery.exceptionsr r �__all__�dictr r' r) r r r r �<module> s 1