SQLAlchemy操作BigQuery：移除ArrayAgg自动生成的列标签

问题原因 #

SQLAlchemy会为聚合函数这类非直接列的表达式自动生成别名（如anon_1），导致原本预期的ARRAY_AGG(t ORDER BY time_col DESC LIMIT 1)[OFFSET(0)].*被生成为ARRAY_AGG(...)[OFFSET(0)].anon_1.*，不符合BigQuery语法，引发功能异常。

解决方案 #

方法一：直接用text()构造完整表达式 #

适合快速实现，直接写入目标SQL片段：

from sqlalchemy import select, text

# 定义你的子查询
subquery = ... 

# 构造主查询
query = select(
    text('t.id'),
    text("ARRAY_AGG(t ORDER BY `time_col` DESC LIMIT 1)[OFFSET(0)].*")
).select_from(subquery.alias('t')).group_by(text('t.id'))

方法二：自定义表达式保持ORM风格 #

通过SQLAlchemy的UnaryExpression自定义运算符，避免硬编码SQL，同时绕过自动别名机制：

from sqlalchemy import func, select
from sqlalchemy.sql.expression import UnaryExpression

def array_agg_fetch_all_first(entity, order_by_col):
    # 构造ARRAY_AGG聚合表达式
    agg_expr = func.array_agg(entity, order_by=order_by_col.desc()).limit(1)
    # 拼接[OFFSET(0)].*运算符
    return UnaryExpression(
        agg_expr,
        operator='[OFFSET(0)].*',
        precedence=100,
    )

# 使用示例
subquery = ...  # 你的子查询定义
fetch_all_fields = array_agg_fetch_all_first(subquery, subquery.c.time_col)

query = select(
    subquery.c.id,
    fetch_all_fields
).group_by(subquery.c.id)

验证效果 #

生成的SQL会完全符合预期：

SELECT
 t.id,
 ARRAY_AGG(t ORDER BY `time_col` DESC LIMIT 1)[OFFSET(0)].*
FROM (
    -- 你的子查询内容
) t
GROUP BY t.id

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