如何将SQLAlchemy查询的Task按BusinessUnit分组并序列化为字典列表

你已经完成了按业务单元分组的核心逻辑，只需要将现有的分组字典转换为目标格式的字典列表即可，以下是修正后的代码：

方法一：使用原生字典（无需额外导入） #

# 第一步：正确构建分组字典（去掉之前错误的serialized_tasks.append(res)）
res = {}
for item in tasks:
    res.setdefault(item.business_unit.name, []).append(item.serialize)

# 第二步：转换为目标格式的字典列表
serialized_tasks = [
    {"business_unit_name": bu_name, "tasks": task_list}
    for bu_name, task_list in res.items()
]

方法二：使用collections.defaultdict简化分组 #

如果想让分组代码更简洁，可以导入defaultdict：

from collections import defaultdict

# 构建分组字典
res = defaultdict(list)
for item in tasks:
    res[item.business_unit.name].append(item.serialize)

# 转换为目标格式
serialized_tasks = [
    {"business_unit_name": bu_name, "tasks": task_list}
    for bu_name, task_list in res.items()
]

说明 #

• 第一步的分组逻辑你原本已经正确实现，但之前错误地在循环内重复将res添加到serialized_tasks中，这会导致列表包含重复的字典引用，只需去掉这行代码即可。
• 第二步通过列表推导式遍历分组字典的键值对，将每个键（业务单元名称）和值（序列化任务列表）映射为目标要求的字典结构，最终得到符合格式的字典列表。

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