SQLalchemy查询获取dicts的聚合数组

时间:2017-11-09 13:43:42

标签: python postgresql sqlalchemy

是否可以从SQLAlchemy查询中获取包含字典的聚合数组? e.g。

session.query(
    Object.name,
    func.array_agg({Location.id: Location.name}).label('locations')
)\
    .join(Location)\
    .all()

所以预期结果将是:

[
 ('Horizontal neutral circuit',
  [{143:'A5'},{145:'A8'},{765:'B12'}]),
 ('Fletcher, Lopez and Edwards',
  [{41:'A1'},{76:'B8'},{765:'B12'}]),
]

1 个答案:

答案 0 :(得分:3)

我猜您可以使用json_build_object()来构建词典:

from sqlalchemy.dialects import postgresql

session.query(
        Object.name,
        postgresql.array_agg(
            func.json_build_object(Location.id,
                                   Location.name)).label('locations'))\
    .join(Location)\
    .group_by(Object.name)\
    .all()

警告说,与示例的整数键相比,键将是字符串。