如何在SQLAlchemy关联查询中返回额外布尔值is_in_A

假设你已经通过SQLAlchemy定义了如下模型（对应你的表结构）：

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class A(Base):
    __tablename__ = "A"
    id = Column(Integer, primary_key=True)
    id_B = Column(Integer)

class B(Base):
    __tablename__ = "B"
    id = Column(Integer, primary_key=True)
    name = Column(String)
    # 其他字段按实际定义补充

你可以通过左连接+条件判断实现需求，以下是两种常用写法：

ORM 风格写法 #

from sqlalchemy import func, Boolean
from sqlalchemy.orm import Session

# 假设已创建Session实例
session = Session()

# 构造查询
query_result = session.query(
    B.id.label("id_B"),
    B.name,
    # 通过判断A表关联字段是否非空，生成is_in_A布尔值
    func.cast(func.coalesce(A.id, 0), Boolean).label("is_in_A")
).outerjoin(A, B.id == A.id_B).all()

或者用case表达式更直观：

query_result = session.query(
    B.id.label("id_B"),
    B.name,
    func.case(
        [(A.id.isnot(None), True)],
        else_=False
    ).label("is_in_A")
).outerjoin(A, B.id == A.id_B).all()

Core 风格写法（直接操作表对象） #

from sqlalchemy import select, case

a_table = A.__table__
b_table = B.__table__

query = select(
    b_table.c.id.label("id_B"),
    b_table.c.name,
    case(
        [(a_table.c.id.isnot(None), True)],
        else_=False
    ).label("is_in_A")
).select_from(b_table.outerjoin(a_table, b_table.c.id == a_table.c.id_B))

# 执行查询
with Session() as session:
    result = session.execute(query).all()

逻辑说明 #

1. 使用outerjoin（左连接）关联B表和A表，确保B表的所有记录都会被返回，A表中无匹配的记录会以NULL填充关联字段。
2. 通过func.coalesce或func.case判断A表的关联字段是否存在：
   • 若A.id不为空，说明当前B记录在A表中有对应项，is_in_A为True
   • 否则为False
3. 用label方法给字段重命名，匹配你需要的输出格式。

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