django group by
概要
如何用 django 自带的 ORM 框架来完成 SQL 中的 group by 操作呢？
环境介绍
foo 这个应用定义了如下模型。
class PersonModel(models.Model):
name = models.CharField('姓名', max_length=64)
age = models.PositiveIntegerField('年龄')
def __str__(self):
return f"{self.name} - {self.age}"
数据库层面看到的表结构如下。
CREATE TABLE `foo_personmodel` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`name` varchar(64) COLLATE utf8mb4_general_ci NOT NULL,
`age` int(10) unsigned NOT NULL,
PRIMARY KEY (`id`)
) ENGINE=InnoDB
表中的数据如下。
mysql> select * from foo_personmodel;
+----+-------+-----+
| id | name | age |
+----+-------+-----+
| 1 | tim | 16 |
| 2 | jerry | 27 |
| 3 | marry | 18 |
| 4 | tim | 20 |
| 5 | alis | 17 |
+----+-------+-----+
5 rows in set (0.00 sec)
全表聚合
一个常见的需求，查询一下表中有多少行数据。
SQL 写法。
mysql> select count(id) as counts from foo_personmodel;
+--------+
| counts |
+--------+
| 5 |
+--------+
1 row in set (0.00 sec)
ORM 实现。
from django.db.models import Count
from apps.foo.models import PersonModel
PersonModel.objects.aggregate(counts = Count(id))
{'counts': 5}
注意 aggregate 返回的不再是 queryset 而是一个字典。
values 方法
要查询出每一行 name 列的取值。
SQL 写法。
mysql> select name from foo_personmodel;
+-------+
| name |
+-------+
| tim |
| jerry |
| marry |
| tim |
| alis |
+-------+
5 rows in set (0.00 sec)
ORM 实现。
PersonModel.objects.values("name")
<QuerySet [{'name': 'tim'}, {'name': 'jerry'}, {'name': 'marry'}, {'name': 'tim'}, {'name': 'alis'}]>
django 会返回一个 queryset 并且不会去重。
分组聚合
查询同一个名字在表中出现了多少次。
SQL 写法。
mysql> select name,count(id) as counts from foo_personmodel group by name;
+-------+--------+
| name | counts |
+-------+--------+
| tim | 2 |
| jerry | 1 |
| marry | 1 |
| alis | 1 |
+-------+--------+
4 rows in set (0.00 sec)
ORM 写法。
PersonModel.objects.values("name").annotate(counts=Count(id))
<QuerySet [{'name': 'tim', 'counts': 2}, {'name': 'jerry', 'counts': 1}, {'name': 'marry', 'counts': 1}, {'name': 'alis', 'counts': 1}]>
神奇的事情发生了，当 values 和 annotate 一起使用的时候，values 就承担起了 group by 的角色。并且自动去掉了重项!
结论
django 中并没有为 group by 设置单独的方法，而是通过 values + annotate 的组合来实现的。