SQLAlchemy中如何为Base.metadata.drop_all添加级联解决删除报错

我正在为pytest测试搭建初始化（setup）与清理（teardown）流程。由于表之间存在大量依赖关系，执行drop_all时抛出psycopg2.errors.DependentObjectsStillExist错误，无法删除表，但仍需要在每次测试后清空所有表。

当前的pytest fixture代码：

@pytest.fixture()
def test_db():
    model.Base.metadata.create_all(bind=test_engine)
    db = TestSessionLocal()
    yield db
    model.Base.metadata.drop_all(bind=test_engine)

已在模型中添加级联设置但未生效：

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    email = Column(String, unique=True, index=True)
    name = Column(String)
    surname = Column(String)
    position = Column(String)
    skills = relationship("UserSkills", cascade="all, delete", passive_deletes=True)
    manager = Column(Integer, ForeignKey("manager.id", ondelete='CASCADE'))
    account_type = Column(Enum(AccountType), default=AccountType.USER)
    created = Column(DateTime, default=datetime.datetime.utcnow)

请问如何为model.Base.metadata.drop_all(bind=test_engine)添加级联选项？

方法1：按依赖逆序删除表 #

SQLAlchemy默认按表创建顺序执行drop_all，这会导致父表先被删除、子表依赖未清理的问题。可以手动指定删除顺序，或利用sorted_tables获取依赖逆序的表列表：

@pytest.fixture()
def test_db():
    model.Base.metadata.create_all(bind=test_engine)
    db = TestSessionLocal()
    yield db
    # 按依赖逆序删除表
    for table in reversed(model.Base.metadata.sorted_tables):
        table.drop(bind=test_engine, cascade=True)

方法2：执行原生SQL级联清空Schema #

直接通过原生SQL清空整个测试Schema，适合快速清理所有表对象：

@pytest.fixture()
def test_db():
    model.Base.metadata.create_all(bind=test_engine)
    db = TestSessionLocal()
    yield db
    # 清空public schema并重建
    with test_engine.connect() as conn:
        conn.execute("DROP SCHEMA public CASCADE; CREATE SCHEMA public;")
        conn.commit()

注意：此方法会删除Schema下所有对象（表、视图、序列等），仅适合测试环境使用。

方法3：为drop_all指定级联参数 #

SQLAlchemy 1.4+版本支持在drop_all中传入cascade参数，直接生成带级联的DDL语句：

@pytest.fixture()
def test_db():
    model.Base.metadata.create_all(bind=test_engine)
    db = TestSessionLocal()
    yield db
    # 添加cascade=True启用级联删除
    model.Base.metadata.drop_all(bind=test_engine, cascade=True)

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