Django 获取兄弟姐妹的字典而不使用预取相关



我有三个模型,如下所示,并有一个男孩列表。我怎样才能得到他们每个姐妹的名单?不幸的是,我无法使用与预取相关的,因为我被困在使用非常旧版本的 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: [ .. ] }

提前感谢任何帮助!

首先,获取所有男孩父母的列表,并用它来获取所有女孩和这些父母。

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您将获得额外的查询,以便为每个父母获取女孩。

最新更新