如何在Django中基于Transaction的execution_date过滤prefetch_related关联数据

模型定义（补全关联关系） #

首先补全PersonData与Person的关联字段（逻辑上必须存在，否则无法关联Transaction与PersonData）：

from django.db import models
from datetime import date

class Person(models.Model):
    birth_date = models.DateField()

class PersonData(models.Model):
    person = models.ForeignKey(Person, on_delete=models.CASCADE, related_name='data_entries')
    last_name = models.CharField(max_length=255)
    valid_from = models.DateField()  # 默认值为关联Person的birth_date
    valid_to = models.DateField(default=date(year=2999, month=12, day=31))
    
class Transaction(models.Model): 
    person = models.ForeignKey(Person, on_delete=models.RESTRICT)
    execution_date = models.DateField()
    # 其他字段...

需求说明 #

需要筛选出execution_date ≤ context['context_date']的Transaction数据，同时预取关联的PersonData时，基于每条Transaction自身的execution_date过滤，只保留满足valid_from ≤ execution_date ≤ valid_to的PersonData条目。

解决方案 #

利用Django的Prefetch对象结合OuterRef实现跨层级的字段引用，具体代码如下：

from django.db.models import Prefetch, OuterRef

# 1. 基础Transaction筛选：execution_date不晚于context_date
base_query = Transaction.objects.filter(execution_date__lte=context['context_date'])

# 2. 构建PersonData的过滤查询，引用外层Transaction的execution_date
filtered_person_data = PersonData.objects.filter(
    person=OuterRef('person'),
    valid_from__lte=OuterRef('execution_date'),
    valid_to__gte=OuterRef('execution_date')
)

# 3. 预取过滤后的PersonData，指定自定义属性名方便访问
result = base_query.prefetch_related(
    Prefetch(
        'person__data_entries',
        queryset=filtered_person_data,
        to_attr='filtered_data'
    )
)

关键说明 #

• OuterRef用于引用外层查询（Transaction）的字段，解决了直接用F('execution_date')只能引用当前PersonData查询集字段的问题。
• 通过to_attr指定自定义属性名（如filtered_data），后续可通过transaction.person.filtered_data直接获取当前Transaction对应的有效PersonData条目。
• 确保PersonData模型中person字段的related_name设置正确（示例中为data_entries），否则需要调整Prefetch中的关联路径。

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