基于数据库引擎的 Alembic 条件答案

【问题标题】：Alembic conditional based on database engine基于数据库引擎的 Alembic 条件
【发布时间】：2019-01-24 20:07:48
【问题描述】：

alembic 能否根据数据库类型运行略有不同的迁移代码？例如，只在 MySQL 上运行 ALTER TABLE object AUTO_INCREMENT = 6000;，而在 SQLite 上跳过这个？

一些背景知识：我们正在使用 alembic 来运行迁移。在许多开发设置中，我们使用 sqlite，而在生产环境中，我们使用 mysql。在生产环境中，我们希望以特定值启动一些主键，但在使用 sqlite 的开发设置中，这是不可能的，也不需要，因此我们可以跳过这一步。

【问题讨论】：

标签： python migration alembic

【解决方案1】：

可以从绑定中获取引擎名称。我在迁移中添加了以下内容：

def upgrade():
    # create table call
    bind = op.get_bind()
    if bind.engine.name == 'mysql':
        op.execute("ALTER TABLE object AUTO_INCREMENT = 5000")
    else:
        print("Skipping setting initial ID value")

【讨论】：