Агрегация и запросы 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

SQL-запрос, который Django выполняет за кулисами, ограничивает вас только строками, которые содержат type_element_type.element_id = 1.Таким образом, аннотированное количество применяется только к ним, следовательно, по 1 каждому.

Ваш запрос [7] выше становится этим SQL (как показано this ):

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 2

Я закончил делать

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

Answer 3

Учитывая, что это стандартное множество ко многим, вы можете вызвать 'my_type.element_set.all ()', который должен дать вам экземпляры Element этого типа.Вы также можете пойти другим путем (используя множественную версию названия модели, а не '_set'), чтобы получить типы элементов, например, так: 'my_element.types.all ()'.

Чтобы получить количество, простосделать 'my_type.element_set.count ()'.