在执行 Sqlalchemy 升级时禁用外键检查

在 Sqlalchemy 中，使用 Alembic 进行数据库迁移时，如果涉及到外键的增删改操作，会默认启用外键检查，如果出现了外键冲突，就会报错。

如果你确定数据库中现有的数据是符合外键约束的，并且升级过程中不会破坏外键约束，可以在执行迁移时禁用外键检查，这样可以加快迁移的速度。

在 Alembic 迁移文件中，可以通过在 op.execute 方法前后执行 op.drop_constraint 和 op.create_foreign_key 来删除和添加外键，具体操作可以参考 Sqlalchemy 文档。

禁用外键检查的方法如下：

from alembic import context

# 增加以下代码
def include_object(object, name, type_, reflected, compare_to):
    if type_ == "table" and "sqlite_sequence" in name:
        return False
    if not reflected:
        return True
    return False

context.configure(
    connection=connection,
    target_metadata=target_metadata,
    process_revision_directives=process_revision_directives,
    # 禁用外键检查
    include_object=include_object,
)

在 context.configure 中新增了一个参数 include_object，根据传入的函数决定是否包含某个对象（比如表、索引、外键等），在这个函数中返回 False 即可禁用对应对象的检查。

注意，需要判断是否为反射（reflected=True）的对象，因为不是反射的对象不需要禁用检查。上文代码中判断了非反射对象（not reflected）也返回了 True，这是因为反射的对象已经包含了引用对象的定义，所以不需要在这里再次定义。

以上就是禁用外键检查的方法，看到了其他有趣的东西也别忘了留意哦！