Django中如何预取关联集合以避免循环内重复数据库查询？

问题背景 #

你的模型定义如下：

class Convocacao(models.Model):
    cursos = models.ForeignKey(Cursos)

class RegistroConvocacao(models.Model):
    convocacao = models.ForeignKey(Convocacao)

你通过prefetch_related预取了Convocacao对象的cursos和registroconvocacao_set，但在遍历cursos时调用obj.registroconvocacao_set.filter()仍会触发重复数据库查询——原因是registroconvocacao_set是Django的RelatedManager，调用filter()会生成新的QuerySet，不会复用预取到内存中的数据，因此每次循环都会发起新查询。

优化方案 #

方案1：将预取数据转存到内存列表，在内存中过滤

先把预取的RegistroConvocacao对象一次性转成内存列表，之后循环时直接在列表中筛选，避免触发新查询：

# 获取对象时保持原预取逻辑
obj = get_object_or_404(
    Convocacao.objects.prefetch_related("cursos", "registroconvocacao_set"),
    pk=pk,
)

# 把预取的所有RegistroConvocacao对象转存到内存列表
all_registros = list(obj.registroconvocacao_set.all())

# 遍历cursos时，在内存中过滤数据
for curso in obj.cursos.all():
    # 用列表推导式在内存中筛选（替换成你的实际过滤条件）
    filtered_registros = [r for r in all_registros if r.curso == curso]
    # 处理筛选后的结果

方案2：使用Prefetch自定义预取属性（更清晰的写法）

通过Prefetch对象将预取的结果存到自定义属性中，避免和默认的RelatedManager混淆，之后同样在内存中过滤：

from django.db.models import Prefetch

# 获取对象时，用Prefetch指定预取结果的存储属性
obj = get_object_or_404(
    Convocacao.objects.prefetch_related(
        "cursos",
        Prefetch(
            "registroconvocacao_set",
            queryset=RegistroConvocacao.objects.all(),  # 可在此添加全局过滤条件
            to_attr="registros_list"  # 自定义属性名，存储预取的所有对象
        )
    ),
    pk=pk,
)

# 遍历cursos时直接使用自定义属性的列表进行内存过滤
for curso in obj.cursos.all():
    filtered_registros = [r for r in obj.registros_list if r.curso == curso]
    # 处理筛选后的结果

关键原理 #

预取的数据只会被RelatedManager的all()方法复用，而filter()/exclude()等方法会生成新的QuerySet，绕过预取缓存。将预取数据转成列表后，所有筛选操作都在内存中完成，不会触发新的数据库请求。

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