pytest测试套件运行结束后如何执行退出命令清理SQLAlchemy动态表

pytest 原生提供了测试生命周期管理能力，远比 atexit 适配性更强，不会出现回调不触发的问题，推荐以下两种方案：

• 方案1：使用 session 级自动执行 fixture（最常用）
  将清理逻辑放在 session 级 fixture 的 yield 语句后，pytest 会在整个测试会话全部运行完成后自动执行该部分逻辑。示例代码如下：

import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import declarative_base

Base = declarative_base()
engine = create_engine("你的数据库连接字符串")
# 全局存储动态生成的表对象，若你所有表都注册到Base可不用该变量
GLOBAL_DYNAMIC_TABLES = []

@pytest.fixture(scope="session", autouse=True)
def auto_clean_dynamic_tables(request):
    # 测试会话启动前的预处理逻辑（可选）
    yield
    # 仅当无测试用例失败时才执行清理，若需要无论结果都清理可去掉判断
    if request.session.testsfailed == 0:
        # 方式1：所有表都注册到Base时直接调用drop_all
        Base.metadata.drop_all(bind=engine, checkfirst=True)
        # 方式2：动态生成的表未注册到Base时，遍历全局列表删除
        # for table in GLOBAL_DYNAMIC_TABLES:
        #     table.drop(bind=engine, checkfirst=True)

如果需要每个测试用例/模块运行完就清理对应表，可调整 fixture 的 scope 参数为 function/module，适配不同的隔离需求。

• 方案2：使用 pytest_sessionfinish 全局钩子
  在项目根目录的 conftest.py 文件中添加 pytest_sessionfinish 钩子函数，该函数会在 pytest 整个测试会话结束时固定触发：

def pytest_sessionfinish(session, exitstatus):
    from your_module import GLOBAL_DYNAMIC_TABLES, engine
    for table in GLOBAL_DYNAMIC_TABLES:
        table.drop(bind=engine, checkfirst=True)

注意事项 #

• 删表操作添加 checkfirst=True 参数，避免表不存在时抛出异常
• 动态创建表时统一维护全局存储列表，避免漏删或错删非测试生成的业务表
• 若使用多进程运行 pytest，需确保全局表存储逻辑适配多进程场景，避免跨进程变量无法共享导致清理不全

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