基于数据库引擎的 Alembic 条件

Alembic conditional based on database engine

alembic 运行 迁移代码是否可以根据数据库类型略有不同?例如,运行 ALTER TABLE object AUTO_INCREMENT = 6000; 仅在 MySQL 上,但对于 SQLite 跳过此?

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

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

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")