Django查询多对多子集包含

问题描述 投票:8回答:1

有没有办法查询具有多对多字段的子集或超集包含?

假设每个人都有他们想要看到的鸟类列表,每个鸟类都有一个鸟类列表。对于给定的Person实例,我如何进行查询以查找哪些Aviaries拥有该人员列表中的每只鸟?同样,对于给定的Person实例,我如何找到哪个Aviary在该人员列表中只有鸟类(但不一定是所有鸟类)。

这是我的Django 1.5型号:

class Bird(models.Model):
    name = models.CharField(max_length=255, unique=True)

class Aviary(models.Model):
    name = models.CharField(max_length=255, unique=True)
    birds = models.ManyToManyField(Bird)

class Person(models.Model):
    name = models.CharField(max_length=255, unique=True)
    birds_to_see = models.ManyToManyField(Bird)

我知道如何找到至少有一个人的鸟类的鸟舍,但我不知道我会如何适应这里。 (参见:django queryset for many-to-many field

如果有一个查询可以满足我的需求,我也有兴趣知道是否/为什么更加“手动”这样做更可取。例如,我可以遍历鸟舍,提取每个鸟舍的鸟类列表,并查看该人的bird_to_see是鸟舍鸟类列表的子集或超集:

def find_aviaries(self):
    person_birds = set(self.birds_to_see.all())
    found_aviaries = []
    for aviary in Aviary.objects.all():
        aviary_birds = set(aviary.birds.all())
        if person_birds.issubset(aviary_birds):
            found_aviaries.append(aviary)            
    return found_aviaries

任何帮助表示赞赏!

python django many-to-many django-queryset django-orm
1个回答
0
投票

Django> = 2.0存在一个很好的解决方案。可以通过匹配的鸟类数量来注释Aviaries,并过滤至少匹配一只Bird或所需数量的Aviaries。

from django.db.models import Count

    ...
    person_birds = set(self.birds_to_see.all())
    aviaries = (
        Aviary.objects
        .annotate(bird_match_count=Count('birds', filter=Q(birds__in=person_birds)))
        .filter(bird_match_count__gt=0)
    )

然后,通过bird_match_count=len(person_birds)过滤新的查询集或者在Python中过滤原始查询集或者通过bird_match_count对其进行排序是微不足道的。

Django <2.0将需要引用中间模型AviaryBirds并且将更加冗长。


通过阅读SQL验证:

>>> print(aviaries.query)
SELECT aviary.id, aviary.name,
  COUNT(CASE WHEN  aviary_birds.bird_id IN (1,..)  THEN aviary_birds.bird_id ELSE NULL END)
    AS bird_match_count
FROM aviary LEFT OUTER JOIN aviary_birds ON (aviary.id = aviary_birds.aviary_id)
GROUP BY aviary.id, aviary.name
HAVING
  COUNT(CASE WHEN (aviary_birds.bird_id IN (1,..)) THEN aviary_birds.bird_id ELSE NULL END)
     > 0
© www.soinside.com 2019 - 2024. All rights reserved.