Django没有使用预取相关的兄弟姐妹的字典?

时间:2018-04-09 14:22:49

标签: python django django-models one-to-many django-select-related

我有三个模型,如下所示,并有一个男孩列表。我怎样才能得到他们每个姐妹的名单? 不幸的是我不能使用相关的预取,因为我被困在使用非常旧版本的django。

class Parent(model.Model):
    name = models.CharField(max_length=50)

class Boy(model.Model):
    parent = models.ForeignKey(Parent)
    name = models.CharField(max_length=50)

class Girl(model.Model):
    parent = models.ForeignKey(Parent)
    name = models.CharField(max_length=50)

所需的输出将是这样的:

{ boy1: [ sister1, sister2 ], boy2: [ .. ] }

提前感谢您的帮助!

1 个答案:

答案 0 :(得分:3)

首先,列出所有男孩的父母,然后用它来让所有女孩和这些父母在一起。

boys = Boy.objects.all()
parents = boys.values_list('parent_id', flat=True)
girls = Girl.objects.filter(parent__in=parents)

然后建立一个父母键入的女孩字典

from collections import defaultdict
girls_by_parent = defaultdict(list)
for girl in girls:
    girls_by_parent(girl.parent_id) = girl

然后你可以为每个男孩建立你最后的姐妹词典:

sisters = { boy.pk: girls_by_parent[boy.parent_pk] for boy in boys } 

如果您可以使用prefetch_related,那么您可以执行以下操作:

boys = Boy.objects.select_related('parent').prefetch_related('parent__girl')
sisters = {b.pk: b.parent.girls_set.all() for b in boys}

如果没有prefetch_related,您将获得额外的查询以获取每个家长的女孩。