Django 聚合和查询

Question

我有许多可以属于一种或多种类型的元素。

class Type(models.Model):
    name = models.CharField(max_length=128, unique=True)

class Element(models.Model):
    name = models.CharField(max_length=128, unique=True)
    type = models.ManyToManyField('Type')

假设我有 3 种类型和 3 个元素：

In [3]: Type.objects.all()
Out[3]: [<Type: His Type>, <Type: My Type>, <Type: Your Type>]

In [4]: [(e,e.type.all()) for e in Element.objects.all()]
Out[4]: 
[(<Element: First Element>, [<Type: My Type>]),
(<Element: Second Element>, [<Type: Your Type>]),
(<Element: Third Element>,
  [<Type: My Type>, <Type: Your Type>, <Type: His Type>])]

我正在尝试获取一个查询集，其中包含仅属于“我的类型”类型的元素

我的想法是获取此类型的元素并检查它们是否仅属于一种类型。

但由于某种原因，它认为“第三元素”只有一种类型，

In [5]: my_type=Type.objects.get(name='My Type')

In [6]: my_type.element_set.annotate(num_types=Count('type')).filter(num_types__exact=1)
Out[6]: [<Element: First Element>, <Element: Third Element>]

In [7]: [(e,e.num_types) for e in my_type.element_set.annotate(num_types=Count('type'))]
Out[7]: [(<Element: First Element>, 1), (<Element: Third Element>, 1)]

而它是三种类型，

In [8]: Element.objects.get(name='Third Element').type.count()
Out[8]: 3

我做错了什么？

Answer 1

鉴于这是标准的多对多，您可以调用“my_type.element_set.all()”，它应该为您提供该类型的 Element 实例。您还可以采用另一种方式（使用模型名称的复数版本，而不是“_set”）来获取元素类型，如下所示：“my_element.types.all()”。

要获取计数，只需执行“my_type.element_set.count()”即可。

Answer 2

Django 在后台执行的 SQL 查询将您限制为仅包含 type_element_type.element_id = 1 的行。因此，带注释的计数仅适用于这些，因此每个 1。

上面的查询 [7] 变为此 SQL（如 这个)：

SELECT "type_element"."id", "type_element"."name", COUNT("type_element_type"."type_id") 
AS "num_types" FROM "type_element" LEFT OUTER JOIN "type_element_type"
ON ("type_element"."id" = "type_element_type"."element_id")
WHERE "type_element_type"."type_id" = 1 
GROUP BY "type_element"."id", "type_element"."name"

以及来自 sqlite3：

sqlite> SELECT * FROM "type_element" LEFT OUTER JOIN "type_element_type" ON ("type_element"."id" = "type_element_type"."element_id") WHERE "type_element_type"."type_id" = 1;
1|First Element|1|1|1
3|Third Element|3|3|1

Answer 3

我结束了

Element.objects.filter(pk__in=my_type.element_set.values_list('pk', flat=True)).annotate(num_types=Count('type')).filter(num_types=1)

这样，我就不会受到从 my_type.element_set 开始时发生的连接的影响。

如果您想知道所涉及的 SQL：

在 my_type.element_set.annotate(num_types=Count) 中('type')).filter(num_types__exact=1) 它的作用是：

SELECT `senalesweb_element`.`id`, `senalesweb_element`.`name`, COUNT(`senalesweb_element_type`.`type_id`) AS `num_types` FROM `senalesweb_element` LEFT OUTER JOIN `senalesweb_element_type` ON (`senalesweb_element`.`id` = `senalesweb_element_type`.`element_id`) WHERE (`senalesweb_element_type`.`type_id` = 1 ) GROUP BY `senalesweb_element`.`id` HAVING COUNT(`senalesweb_element_type`.`type_id`) = 1  ORDER BY NULL LIMIT 21

在 Element.objects.filter(pk__in=my_type.element_set.values_list('pk', flat=True)).annotate 中(num_types=Count('type')).filter(num_types=1) 它的作用是：

SELECT `senalesweb_element`.`id`, `senalesweb_element`.`name`, COUNT(`senalesweb_element_type`.`type_id`) AS `num_types` FROM `senalesweb_element` LEFT OUTER JOIN `senalesweb_element_type` ON (`senalesweb_element`.`id` = `senalesweb_element_type`.`element_id`) WHERE (`senalesweb_element`.`id` IN (SELECT U0.`id` FROM `senalesweb_element` U0 INNER JOIN `senalesweb_element_type` U1 ON (U0.`id` = U1.`element_id`) WHERE U1.`type_id` = 1 )) GROUP BY `senalesweb_element`.`id` HAVING COUNT(`senalesweb_element_type`.`type_id`) = 1  ORDER BY NULL LIMIT 21