| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- # -*- coding: utf-8 -*-
- """
- celery.app.registry
- ~~~~~~~~~~~~~~~~~~~
-
- Registry of available tasks.
-
- """
- from __future__ import absolute_import
-
- import inspect
-
- from importlib import import_module
-
- from celery._state import get_current_app
- from celery.exceptions import NotRegistered
- from celery.five import items
-
- __all__ = ['TaskRegistry']
-
-
- class TaskRegistry(dict):
- NotRegistered = NotRegistered
-
- def __missing__(self, key):
- raise self.NotRegistered(key)
-
- def register(self, task):
- """Register a task in the task registry.
-
- The task will be automatically instantiated if not already an
- instance.
-
- """
- self[task.name] = inspect.isclass(task) and task() or task
-
- def unregister(self, name):
- """Unregister task by name.
-
- :param name: name of the task to unregister, or a
- :class:`celery.task.base.Task` with a valid `name` attribute.
-
- :raises celery.exceptions.NotRegistered: if the task has not
- been registered.
-
- """
- try:
- self.pop(getattr(name, 'name', name))
- except KeyError:
- raise self.NotRegistered(name)
-
- # -- these methods are irrelevant now and will be removed in 4.0
- def regular(self):
- return self.filter_types('regular')
-
- def periodic(self):
- return self.filter_types('periodic')
-
- def filter_types(self, type):
- return dict((name, task) for name, task in items(self)
- if getattr(task, 'type', 'regular') == type)
-
-
- def _unpickle_task(name):
- return get_current_app().tasks[name]
-
-
- def _unpickle_task_v2(name, module=None):
- if module:
- import_module(module)
- return get_current_app().tasks[name]
|
