Django ORM多InactivePeriod场景下错误排除Person的问题咨询

背景与模型 #

使用以下简化的Django模型：

class Person(models.Model):
    name = models.CharField()

class InactivePeriod(models.Model):
    person = models.ForeignKeyField(Person)
    start_date = models.DateField()
    end_date = models.DateField()

class Template(models.Model):
    day = models.CharField() # 可选值：monday, tuesday等
    person = models.ForeignKeyField(Person)

class TheList(models.Model):
    person = models.ForeignKeyField(Person)
    date = models.DateField(default=datetime.today)
    attended = models.BooleanField(default=False)

业务流程：用户通过电话报名每日活动，原本需手动创建TheList记录；为减少操作，新增Template表标记固定日期的常客，通过定时任务自动生成当日TheList记录，同时支持标记Person的InactivePeriod（当日处于该时段则排除该用户）。

问题 #

当前定时任务代码逻辑存在错误：当某Person存在两个均不包含今日的InactivePeriod时，其Template记录被错误排除；仅保留一个InactivePeriod时查询正常。代码如下：

def handle(self, *args, **options):
    today = datetime.today()
    day = str(today.weekday()) # 存为字符串对应charfield类型

    is_inactive_today = (Q(person__inactiveperiod__start_date__lte=today) & \
                         Q(person__inactiveperiod__end_date__gte=today))

    attendance_list = []
    for t in models.Template.objects.filter(day=day).exclude(is_inactive_today):
        attendance_list.append(models.TheList(person=t.person, date=today))
    models.TheList.objects.bulk_create(attendance_list)

示例场景：今日为2024-10-11，某Person的两个InactivePeriod为2024-10-09至2024-10-10、2024-10-12至2024-10-13，均不包含今日，但该用户的Template仍被排除。

原因分析 #

原Q表达式逻辑有误：Q(person__inactiveperiod__start_date__lte=today) & Q(person__inactiveperiod__end_date__gte=today)会匹配不同的InactivePeriod对象——只要用户存在任意一个时段的start_date<=今日，且存在任意一个时段的end_date>=今日（即使是两个不重叠的时段），就会被误判为今日处于无效时段。

解决方案 #

无需额外循环逻辑，直接通过Django ORM修正查询逻辑即可，推荐使用Exists子查询（性能更优）：

from django.db.models import Exists, OuterRef
from datetime import datetime

def handle(self, *args, **options):
    today = datetime.today().date()  # 转为date类型，匹配模型字段类型
    day = str(today.weekday())

    # 子查询：检查当前Person是否存在包含今日的InactivePeriod
    has_active_inactive_period = Exists(
        InactivePeriod.objects.filter(
            person=OuterRef('person'),
            start_date__lte=today,
            end_date__gte=today
        )
    )

    # 筛选今日模板，排除存在有效无效时段的用户
    templates = models.Template.objects.filter(day=day).exclude(has_active_inactive_period)

    # 批量创建记录
    attendance_list = [models.TheList(person=t.person, date=today) for t in templates]
    models.TheList.objects.bulk_create(attendance_list)

也可以使用嵌套Q表达式实现相同逻辑：

from django.db.models import Q

is_inactive_today = Q(person__inactiveperiod__Q(start_date__lte=today, end_date__gte=today))
templates = models.Template.objects.filter(day=day).exclude(is_inactive_today)

额外优化点 #

• 将datetime.today()转为date()类型，避免datetime与date字段的类型匹配问题；
• 用列表推导式替代循环append，代码更简洁。

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