与包含原始模型的字段的筛选器相关的预回迁 [英] Prefetch related with a filter that contains a field of the original model

问题描述

如果我的模型A具有指向模型B的外键，我如何使用模型B中的某些条件从模型B预取模型A？

这正是我想要的(OuterRef显然只在SubQuery中起作用，但这基本上就是我需要的功能)：

class ModelA(models.Model):
    somecriteria = models.CharField()


class ModelB(models.Model):
    somerelation = models.ForeignKey(ModelA)
    someattribute = models.CharField()


qs = ModelA.objects.all()
qs = qs.prefetch_related(
    Prefetch(
       'modelb_set', 
       queryset=ModelB.objects.filter(someattribute=OuterRef('somecriteria')), 
       to_attr='some_attr'
    ),
)

该查询应执行以下操作(在较少的查询中)：

for obj in qs:
    # Change this to new_qs = obj.some_attr if prefetch is working.
    newqs = obj.modelb_set.filter(someattribute=obj.somecriteria)
    if newqs.exists(): 
        # Do something

推荐答案

modelb_qs = models.Subquery(ModelB.objects.filter(someattribute=models.OuterRef('somerelation__somecriteria')).values('id'))
qs = ModelA.objects.prefetch_related(models.Prefetch('modelb_set', queryset=ModelB.objects.filter(id__in=modelb_qs)))

for obj in qs:
    if len(obj.modelb_set.all()): 
        # Do something

这将只进行2个查询，如果您在循环中放置一个Filter()，则将进行N(O^2)个查询...不太好。

我写了一个测试来证明这只需要2次查询，你可以在这里找到它：

https://gist.github.com/kingbuzzman/fd2b635f2cf011f1330a6be088ce3664#file-modelabquery-py-L125

这篇关于与包含原始模型的字段的筛选器相关的预回迁的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！