SQLAlchemy中如何从Label对象关联表并添加parent_id过滤条件？

现有如下SQLAlchemy代码：

from sqlalchemy.orm import DeclarativeBase, Mapped
from sqlalchemy import func, select

class TableA(DeclarativeBase):
    col_name = Mapped[str | None]
    parent_id = Mapped[int]

class TableB(DeclarativeBase):
    col_name = Mapped[str | None]
    parent_id = Mapped[int]

class TableC(DeclarativeBase):
    col_name = Mapped[str | None]
    parent_id = Mapped[int]

MY_MAPPINGS = {
   "a__col_name" : func.coalesce(TableA.col_name, "").label("some_col_name"),
   "b__col_name" : func.coalesce(TableB.col_name, "").label("another__col_name"),
   "c__col_name" : func.coalesce(TableC.col_name, "").label("third__col_name"),
}

def get_query(mapping: str, parent_id: int):
    col = MY_MAPPINGS[mapping]
    return select(col)

这段代码可正常运行，但现在需要为查询添加parent_id的过滤条件，理想写法是：

.filter(col.table.parent_id == parent_id)

但col的类型为sqlalchemy.sql.elements.Label，且col.table的值为None，该如何解决？

方法1：修改映射存储表与字段表达式的配对 #

直接在MY_MAPPINGS中存储包含表对象和字段表达式的元组，直接获取表来添加过滤条件：

MY_MAPPINGS = {
   "a__col_name" : (TableA, func.coalesce(TableA.col_name, "").label("some_col_name")),
   "b__col_name" : (TableB, func.coalesce(TableB.col_name, "").label("another__col_name")),
   "c__col_name" : (TableC, func.coalesce(TableC.col_name, "").label("third__col_name")),
}

def get_query(mapping: str, parent_id: int):
    table, col = MY_MAPPINGS[mapping]
    return select(col).filter(table.parent_id == parent_id)

这种方式逻辑清晰，维护成本低，不依赖SQLAlchemy内部结构，是最稳妥的方案。

方法2：从表达式中提取原始字段的表对象 #

如果不想修改现有映射结构，可以从coalesce函数的参数里取出原始字段，进而获取对应的表：

def get_query(mapping: str, parent_id: int):
    col = MY_MAPPINGS[mapping]
    # 从coalesce的第一个参数获取原始字段，再拿到表对象
    original_col = col.element.clauses[0]
    table = original_col.table
    return select(col).filter(table.parent_id == parent_id)

注意：该方法依赖SQLAlchemy表达式的内部结构，若后续版本中coalesce的表达式结构变更，可能会失效，适合临时快速调整场景。

方法3：通过映射键前缀解析关联表 #

利用当前映射键的命名规则（比如a__col_name的前缀a对应TableA），维护表映射字典，通过键前缀匹配表：

TABLE_MAP = {
    "a": TableA,
    "b": TableB,
    "c": TableC
}

def get_query(mapping: str, parent_id: int):
    col = MY_MAPPINGS[mapping]
    # 拆分映射键得到表前缀
    table_prefix = mapping.split("__")[0]
    table = TABLE_MAP[table_prefix]
    return select(col).filter(table.parent_id == parent_id)

这种方法无需修改原MY_MAPPINGS结构，适合映射键命名规则固定的场景。

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