itertools.groupby 无法通过外键正确分组 django 查询集

Question

假设我有以下模型：

class Category(models.Model):
    name = models.CharField(max_length=10)

class Post(models.Model):
    title = models.CharField(max_length=10)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

然后，我想选择所有帖子并按类别分组：

from itertools import groupby
from operator import attrgetter

qs = Post.objects.all().select_related('category').order_by('category')
posts_group = groupby(qs, key=attrgetter('category'))

但是，posts_group 不包含预期的数据。

例如，我在数据库中有两个帖子，属于不同的类别。查询确实选择了所有类别，但在最后一个类别组中只显示了一个帖子。

Answer 1

这个呢：

def groub_posts_by_cat():
    groups = {}
    categories = Post.objects.values('category')
    for cat in categories:
        groups.update({
            cat.name: Post.objects.filter(category__pk=cat.pk)
        })
    return groups

这里还有一篇关于如何执行 group_by 注释的 excellent 文章。

Answer 2

您可以执行以下操作：

posts = Posts.objects.all()
posts_groupped = {post.category: list() for post in posts}

for post in posts:
    posts_groupped[post.category].append(post)

但是你的

Category

模型应该已经定义了散列：

class Category(models.Model)
    ...
    def __hash__(self):
        return hash(self.title)  # title or any other Category attribute with uniqueness for avoiding hash collisions