Question

我正在寻找通过Huey中的定期任务来实现Peewee的上下文管理器的最佳方法。普通任务具有很好的Huey.context_task（）装饰器，但对于周期性任务似乎没有类似的东西。

我是否正确地假设我只需要在定期任务中使用带（丑陋）的with语句？

Answer 1

应该能够执行以下操作：

from functools import wraps

huey = RedisHuey()
db = PostgresqlDatabase(...)

def db_periodic_task(*args, **kwargs):
    def decorator(fn):
        @wraps(fn)
        def new_task():
            with db:
                fn()
        return huey.periodic_task(*args, **kwargs)(new_task)
    return decorator

@db_periodic_task(crontab('0', '0'))
def my_periodic_task():
    # db will have been opened at start.
    # db will then be closed upon return.

在Huey中为periodic_tasks实现context_task的最佳方法是什么？

1 个答案: