4

这就是我希望 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)

有任何想法吗？我通过源代码进行了调试，但问题所在并不明显。

4

1 回答 1

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 后端，并且对于具有反向 ForeignKey 或 ManyToMany 关系的所有类完全可重用（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)

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

于 2013-07-26T16:04:25.410 回答