Django 查询（聚合和计数）

Question

嘿伙计们，我有一个如下所示的模型：

class Interaction(DateAwareModel, UserAwareModel):
  page = models.ForeignKey(Page)
  container = models.ForeignKey(Container, blank=True, null=True)
  content = models.ForeignKey(Content)
  interaction_node = models.ForeignKey(InteractionNode)
  kind = models.CharField(max_length=3, choices=INTERACTION_TYPES)

我希望能够执行一个查询来获取按容器然后按类型分组的交互计数。这个想法是输出 JSON 数据结构（由活塞负责的序列化）将如下所示：

"data": {
   "container 1": {
       "tag_count": 3, 
       "com_count": 1
   },
   "container 2": {
       "tag_count": 7, 
       "com_count": 12
   },
   ...
}

SQL 将如下所示：

SELECT container_id, kind, count(*) FROM rb_interaction GROUP BY container_id, kind;

关于如何使用 ORM 按多个字段进行分组的任何想法？ （如果我可以避免 id，我不想为这个项目编写原始查询）这似乎是一个简单而常见的查询。

在你提问之前：我已经看过 django 聚合文档和原始查询文档。

更新 根据下面的建议，我创建了一个自定义管理器来处理此问题：

class ContainerManager(models.Manager):
    def get_query_set(self, *args, **kwargs):
        qs = super(ContainerManager, self).get_query_set(*args, **kwargs)
        qs.filter(Q(interaction__kind='tag') | Q(interaction__kind='com')).distinct()
        annotations = {
            'tag_count':models.Count('interaction__kind'),
            'com_count':models.Count('interaction__kind')
        }
        return qs.annotate(**annotations)

这仅计算类型标签或 com 的交互，而不是通过 group by 检索标签和 com 的计数。很明显，从代码来看它是这样工作的，但想知道如何修复它......

Answer 1

创建自定义管理器：

class ContainerManager(models.Manager):
    def get_query_set(self, *args, **kwargs):
        qs = super(ContainerManager, self).get_query_set(*args, **kwargs)
        annotations = {'tag_count':models.Count('tag'), 'com_count':models.Count('com')}
        return qs.annotate(**annotations)

class Container(models.Model):
    ...
    objects = ContainerManager()

然后，Container 查询将始终包含 tag_count 和 com_count 属性。您可能需要修改注释，因为我没有您的模型的副本可供参考；我只是猜测字段名称。

更新：

因此，在更好地了解您的模型之后，注释将不再适用于您正在寻找的内容。事实上，唯一获得有多少Container拥有'tag'或'com'kind的方法是：

tag_count = Container.objects.filter(kind='tag').count()
com_count = Container.objects.filter(kind='com').count()

注释不会给你这些信息。我认为可以编写自己的聚合和注释，因此这可能是一个可能的解决方案。然而，我自己从来没有这样做过，所以我不能给你任何指导。您可能一直坚持使用直接 SQL。