如何使用Flask和Alembic删除外键约束解决删表依赖报错

你的错误来自三个核心问题：

• 执行顺序颠倒：先删除了被外键依赖的table_one，此时table_two的外键约束仍指向table_one，数据库会主动拦截这类操作
• 外键约束名填写错误：脚本中写的约束名是table_two_id_fkey，但实际数据库中对应的外键约束名是报错信息里的table_two_file_id_fkey，导致外键约束没有被成功删除
• 降级逻辑顺序错误：原downgrade函数先创建外键再创建table_one，会出现外键指向的表不存在的报错

方案1：按依赖顺序执行操作（推荐，可控性更高） #

先处理依赖方table_two的外键关联，再删除被依赖的table_one，修正后的迁移脚本如下：

def upgrade():
    # 先删外键约束，注意用正确的约束名
    op.drop_constraint('table_two_file_id_fkey', 'table_two', type_='foreignkey')
    # 再删关联的外键字段
    op.drop_column('table_two', 'file_id')
    # 如果你没有删除table_two主键id的需求，可以删掉下一行语句
    op.drop_column('table_two', 'id')
    # 最后删已经没有被依赖的table_one
    op.drop_table('table_one')


def downgrade():
    # 降级先建被依赖的table_one
    op.create_table('table_one',
    sa.Column('id', sa.BIGINT(), autoincrement=True, nullable=False),
    sa.Column('filename', sa.VARCHAR(length=65535), autoincrement=False, nullable=True),
    sa.Column('mission', sa.VARCHAR(length=65535), autoincrement=False, nullable=True),
    sa.PrimaryKeyConstraint('id', name='table_one_pkey')
    )
    # 再给table_two加字段
    op.add_column('table_two', sa.Column('id', sa.BIGINT(), autoincrement=True, nullable=False))
    op.add_column('table_two', sa.Column('file_id', sa.BIGINT(), autoincrement=False, nullable=True))
    # 最后建外键约束
    op.create_foreign_key('table_two_file_id_fkey', 'table_two', 'table_one', ['file_id'], ['id'])

方案2：使用CASCADE强制删除 #

如果你不需要手动处理依赖关系，可以直接给drop_table方法传入cascade=True参数，数据库会自动删除所有依赖该表的约束、索引等对象：

def upgrade():
    # 传入cascade=True自动删除依赖的外键约束
    op.drop_table('table_one', cascade=True)
    op.drop_column('table_two', 'file_id')
    op.drop_column('table_two', 'id')

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