python - 如何使用 celery 安排在月底运行的任务？

Question

我需要将 celery crontab 设置为在月底运行，甚至与月底不同的一天(28~31)运行。我知道如何在 shell 命令上设置 crontab 在月底运行，如下所示:

55　23　28-31 * * /usr/bin/test $( date -d '+1 day' +%d ) -eq 1 && exec something

但是在 celery 时间表上我不知道如何进行此设置。有什么方法可以安排在 celery 上月底运行的任务吗？
似乎唯一的方法是重写 celery.schedules.crontab 上的 is_due 方法。

Answer 1

如果您不介意一点开销 - 您可以将任务设置为每天在 CELERYBEAT_SCHEDULE 中运行。

然后在任务本身中，您可以检查这一天是否是该月的最后一天:

import calendar
from datetime import datetime

@task
def task_to_run_at_end_of_month():
    today = datetime.today()
    day_in_month = today.day
    month = today.month
    year = today.year
    day_of_week, number_of_days_in_month = calendar.monthrange(year, month)
    if day_in_month != number_of_days_in_month:
        # not last day of month yet, do nothing
        return
    # process stuff on last day of month
    ...