peewee - 在 Huey 中为周期性任务实现 context_task 的最佳方法是什么？

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.