/usr/share/pyshared/celery/registry.py is in python-celery 2.5.3-4.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | # -*- coding: utf-8 -*-
"""
celery.registry
~~~~~~~~~~~~~~~
Registry of available tasks.
:copyright: (c) 2009 - 2012 by Ask Solem.
:license: BSD, see LICENSE for more details.
"""
from __future__ import absolute_import
import inspect
from .exceptions import NotRegistered
class TaskRegistry(dict):
NotRegistered = NotRegistered
def regular(self):
"""Get all regular task types."""
return self.filter_types("regular")
def periodic(self):
"""Get all periodic task types."""
return self.filter_types("periodic")
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:
# Might be a task class
name = name.name
except AttributeError:
pass
self.pop(name)
def filter_types(self, type):
"""Return all tasks of a specific type."""
return dict((name, task) for name, task in self.iteritems()
if task.type == type)
def pop(self, key, *args):
try:
return dict.pop(self, key, *args)
except KeyError:
raise self.NotRegistered(key)
#: Global task registry.
tasks = TaskRegistry()
def _unpickle_task(name):
return tasks[name]
|