具有多个装饰器的芹菜任务不会自动注册任务名称

Question

我有一个看起来像这样的任务

from mybasetask_module import MyBaseTask
@task(base=MyBaseTask)
@my_custom_decorator 
def my_task(*args, **kwargs):
    pass

我的基本任务看起来像这样

from celery import task, Task
class MyBaseTask(Task):
    abstract = True
    default_retry_delay = 10 
    max_retries = 3 
    acks_late = True

我遇到的问题是芹菜工人正在使用名称注册任务

'mybasetask_module.__inner'

当我从任务中删除@my_custom_decorator或如果我像这样为任务提供显式名称时，任务注册得很好（即包+模块+函数）

from mybasetask_module import MyBaseTask
@task(base=MyBaseTask, name='an_explicit_task_name')
@my_custom_decorator 
def my_task(*args, **kwargs):
    pass

这种行为是预期的吗？当我有多个装饰器但没有显式任务名称时，我是否需要执行某些操作，以便在第一种情况下使用默认的自动注册名称注册我的任务？

谢谢

Answer 1

使用 functools.wraps() 修饰器来确保 my_custom_decorator 返回的包装器具有正确的名称：

from functools import wraps
def my_custom_decorator(func):
    @wraps(func)
    def __inner():
        return func()
    return __inner

任务名称取自task装饰器包装的函数调用，但通过在两者之间插入装饰器，您task __inner包装函数。functools.wraps()装饰器将所有必要的元数据从func复制到包装器，以便task()可以选择正确的名称。