通过带注释的查询进行 Django 聚合

Question

这是我希望 Django 在 SQL 中生成的内容:

select avg(subquery.countval) from (
select count(something) countval,user_id from foo group by user_id
 ) subquery

我认为这应该如何与基于 Annotated 聚合文档的 Django 一起使用:

Foo.objects.all().values('user_id').\
                 annotate(countval=Count('id')).\
                 aggregate(Avg('countval'))

问题是 Django 4.x 没有生成正确的查询。你会得到类似的东西:

SELECT FROM (SELECT foo.user_id as user_id,COUNT(foo.id) 
 AS countval from foo 
 group by foo.user_id)

有任何想法吗？我通过源代码进行了调试，但不清楚出了什么问题。

Answer 1

我无法用纯 Django 代码来做，但这是我能做的最好的，尽可能多地依赖 Django 代码而不是原始 sql。

from django.db import connection
from django.db.models import Count

def get_average_count(klass, field_name):
    foo = klass.objects.values(field_name).annotate(countval=Count('id'))
    query = "SELECT AVG(subquery.countval) FROM (%s) subquery" % str(foo.query)
    cursor = connection.cursor()
    cursor.execute(query)
    return float(cursor.fetchone()[0])

这将执行您说要生成的确切 SQL 语句。它也完全独立于您正在使用的 SQL 后端，并且完全可重用 (yay DRY) 用于具有反向外键或多对多关系的所有类。

如果您真的不想使用原始 SQL，另一种选择是在 Django 中计算平均值:

from __future__ import division # no need to cast to float now

def get_average_count(klass, field_name):
    counts = klass.objects.values(field_name).annotate(countval=Count('id')).\
        values_list('countval', flat=True)
    return reduce(lambda x, y: x + y / len(counts), counts, 0)

如果您计划在数据库中拥有大型数据集，您可能需要检查任何性能差异。