如何用django-recurrence在Django中处理重复训练的日期筛选与排序？

问题背景 #

我正在开发一个健身房网站，用django-recurrence处理重复训练课程，但不确定如何在Django QuerySet中处理重复日期：

• 想按下次即将到来的日期对训练进行排序
• 想按特定日期筛选训练（比如显示选定日期的所有课程）

我考虑过给Training模型加next_training_date字段，用Celery定时任务在每次课程结束后更新这个字段，但觉得这种方式开销太大，没必要为了筛选排序搞这么复杂。想知道用django-recurrence实现需求的简洁高效方法，以及Django中有没有更好的重复事件处理模式。

附我的Training模型代码：

class Training(TimestampModel):
    template = models.ForeignKey(TrainingTemplate, on_delete=models.CASCADE, related_name='trainings')
    start_time = models.TimeField()
    end_time = models.TimeField()
    location = models.ForeignKey(TrainingLocation, on_delete=models.PROTECT, related_name='trainings')
    recurrences = RecurrenceField()
    members = models.ManyToManyField(Member, through=MemberTraining, related_name='trainings', blank=True)

    objects = TrainingManager()

解决方案 #

1. 按特定日期筛选课程 #

django-recurrence的RecurrenceField存储的是重复规则，数据库无法直接解析Python层面的重复逻辑，所以可以在自定义TrainingManager里实现筛选逻辑：

from recurrence import recurrence_from_dict
from datetime import date, datetime

class TrainingManager(models.Manager):
    def on_date(self, target_date):
        trainings = self.all()
        matching_ids = []
        target_start = datetime.combine(target_date, datetime.min.time())
        target_end = datetime.combine(target_date, datetime.max.time())
        
        for training in trainings:
            # 获取目标日期范围内的所有重复事件
            occurrences = training.recurrences.occurrences(start=target_start, end=target_end)
            if list(occurrences):
                matching_ids.append(training.pk)
        
        return self.filter(pk__in=matching_ids)

使用示例：

# 获取2024年5月20日的所有课程
target_date = date(2024, 5, 20)
trainings_on_date = Training.objects.on_date(target_date)

2. 按下次即将到来的日期排序 #

数据库无法直接计算每个训练的下一次发生时间，可先在模型中添加方法获取下一次事件时间，再在Python层面排序：

from datetime import datetime, timedelta

class Training(TimestampModel):
    # 原模型字段...
    
    def get_next_occurrence(self):
        now = datetime.now()
        next_occur_date = self.recurrences.next_occurrence(now)
        if next_occur_date:
            # 结合start_time得到完整的开始时间
            return datetime.combine(next_occur_date.date(), self.start_time)
        return None

排序使用示例：

trainings = Training.objects.all()
# 过滤无后续事件的训练，再按下次发生时间排序
sorted_trainings = sorted(
    [t for t in trainings if t.get_next_occurrence()],
    key=lambda x: x.get_next_occurrence()
)

如果数据量较大、排序需求频繁，可以考虑轻量预计算方案：

• 在训练的save方法中计算并更新next_training_date字段
• 每天用Django的django-cron跑一次批量更新脚本，刷新所有训练的下次事件时间，避免Celery任务的高开销

3. 替代方案：数据库原生重复事件处理 #

如果django-recurrence的效率无法满足需求，可尝试：

• 用PostgreSQL的daterange和generate_series函数，直接在数据库层生成、筛选重复事件
• 使用django-orm-recurrence这类库，将重复规则转换为数据库可查询的表达式，支持直接在QuerySet中筛选和排序

总结 #

• 小数据量场景：直接在Python层面计算重复事件，实现筛选和排序，简单易维护
• 大数据量场景：预计算next_training_date字段配合轻量定时更新，或使用支持数据库层查询的重复事件库

内容的提问来源于stack exchange，提问作者Dmytro Hryshchenko