18

如何使用 Alembic--autogenerate迁移SQL Alchemy 模型中未硬编码的多个Postgres 模式？（关于 Postgres Schemas的 SQLAlchemy 支持的镜像问题，但对于 Alembic）。

特别是，我们使用 Postgres 模式来分隔共享同一组表的不同客户端。此外，客户端之间有一个共享内容的模式。SQL Alchemy 模型不了解模式，模式是在运行时使用session.execute("SET search_path TO client1,shared").

默认设置--autogenerate根本没有帮助，因为它会检测模型中不存在的多个模式并最终删除模式并重新创建默认模式中的每个表。

不过，我真的很想使用--autogenerate正确的管道来正确设置模式。关于 Alembic 的 API 是否/如何做到这一点的任何建议？

4

3 回答 3

9

从问题 409开始，升级/降级操作的租户特定模式的应用程序可以使用翻译的模式名称最容易地完成，这也是您通常为多租户执行主应用程序的方式。

进入 env.py：

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:
        for tenant_schema_name in all_my_tenant_names:
             conn = connection.execution_options(schema_translate_map={None: tenant_schema_name}

            logger.info("Migrating tenant schema %s" % tenant_schema_name)
            context.configure(
                connection=conn,
                target_metadata=target_metadata
            )

            # to do each tenant in its own transaction.
            # move this up to do all tenants in one giant transaction
            with context.begin_transaction():
                context.run_migrations()

上面会将“无”模式名称转换为给定的租户名称。如果应用程序与具有全局表的默认模式共享基于租户的模式，那么您将使用像“tenant_schema”这样的标记作为符号：

for tenant_schema_name in all_my_tenant_names:
     conn = connection.execution_options(schema_translate_map={"tenant_schema": tenant_schema_name}

并在迁移文件中引用“tenant_schema”，其中实际租户特定的架构名称所在的位置：

def upgrade():
    op.alter_column("some_table", "some_column", <migration options>, schema="tenant_schema")

对于“自动生成”的情况，@nick-retallack 提供的解决方案提供了更多您将在这方面使用的部分，即使用 include_schemas 以便自动生成只查看代表最新版本的“样本”模式特定于租户的架构。

为了设置 env.py 为正确的命令使用正确的系统，可以使用带有migration_context.get_x_argument()的用户定义选项来控制行为。

于 2017-07-05T14:31:50.587 回答
1

这是一个可行的解决方案：https ://gist.github.com/nickretallack/bb8ca0e37829b4722dd1

它仍然需要您在每次迁移生成后编辑模式名称，但至少有趣的工作是由 Alembic 完成的。感谢 Michael Bayer 在邮件列表中提供的帮助。

于 2014-11-18T21:07:09.267 回答
1

这是一个相当古老的问题，但对于遇到同样问题的任何人，我通过在 env.py 中使用以下内容来解决它：

def run_migrations_online():
"""Run migrations in 'online' mode.

In this scenario we need to create an Engine
and associate a connection with the context.

"""
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,
        include_schemas=True,
        version_table_schema=build  # <-- This is the relevant line
    )

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

wherebuild是定义所需模式名称的字符串。我的用例略有不同（多个分布式构建，单个数据库包含多个相同的模式），但是我遇到了同样的问题，即 alembic 没有正确检测到我试图连接的模式。

我使用环境变量来确定正确的构建，因为它与 Zappa 配合得很好。

于 2017-10-20T01:23:00.430 回答