基于数据库引擎的 Alembic 条件

Question

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

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

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