django 与包含原始模型字段的过滤器相关的预取

wkftcu5l  于 2022-12-14  发布在  Go
关注(0)|答案(1)|浏览(105)

如果模型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
gab6jxml

gab6jxml1#

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

相关问题