如何仅在数据库迁移时设置pragma foreign_keys=off？

我通过@sqlalchemy.event.listens_for(Engine, "connect")连接处理器，全局启用了外键约束检查（执行PRAGMA foreign_keys=ON）。现在需要运行一段数据库迁移，必须临时禁用该PRAGMA，但尝试了三种方式都无效，batch_op.drop_column仍触发IntegrityError：

def upgrade_database():
    session = orm.Session(bind=op.get_bind())
    session.execute("PRAGMA foreign_keys = OFF;")  # 第一次尝试
    session.commit()
    session.close()
    op.get_bind().execute("PRAGMA foreign_keys=OFF;")  # 第二次尝试
    with op.batch_alter_table("user", schema=None) as batch_op:
        batch_op.get_bind().execute("PRAGMA foreign_keys=OFF;")  # 第三次尝试
        batch_op.drop_column("outdated_column")  # 目标操作

目前只能通过提前区分迁移/应用运行场景，不在处理器中设置PRAGMA来完成迁移，但希望能仅针对该特定迁移关闭外键检查。

你的问题核心在于：PRAGMA foreign_keys是连接级配置，而batch_alter_table可能会创建新的数据库连接；同时，全局的connect事件监听器会在每次新连接建立时强制开启外键检查，导致你之前在旧连接上设置的OFF对新连接无效。

可以通过临时移除全局connect事件监听器的方式，仅在当前迁移过程中禁用外键检查，操作完成后恢复监听器：

步骤1：给全局监听器命名（若未命名） #

确保全局的connect监听器是可引用的命名函数（如果之前用匿名函数，改成命名函数）：

from sqlalchemy import event, Engine

def enable_foreign_keys(dbapi_conn, conn_record):
    cursor = dbapi_conn.cursor()
    cursor.execute("PRAGMA foreign_keys=ON")
    cursor.close()

# 注册全局监听器
event.listen(Engine, "connect", enable_foreign_keys)

步骤2：在迁移函数中临时移除并恢复监听器 #

def upgrade_database():
    # 临时移除全局外键启用监听器
    event.remove(Engine, "connect", enable_foreign_keys)
    try:
        # 执行需要禁用外键的迁移操作
        with op.batch_alter_table("user", schema=None) as batch_op:
            batch_op.drop_column("outdated_column")
    finally:
        # 无论操作成功与否，都恢复监听器
        event.listen(Engine, "connect", enable_foreign_keys)

方法生效原因 #

• 移除监听器后，迁移过程中新建的连接不会被强制设置PRAGMA foreign_keys=ON
• batch_alter_table创建的新连接会使用数据库默认的外键检查状态（SQLite默认关闭，刚好满足迁移需求）
• finally块确保监听器一定会被恢复，不会影响后续的应用正常运行

内容的提问来源于stack exchange，提问作者Raubtier