按相关字段值分组查询集

Question

我有以下模型结构：

class Chapter:
    name = ...


class Atom:
    name = ...
    chapter = FK(Chapter)


class Question:
    title = ...
    atom = FK(Atom)

我想获取所有与Chapter 相关的Questions，在本章中按Atoms 分组。

如何处理我知道的许多查询：

atoms = Question.objects.filter(atom__chapter__id=1).values_list("atom", flat=True).distinct()
result = {}
for atom in atoms:
    result["atom"] = Question.objects.filter(atom__id=atom)

我想这可以通过使用annotate、Subquery 和其他Django 的员工的一个查询来完成。这可能吗？

Answer 1

您可以在一个查询中获取此信息，但您需要自己进行抓取：

from itertools import groupby
from operator import attrgetter

questions = Question.objects.filter(
    atom__chapter_id=1
).order_by('atom')

result = {
    k: list(vs)
    for k, vs in groupby(questions, attrgetter('atom_id'))
}

这里的result 是一个字典，它将atom_ids 的主键映射到Question 对象列表与atom_id。