python - SQLalchemy 查询获取字典的聚合数组

Question

是否可以从 SQLAlchemy 查询中获取包含字典的聚合数组？例如

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'}]),
]

Answer 1

我想你可以使用 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()

需要注意的是，与您的示例的整数键相比，键将是字符串。