SQLAlchemy执行Union查询后query_expression值丢失问题求助

当使用SQLAlchemy的with_expression给模型添加临时属性后，用union_all联合多个查询时，ORM无法自动将联合查询结果中的临时字段映射到模型属性。这是因为union_all生成的新查询会重置部分ORM上下文，原查询中绑定的临时属性关联不会被继承。而你遇到的“同会话先执行单独查询就正常”的情况，是因为Session的identity map已经缓存了带有临时属性的对象，联合查询直接复用了缓存对象，而非从数据库结果重新加载。

下面提供几种可行的解决方式：

方式一：改用with_entities返回元组（最直接） #

放弃给模型绑定临时属性，直接在查询中返回模型对象和目标字段，这样联合查询后能直接获取对应值：

from sqlalchemy import literal

# 子查询返回Tweet对象和liked字段
q1 = session.query(Tweet, literal(True).label('liked'))
q2 = session.query(Tweet, literal(False).label('liked'))

# 联合查询并获取结果
union_result = q1.union_all(q2).all()

# 遍历结果，直接拿到每个Tweet对应的liked值
liked_values = [item[1] for item in union_result]
print(liked_values)  # 输出 [True, False]

方式二：在联合查询中重新绑定临时属性 #

如果需要继续通过模型属性访问liked，可以在联合查询后显式将结果中的liked字段映射到Tweet的临时属性：

from sqlalchemy import literal, column

# 原查询保持不变
q1 = session.query(Tweet).with_expression(Tweet.liked, literal(True))
q2 = session.query(Tweet).with_expression(Tweet.liked, literal(False))

# 联合查询后，重新绑定liked字段到模型属性
union_q = q1.union_all(q2).with_expression(Tweet.liked, column('liked'))

# 执行查询并获取liked值
liked_values = [tweet.liked for tweet in union_q.all()]
print(liked_values)  # 输出 [True, False]

方式三：通过子查询显式指定返回字段 #

将子查询转为子查询对象，再联合后从子查询中选择字段并映射：

from sqlalchemy import literal, column

# 原查询转为子查询
subq1 = session.query(Tweet).with_expression(Tweet.liked, literal(True)).subquery()
subq2 = session.query(Tweet).with_expression(Tweet.liked, literal(False)).subquery()

# 联合子查询
union_subq = subq1.union_all(subq2)

# 从联合子查询中选择Tweet实体和liked字段
result = session.query(Tweet, column('liked')).select_from(union_subq).all()

liked_values = [item[1] for item in result]
print(liked_values)  # 输出 [True, False]

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