Django ORM通过注释多个聚合列来删除不需要的组

问题描述

我想在Django ORM中创建类似这样的查询。

SELECT COUNT(CASE WHEN myCondition THEN 1 ELSE NULL end) as numyear
FROM myTable

以下是我写的djang ORM查询

year_case = Case(When(added_on__year = today.year,then=1),output_field=IntegerField())

qs = (ProfaneContent.objects
                    .annotate(numyear=Count(year_case))
                    .values('numyear'))

这是由django orm生成的查询。

SELECT COUNT(CASE WHEN "analyzer_profanecontent"."added_on" BETWEEN 2020-01-01 00:00:00+00:00 AND 2020-12-31 23:59:59.999999+00:00 THEN 1 ELSE NULL END) AS "numyear" FROM "analyzer_profanecontent" GROUP BY "analyzer_profanecontent"."id"

其他所有东西都很好，但是django在末尾放置 GROUP BY ，导致出现多行和错误的答案。我一点都不想要。现在只有一列，但我会放置更多这样的列。

基于评论的编辑 我将使用qs变量来获取有关在当前年，月，周中如何进行分类的值。

更新 根据我的评论和回答，请让我澄清一下。我只想在数据库端执行此操作（显然使用Django ORM而不是RAW SQL）。它是一个简单的SQL查询。由于数据可能太大，因此在Python末尾做任何事情都会效率低下。这就是为什么我希望数据库根据CASE条件为我提供记录的总和。 将来我会添加更多此类列，因此len（）或.count之类的内容将无法正常工作。

我只想使用Django ORM（没有自动附加的GROUP BY）创建上述查询。

解决方法

在注释中使用聚合时，django需要进行某种分组，如果没有，则默认为主键。因此，您需要在.values()之前使用.annotate()。 Please see django docs。

但是要完全删除group by可以使用一个静态值，而django足够聪明，可以完全删除它，因此您可以使用如下ORM查询获得结果：

year_case = Case(When(added_on__year = today.year,then=1),output_field=IntegerField())

qs = (ProfaneContent.objects
                    .annotate(dummy_group_by = Value(1))
                    .values('dummy_group_by')
                    .annotate(numyear=Count(year_case))
                    .values('numyear'))

,

如果您只需要汇总到一行，则应该使用.aggregate()方法而不是annotate（）。

result = ProfaneContent.objects.aggregate(
    numyear=Count(year_case),# ... more aggregated expressions are possible here
)

您将获得一个简单的结果列字典：

>>> result
{'numyear': 7,...}

生成的SQL查询没有分组，完全按照要求：

SELECT
  COUNT(CASE WHEN myCondition THEN 1 ELSE NULL end) as numyear
  -- and more possible aggregated expressions
FROM myTable

,

列表理解如何？

# get all the objects
profane = ProfaneContent.objects.all()

# Something like this 
len([pro for pro in profane if pro.numyear=today.year])

如果num years相等，它将添加到列表中，因此在，您可以检查len（）

获得计数

希望这对您有所帮助！

,

这就是我用SQL编写的方式。

SELECT SUM(CASE WHEN myCondition THEN 1 ELSE 0 END) as numyear
FROM myTable

SELECT 
    SUM(CASE WHEN "analyzer_profanecontent"."added_on" 
                 BETWEEN 2020-01-01 00:00:00+00:00 
                     AND 2020-12-31 23:59:59.999999+00:00 
             THEN 1 
             ELSE 0 
         END) AS "numyear" 
FROM "analyzer_profanecontent" 
GROUP BY "analyzer_profanecontent"."id"

如果您打算在SELECT子句中使用其他项目，我建议您也使用group by，如下所示：

SELECT SUM(CASE WHEN myCondition THEN 1 ELSE 0 END) as numyear
FROM myTable
GROUP BY SUM(CASE WHEN myCondition THEN 1 ELSE 0 END)