SQLAlchemy如何在SELECT语句中省略列的schema限定符

方案1：全局重写列渲染逻辑（最直接适配需求） #

不需要修改现有模型或查询代码，全局生效，直接去掉SELECT列前的schema前缀：

from sqlalchemy.ext.compiler import compiles
from sqlalchemy.sql.expression import ColumnClause

@compiles(ColumnClause)
def render_column_without_schema(element, compiler, **kw):
    original = compiler.visit_column(element, **kw)
    parts = original.split(".")
    # 只要列名包含三级结构（schema.table.column）就截断取后两级
    if len(parts) >= 3:
        return ".".join(parts[-2:])
    return original

适配场景：单查询内不会出现相同表名、不同schema的表，不会产生命名冲突的场景，完全匹配输出user.id的需求。

方案2：自动为所有表生成别名（无冲突风险，适合多schema场景） #

如果查询可能涉及同表名不同schema的表，可以用这个方案，SQLAlchemy会自动给所有FROM子句的表生成匿名别名，列会用别名前缀，自然不会带schema，完全不需要手动调用aliased：

from sqlalchemy import event
from sqlalchemy.sql import Select

@event.listens_for(Select, "before_compile")
def auto_alias_all_tables(select_stmt):
    return select_stmt.alias_all_tables()

生成的SQL示例：

select
  anon_1.id,
  anon_1.name
from
  myschema.user AS anon_1

该写法所有SQL方言都兼容，不会有语法问题。

方案3：Presto方言专属配置 #

如果你用的是pyhive或者官方Presto SQLAlchemy方言，大部分新版本已经内置了关闭列前缀schema的配置项，创建引擎时开启即可，不需要额外写钩子：

engine = create_engine(
    "presto://<username>@<host>:<port>/<catalog>/<schema>",
    connect_args={"session_properties": {"omit_schema_in_select": "true"}}
)
# 不同方言的参数名可能略有差异，可对应查询你所用方言的配置文档

验证方法 #

可以用下面的代码测试生成的SQL是否符合要求：

stmt = sess.query(User).statement
print(stmt.compile(engine))

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