时间戳

首页 > 解决方案 > 使用子对象字段过滤查询集

python - 使用子对象字段过滤查询集

问题描述

我有两个模型父，子

class Parent(models.Model):
    id = models.IntegerField(...)

class Child(models.Model)
    id = models.IntegerField(...)
    parent = models.ForeignKey(Parent, ...)
    wanted = models.CharField(default="yes")

我想过滤所有与该父母一起出现的孩子将“想要”为“是”的所有父母对象

我的代码：

    def containsYes(self):
        yes_ids = []
        qs = self.get_queryset()
        for q in qs:
            children = Child.objects.filter(parent_id = q.id)
            count = children .count()
            if children.filter(wanted = 'yes').count() == count
            yes_ids.append(q.id)
        return qs.filter(id__contains = yes_ids)

我知道这段代码效率很低，并且想要一个只使用查询的更好的解决方案

PS：我是 Django 新手

标签： pythondjangodjango-queryset

解决方案

我们可以排除Parent存在不想要的孩子的 s，因此我们可以使用：

from django.db.models import F, Count, Q

Parent.objects.annotate(
    nchild=Count('child')
    nchild_wanted=Count('child', filter=Q(child__wanted=True))
).filter(
    nchild=F('nchild_wanted')
)

因此，我们首先计算相关Childs 的数量，并将相关Childs的数量wanted设置为True。然后我们过滤并只保留Parent这两个注释相同的对象。

从开始，可以使用.alias(…)[Django-doc]来防止在SELECT子句和HAVING子句中计数：

from django.db.models import F, Count, Q

Parent.objects.alias(
    nchild=Count('child')
    nchild_wanted=Count('child', filter=Q(child__wanted=True))
).filter(
    nchild=F('nchild_wanted')
)

推荐阅读