django celery中动态添加周期任务

常用实现方式

@periodic_task(run_every=crontab(minute='*/5', hour='*', day_of_week="*"))
def get_time():
    """
    celery 周期任务示例

    run_every=crontab(minute='*/5', hour='*', day_of_week="*")：每 5 分钟执行一次任务
    """
    now = datetime.datetime.now()
    logger.error(u"celery 周期任务调用成功，当前时间：{}".format(now))

这样实现方式无法满足前端页面的动态修改或添加周期任务的要求
我们对django celery 周期任务进行二次开发

celery的models

PeriodicTask 模型定义要运行的周期性任务
PeriodicTasks 模型仅用作索引以跟踪计划何时更改
IntervalSchedule 以特定间隔（例如，每5秒）运行的计划。
CrontabSchedule 以特定时刻表（例如，每天早晨8点）运行的计划。

1. 创建时刻表

# 第一种
# 创建基于间隔时间的周期性任务，每多少分钟/小时/天执行的周期任务
# PERIOD_CHOICES = (('days', _('Days')), # 固定间隔天数
                  ('hours', _('Hours')), # 固定间隔小时数
                  ('minutes', _('Minutes')),# 固定间隔分钟数
                  ('seconds', _('Seconds')),
                  ('microseconds', _('Microseconds')))
schedule, _ = IntervalSchedule.objects.get_or_create(every=cycle_time, period="minutes")
# 第二种
# 创建基于 crontab 的周期性任务
crontab, _ = CrontabSchedule.objects.get_or_create(minute=f"*/{cycle_time}")

2. 创建任务

#第一种
PeriodicTask.objects.update_or_create(name=f"check_task.{check_task_id}", defaults={
            "task": "home_application.check.tasks.now_run",
            "args": json.dumps([check_task_id, username]),
            "interval": schedule
        })
#第二种
PeriodicTask.objects.update_or_create(name=f"check_task.{check_task_id}", defaults={
            "task": "home_application.check.tasks.now_run",
            "args": json.dumps([check_task_id, username]),
            "crontab": crontab
        })

在数据库中添加了PeriodicTask后 celery beat 会周期性的遍历中的任务 发送到任务队列
[2021-06-21 14:54:05,212: INFO/MainProcess] Scheduler: Sending due task check_task.15 (home_application.check.tasks.now_run)
3 删除任务

PeriodicTask.objects.filter(name=name).delete()

4 暂停任务

PeriodicTask.objects.filter(name=name).update(enabled=False)

celery的 workers 和 beat 定时服务都需要同时开启。