python - 为每个类别选择一个项目的查询语法

Question

class Category(models.Model):
    pass

class Item(models.Model):
    cat = models.ForeignKey(Category)

我想为每个类别只选择一个项目，这是执行此操作的查询语法吗？

Answer 1

您的问题并不完全清楚:既然您没有另行说明，我假设您不关心为每个类别选择了哪个项目，只是您需要任何人。如果不是这种情况，请更新问题以进行澄清。

tl;dr version: there is no documented way to explicitly use GROUP BY statements in Django, except by using a raw query. See the bottom for code to do so.

问题在于，要在 SQL 中执行您正在寻找的操作，需要一点技巧。您可以通过在命令行输入 sqlite3 :memory: 轻松尝试此示例:

CREATE TABLE category
(
  id INT
);

CREATE TABLE item
(
  id INT,
  category_id INT
);

INSERT INTO category VALUES (1);
INSERT INTO category VALUES (2);
INSERT INTO category VALUES (3);


INSERT INTO item VALUES (1,1);
INSERT INTO item VALUES (2,2);
INSERT INTO item VALUES (3,3);
INSERT INTO item VALUES (4,1);
INSERT INTO item VALUES (5,2);

SELECT id, category_id, COUNT(category_id) FROM item GROUP BY category_id;

返回

4|1|2
5|2|2
3|3|1

这就是您要查找的内容(每个类别 ID 对应一个项目 ID)，尽管有一个无关的 COUNT。需要计数(或其他一些聚合函数)才能应用 GROUP BY。

注意:这将忽略不包含任何项目的类别，这似乎是明智的行为。

现在的问题是，如何在 Django 中执行此操作？

显而易见的答案是使用 Django's aggregation/annotation support ，特别是结合annotate与 values原样recommend elsewhere在 Django 中进行 GROUP 查询。

阅读这些帖子，似乎我们可以完成我们正在寻找的东西

Item.objects.values('id').annotate(unneeded_count=Count('category_id'))

但是这不起作用。 Django 在这里所做的不仅仅是GROUP BY "category_id"，而是根据所有选择的字段进行分组(即GROUP BY "id", "category_id") 1 .我认为没有办法(至少在公共(public) API 中)可以改变这种行为。

解决方案是回退到原始 SQL:

qs = Item.objects.raw('SELECT *, COUNT(category_id) FROM myapp_item GROUP BY category_id')

---

1 : 请注意，您可以检查 Django 正在运行哪些查询:

from django.db import connection
print connection.queries[-1]

---

编辑:

还有许多其他可能的方法，但大多数都有(可能是严重的)性能问题。这里有一对:

1。从每个类别中选择一个项目。

items = []
for c in Category.objects.all():
    items.append(c.item_set[0])

这是一种更清晰、更灵活的方法，但明显的缺点是需要更多的数据库命中。

2。使用 select_related

items = Item.objects.select_related()

然后自己进行分组/过滤(使用 Python)。

同样，这可能比使用原始 SQL 更清楚并且只需要一个查询，但是这个查询可能非常大(它将返回所有项目及其类别)并且自己进行分组/过滤可能效率低于让数据库为您完成。