如何在Alembic迁移中传递PostgreSQL Schema且避免硬编码？

好问题！要实现无需硬编码，直接从DSN的schema参数中获取名称并传入op.create_table()等迁移操作，你可以通过Alembic的上下文配置和迁移脚本的上下文调用实现，具体步骤如下：

1. 在env.py中解析DSN并传递Schema到上下文 #

Alembic的env.py是迁移的核心配置文件，我们可以在这里解析数据库URL中的schema参数，并将其存入配置上下文，供迁移脚本调用：

from sqlalchemy.engine import make_url
from alembic import context
from sqlalchemy import engine_from_config
from sqlalchemy.pool import NullPool

# ... 保留文件中原有其他代码 ...

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

    # 解析数据库URL，提取schema参数
    db_url = make_url(connectable.url)
    # 默认使用public，如果URL里没有schema参数的话
    target_schema = db_url.query.get("schema", "public")

    with connectable.connect() as connection:
        # 配置迁移上下文，同时把schema存入配置项
        context.configure(
            connection=connection,
            target_metadata=target_metadata,
            # 可选：如果希望alembic_version表也放在目标schema里，就加上这行
            version_table_schema=target_schema,
            include_schemas=True,
            # 将schema名称存入配置，方便迁移脚本获取
            config=context.config.set_section_option(
                context.config.config_ini_section,
                "target_schema",
                target_schema
            )
        )

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

# ... 保留文件中原有其他代码 ...

2. 在迁移脚本中动态获取Schema名称 #

修改你的迁移脚本（比如versions/xxxxxx_create_my_table.py），从Alembic的上下文中读取我们刚才存入的schema名称，不用硬编码：

from alembic import op
import sqlalchemy as sa

def upgrade():
    # 从上下文配置中获取schema名称，默认public
    schema_name = op.get_context().config.get_section_option("target_schema", "public")
    op.create_table(
        "my_table",
        sa.Column("id", sa.Integer, primary_key=True),
        sa.Column("name", sa.String(50), nullable=False),
        schema=schema_name
    )

def downgrade():
    schema_name = op.get_context().config.get_section_option("target_schema", "public")
    op.drop_table("my_table", schema=schema_name)

3. 验证效果 #

现在你可以直接使用带schema参数的DSN运行迁移，比如：

alembic -x sqlalchemy.url="postgresql://login:password@postgreshost/dbname?schema=my_schema" upgrade head

或者在alembic.ini的sqlalchemy.url中配置带schema的URL，迁移脚本会自动使用指定的schema创建/删除表，完全不需要修改脚本中的硬编码值。

另外要注意：如果你的模型类也需要对应到这个schema，记得在模型的__table_args__中指定schema=schema_name，或者通过SQLAlchemy的默认schema配置来统一处理，但迁移脚本中还是需要显式传入schema参数，上面的方法已经覆盖了迁移的核心需求。

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