Django中如何按group合并查询集并关联文件列表？

你可以通过一次数据库查询完成分组聚合，同时获取每组的基本字段和文件列表，不需要分开查询再合并。以下分两种场景给出实现方式：

场景1：同一group下的path、user、canceled字段值一致 #

如果同一个group对应的所有MyFile实例的path、user、canceled字段值都相同，可以直接将这些字段加入values，配合聚合函数获取所需数据：

from django.contrib.postgres.aggregates import ArrayAgg
from django.db.models import Max, Q

# 构造查询
result = MyFile.objects.values(
    'group', 'path', 'user', 'canceled'
).annotate(
    # 聚合该组所有非空文件，按when倒序排列
    file=ArrayAgg('file', ordering='-when', filter=Q(file__isnull=False)),
    # 取该组最新的时间
    when=Max('when'),
    # 聚合该组关联的所有非空Device的uuid（可替换为Device的其他字段）
    status=ArrayAgg('status__uuid', filter=Q(status__isnull=False))
).distinct('group')

# 转换为目标格式
formatted_result = []
for item in result:
    formatted_item = {
        'group': item['group'],
        'file': item['file'],
        'when': item['when'].isoformat() if item['when'] else None,
        'path': item['path'],
        'user': item['user'].id if item['user'] else None,  # 按需替换为item['user'].username
        'canceled': item['canceled'],
        'status': item['status'] or []
    }
    formatted_result.append(formatted_item)

场景2：同一group下的path、user、canceled字段值可能不同 #

如果同一个group下的MyFile实例字段值不统一，需要取该组最新一条记录的字段值，可以结合Subquery实现：

from django.db.models import Subquery, OuterRef
from django.contrib.postgres.aggregates import ArrayAgg
from django.db.models import Q

# 子查询：获取每个group下最新的MyFile记录
latest_file_subquery = MyFile.objects.filter(
    group=OuterRef('group')
).order_by('-when')[:1]

# 主查询：分组聚合文件列表，同时关联最新记录的字段
result = MyFile.objects.values('group').annotate(
    file=ArrayAgg('file', ordering='-when', filter=Q(file__isnull=False)),
    when=Subquery(latest_file_subquery.values('when')),
    path=Subquery(latest_file_subquery.values('path')),
    user=Subquery(latest_file_subquery.values('user__id')),  # 按需替换为user__username
    canceled=Subquery(latest_file_subquery.values('canceled')),
    status=ArrayAgg('status__uuid', filter=Q(status__isnull=False))
)

# 转换为目标格式
formatted_result = []
for item in result:
    formatted_item = {
        'group': item['group'],
        'file': item['file'],
        'when': item['when'].isoformat() if item['when'] else None,
        'path': item['path'],
        'user': item['user'],
        'canceled': item['canceled'],
        'status': item['status'] or []
    }
    formatted_result.append(formatted_item)

关键说明 #

• ArrayAgg用于将分组内的字段聚合为数组，filter参数可过滤掉null值，避免数组中出现无效元素；
• Max('when')用于获取分组内最新的时间，若需其他逻辑可替换为Min等聚合函数；
• Subquery可以精准获取每个分组内特定记录的字段值，避免字段不一致导致的混乱；
• 最终的格式化步骤可根据需求调整字段展示形式（比如将datetime转为自定义格式字符串）。

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