Question

可能重复：
How can I find all subclasses of a given class in Python?

在我的Django项目中，我有一些Celery的Task和PeriodicTask的子类：

class CustomTask(Task):
    # stuff

class CustomPeriodicTask(PeriodicTask):
    # stuff

我需要所有Task类来添加一些自定义日志记录配置。所以我认为我可以__subclasses__，但这不起作用：

>>> Task.__subclasses__()
[<unbound PeriodicTask>, <class handle_register of <Celery default:0xa1cc3cc>>]

以某种方式可以动态地获取我的所有Task和PeriodicTask子类吗？

Answer 1

Celery维护着所有任务的注册表。这是工作人员在收到任务消息时可以按名称查找任务的方式：

from celery import current_app
all_task_names = current_app.tasks.keys()
all_tasks = current_app.tasks.values()
foo_task = current_app.tasks['tasks.foo']

all_task_classes = [type(task) for task in current_app.tasks.itervalues()]

仅在导入包含任务的模块时填充任务注册表。如果您没有导入所有模块，您可以像芹菜工作者那样进行，并且导入所有已配置的任务模块源：

current_app.loader.import_default_modules()

注意：在Celery 2.5之前不存在import_default_modules，那时你必须这样做做：

from celery.loaders import current_loader
current_loader().init_worker()

注意2：您确定要编辑这样的所有任务类吗？芹菜附带一个号码配置任务类的方法，例如：

CELERY_ANNOTATIONS = {
     '*': {
         'loglevel': logging.DEBUG,
         'logger': logging.getLogger('foo'),
     },
}

将设置所有任务的“loglevel”和“logger”属性。或者，您可以使用公共基类：

class MyTask(Task):
    abstract = True   # means this base task won't be registered
    loglevel = logging.DEBUG
    logger = logging.getLogger('foo'),

如何在Celery中获取所有任务和定期任务

1 个答案: