如何在 SQLAlchemy 中删除外键约束？

Question

我正在使用 SQLAlchemy Migrate 来跟踪数据库更改，但在删除外键时遇到了问题。我有两个表，t_new 是一个新表，t_exists 是一个现有表。我需要添加 t_new，然后向 t_exists 添加外键。然后我需要能够反转操作（这就是我遇到麻烦的地方）。

t_new = sa.Table("new", meta.metadata,
    sa.Column("new_id", sa.types.Integer, primary_key=True)
)
t_exists = sa.Table("exists", meta.metadata,
    sa.Column("exists_id", sa.types.Integer, primary_key=True),
    sa.Column(
        "new_id", 
        sa.types.Integer,
        sa.ForeignKey("new.new_id", onupdate="CASCADE", ondelete="CASCADE"),
        nullable=False
    )
)

这工作正常：

t_new.create()
t_exists.c.new_id.create()

但这不行：

t_exists.c.new_id.drop()
t_new.drop()

尝试删除外键列会出现错误：1025，“将'.\my_db_name\#sql-1b0_2e6'重命名为'.\my_db_name\exists'时出错（errno：150） “

如果我使用原始 SQL 执行此操作，我可以手动删除外键，然后删除该列，但我一直无法弄清楚如何使用 SQLAlchemy 删除外键？如何删除外键，然后删除列？

Answer 1

您可以使用 sqlalchemy.migrate 来完成此操作。

为了使其工作，我必须显式创建外键约束，而不是使用 Column('fk',foreignKey('fk_table.field')) 隐式创建外键约束：

唉，而不是这样做：

p2 = Table('tablename',
            metadata,
            Column('id', Integer, primary_key=True),
            Column('fk', ForeignKey('fk_table.field')),
            mysql_engine='InnoDB',
           )

这样做：

p2 = Table('tablename',
            metadata,
            Column('id', Integer, primary_key=True),
            Column('fk', Integer, index=True),
            mysql_engine='InnoDB',
            )
ForeignKeyConstraint(columns=[p2.c.fk], refcolumns=[p3.c.id]).create()

然后删除过程如下所示：

def downgrade(migrate_engine):
     # First drop the constraint
     ForeignKeyConstraint(columns=[p2.c.fk], refcolumns=[p3.c.id]).drop()
     # Then drop the table
     p2.drop()

Answer 2

我通过创建一个单独的元数据实例并使用 Session.execute() 运行原始 SQL 来完成此任务。理想情况下，会有一个专门使用 sqlalchemy 的解决方案，这样我就不必使用特定于 MySQL 的解决方案。但到目前为止，我还不知道有这样的解决方案。

Answer 3

我相信您可以通过 SQLAlchemy-Migrate 实现这一目标。请注意，ForeignKey 位于隔离列上。 foreignKeyConstraint 位于表级别并将列关联在一起。如果您查看列上的ForeignKey 对象，您将看到它引用了ForeignKeyConstraint。

我无法测试这个想法，因为我使用的两个数据库 MS SQL 不受 SqlAlchemy-Migrate 支持，并且 sqlite 不支持约束的“alter table”。我确实让 SQLAlchemy 尝试通过删除 sqlite 表上的引用约束来删除 FK，所以它看起来不错。 YMMV。

Answer 4

好吧，您可以在 sqlalchemy 中实现此目的：只需在 drop() 列之前 drop() 删除所有约束（理论上，您可能有多个约束）：

def drop_column(column):
    for fk in column.table.foreign_keys:
        if fk.column == column:
            print 'deleting fk ', fk
            fk.drop()
    column.drop()

drop_column(t_exists.c.new_id)