SQLAlchemy 2.0中重建ForeignKeyConstraint报错，如何正确实现？

我尝试在SQLAlchemy中临时删除表的ForeignKeyConstraint以高效执行插入操作，为此编写了如下上下文管理器来处理约束的删除与重建：

from contextlib import contextmanager
from typing import Iterator

from sqlalchemy import ForeignKeyConstraint
from sqlalchemy.orm import Session
from sqlalchemy.sql.ddl import CreateConstraint, DropConstraint


@contextmanager
def drop_constraint(
    session: Session, constraint: ForeignKeyConstraint
) -> Iterator[None]:
    session.execute(DropConstraint(constraint))
    yield
    session.execute(CreateConstraint(constraint))

删除约束的操作正常，但在退出上下文管理器重建约束时，出现如下错误：

ArgumentError: Executable SQL or text() construct expected, got <sqlalchemy.sql.ddl.CreateConstraint object at ...>.

请问如何使用纯SQLAlchemy 2.0动态创建该约束？

在SQLAlchemy 2.0中，CreateConstraint无法直接通过session.execute()执行，需改用数据库连接对象执行DDL操作，或直接调用约束对象的create()方法完成重建。

修正后的上下文管理器代码如下：

from contextlib import contextmanager
from typing import Iterator

from sqlalchemy import ForeignKeyConstraint
from sqlalchemy.orm import Session
from sqlalchemy.sql.ddl import DropConstraint


@contextmanager
def drop_constraint(
    session: Session, constraint: ForeignKeyConstraint
) -> Iterator[None]:
    conn = session.connection()
    # 删除约束
    conn.execute(DropConstraint(constraint))
    yield
    # 调用约束对象的create方法重建
    constraint.create(conn)

关键说明 #

• 改用session.connection()获取数据库连接，这是SQLAlchemy 2.0中执行DDL操作的推荐方式，比直接用session.execute()更适配DDL语句的执行逻辑。
• 重建约束时调用constraint.create(conn)，该方法会自动生成正确的CREATE CONSTRAINT语句并执行，避免了CreateConstraint对象的兼容性问题。
• 确保传入的ForeignKeyConstraint对象已绑定到具体表（比如从表对象的constraints属性获取，或创建时指定了table参数），否则create()方法无法生成正确的SQL语句。

内容的提问来源于stack exchange，提问作者Roméo Desprès