如何在SQLAlchemy中自动映射后端无关Base并解决Oracle转MSSQL类型问题

这个问题我之前迁移Oracle到SQL Server时也碰到过——本质是SQLAlchemy的automap在反射Oracle schema时，没有把Oracle特有的NUMBER类型自动转换成跨数据库通用的SQLAlchemy标准类型，而是原封不动保留了Oracle方言类型，导致MSSQL的编译器根本认不出它。下面给你几个可行的解决思路：

1. 手动遍历反射后的表，替换NUMBER类型（最灵活） #

如果你的表结构里NUMBER既有整数又有小数，这种方法能精准控制每个列的转换逻辑：

from sqlalchemy.ext.automap import automap_base
from sqlalchemy.dialects.oracle.base import NUMBER
from sqlalchemy import Integer, Numeric
from custom_connections import connect_to_oracle, connect_to_mssql

Base = automap_base()
oracle_engine = connect_to_oracle()
mssql_engine = connect_to_mssql()

# 先完成Oracle schema的反射
Base.prepare(oracle_engine, reflect=True, schema='ORACLE_MAIN_DB')

# 遍历所有表和列，把Oracle NUMBER转成标准类型
for table in Base.metadata.tables.values():
    for col in table.columns.values():
        if isinstance(col.type, NUMBER):
            # 分情况处理：无精度刻度的NUMBER默认按整数处理（根据你的业务调整）
            if col.type.precision is None and col.type.scale is None:
                col.type = Integer()
            # 有精度刻度的转成通用Numeric类型
            else:
                col.type = Numeric(precision=col.type.precision, scale=col.type.scale)

# 现在再创建MSSQL的表就不会报错了
Base.metadata.create_all(mssql_engine)

这种方式的好处是你能根据实际业务逻辑调整转换规则，比如某些NUMBER(20)实际上是大整数，就转成BigInteger而不是普通Integer。

2. 全局修改Oracle类型映射（最快） #

如果你的场景里所有NUMBER都可以统一转成Numeric或者Integer，可以直接全局替换Oracle的NUMBER类型定义，让automap反射时自动用标准类型：

from sqlalchemy.ext.automap import automap_base
from sqlalchemy.dialects.oracle import base as oracle_base
from sqlalchemy import Numeric
from custom_connections import connect_to_oracle, connect_to_mssql

# 把Oracle的NUMBER默认映射成SQLAlchemy标准Numeric
oracle_base.NUMBER = lambda precision=None, scale=None: Numeric(precision=precision, scale=scale)

Base = automap_base()
oracle_engine = connect_to_oracle()
mssql_engine = connect_to_mssql()

# 现在反射出来的类型都是标准类型了
Base.prepare(oracle_engine, reflect=True, schema='ORACLE_MAIN_DB')
Base.metadata.create_all(mssql_engine)

注意这种全局修改可能会影响其他使用Oracle连接的代码，如果你只是做迁移用，这个方法最省事。

3. 针对Alembic迁移的解决方案 #

如果你用Alembic生成迁移脚本，同样可以处理类型问题：

方法一：在Alembic配置里自动转换 #

修改env.py中的迁移逻辑，添加类型转换的处理：

from sqlalchemy.dialects.oracle.base import NUMBER
from sqlalchemy import Numeric

def run_migrations_online():
    connectable = engine_from_config(
        config.get_section(config.config_ini_section),
        prefix="sqlalchemy.",
        poolclass=pool.NullPool,
    )

    with connectable.connect() as connection:
        context.configure(
            connection=connection,
            target_metadata=target_metadata,
            render_as_batch=True,
            # 给Oracle NUMBER类型指定转换规则
            type_visitor={NUMBER: lambda _, col: Numeric(col.precision, col.scale)}
        )

        with context.begin_transaction():
            context.run_migrations()

方法二：手动修改迁移脚本 #

生成迁移文件后，直接把里面的oracle.NUMBER()替换成sa.Integer()或者sa.Numeric()，再执行迁移就行——适合表数量不多的场景。

核心逻辑说明 #

SQLAlchemy的automap默认会保留数据库方言的专属类型，而不同数据库的方言类型是不兼容的。解决的关键就是把Oracle的NUMBER这类方言类型，转换成SQLAlchemy提供的通用标准类型（比如Integer、Numeric），这样MSSQL的编译器就能正确识别并生成对应的SQL Server类型（比如INT、DECIMAL）。

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