如何使用Django ORM实现用户与礼物的分组字典列表？

先修正模型语义问题 #

• 当前Gift模型的related_name="users"命名不符合语义，建议改为related_name="gifts"，这样User实例可以通过user.gifts.all()直接获取关联的礼物集合，更直观。

方法一：使用ArrayAgg（推荐，需数据库支持） #

Django 1.10+提供了ArrayAgg聚合函数，可直接在数据库层面将用户关联的礼物ID聚合为数组，支持PostgreSQL、MySQL 8.0.14+等数据库。

from django.db.models import ArrayAgg, F

# 从User模型出发，为每个用户标注对应的礼物ID数组
user_gift_data = User.objects.annotate(
    gift_ids=ArrayAgg(F('gifts__id'))  # 这里的gifts对应修正后的related_name
).values('id', 'gift_ids')

# 转换为目标字典
gifts_grouped = {item['id']: item['gift_ids'] for item in user_gift_data}

方法二：从Gift模型出发分组 #

也可以直接从Gift模型入手，按user_id分组并聚合礼物ID：

from django.db.models import ArrayAgg, F

gift_group_data = Gift.objects.values('user_id').annotate(
    gift_ids=ArrayAgg(F('id'))
).values('user_id', 'gift_ids')

gifts_grouped = {item['user_id']: item['gift_ids'] for item in gift_group_data}

兼容旧版/不支持ArrayAgg的数据库（如低版本MySQL） #

如果数据库不支持ArrayAgg，可以用GROUP_CONCAT将ID拼接为字符串，再在Python层面转换为列表：

from django.db.models import Func, F

# 自定义GROUP_CONCAT聚合函数
class GroupConcat(Func):
    function = 'GROUP_CONCAT'
    template = '%(function)s(%(expressions)s)'

gift_group_data = Gift.objects.values('user_id').annotate(
    gift_ids_str=GroupConcat(F('id'))
).values('user_id', 'gift_ids_str')

# 转换字符串为整数列表
gifts_grouped = {
    item['user_id']: list(map(int, item['gift_ids_str'].split(',')))
    for item in gift_group_data
}

与itertools.groupby的对比 #

• ORM聚合的方式是在数据库端完成分组，返回的数据量更小（仅传输用户ID和礼物ID数组），无需提前对QuerySet排序（groupby要求输入有序，否则会出现分组不完整的情况）。
• 减少了Python内存占用，无需加载所有Gift对象到内存中，性能更优。

内容的提问来源于stack exchange，提问作者Amir Shabani