能否通过Django ORM实现按Person聚合关联Place ID列表的查询?
当然可以用Django ORM直接实现这个聚合需求!
完全不需要手动写Python代码来拼接列表,Django的ORM提供了数据库层面的聚合函数来搞定这个事儿。下面分两种常见数据库场景给你具体方案:
针对PostgreSQL数据库(推荐,直接返回列表)
PostgreSQL支持ArrayAgg聚合函数,可以直接将关联的Place ID聚合为列表,步骤如下:
- 首先导入
ArrayAgg:
from django.contrib.postgres.aggregates import ArrayAgg from django.db.models import Value
- 执行查询并生成类字典结构:
# 给每个Person添加place_ids字段,聚合对应的Place ID,default确保无地点时返回空列表 person_place_ids = Person.objects.annotate( place_ids=ArrayAgg('personplace__place__id', default=Value([])) ).values('id', 'place_ids') # 转成你需要的字典格式 result_dict = dict(person_place_ids.values_list('id', 'place_ids'))
这样result_dict就会是{1: [4,5,6], 2: [1,2,5]}这种结构,所有聚合操作都是在数据库层面完成的,性能很高效。
针对MySQL数据库
MySQL没有原生的数组类型,所以需要用GROUP_CONCAT先把ID拼接成字符串,再转成列表:
- 导入需要的函数:
from django.db.models import Func, Value
- 执行查询:
# 用GROUP_CONCAT聚合ID为逗号分隔的字符串 person_place_str = Person.objects.annotate( place_ids_str=Func( 'personplace__place__id', Value(','), function='GROUP_CONCAT', output_field=models.CharField() ) ).values('id', 'place_ids_str') # 转成字典并将字符串分割为列表 result_dict = { item['id']: item['place_ids_str'].split(',') if item['place_ids_str'] else [] for item in person_place_str }
小提示
如果觉得personplace__place__id这种反向关联名称太啰嗦,可以给PersonPlace模型里的person字段加上related_name:
class PersonPlace(models.Model): person = models.ForeignKey(Person, on_delete=models.CASCADE, related_name='visited_places') place = models.ForeignKey(Place, on_delete=models.CASCADE)
之后查询就可以写成ArrayAgg('visited_places__place__id'),可读性会更好~
内容的提问来源于stack exchange,提问作者John
相关产品推荐
相关产品推荐

