在Python中如何通过异常回滚自动管理数据库事务-Python教程-PHP中文网

不自动。with语句配合数据库连接（如sqlite3、psycopg2、pymysql）仅自动关闭连接，不捕获异常也不调用rollback()；事务回滚需显式处理或借助封装层（如sqlalchemy session）。

在python中如何通过异常回滚自动管理数据库事务

Python中`with`语句配合数据库连接是否自动回滚？

不自动。标准库如sqlite3或主流驱动（psycopg2、pymysql）的Connection对象本身**不提供事务级异常回滚语义**；with conn:只保证连接关闭，不捕获异常也不调用rollback()。

常见误解是以为写了with conn:就能像Django ORM那样出错自动回滚——实际不会。必须显式处理异常分支。

sqlite3.Connection 的 __enter__ 返回自身，__exit__ 仅调用 close()，不碰事务状态
psycopg2.connect() 同理，__exit__ 不执行 rollback()
只有封装层（如SQLAlchemy的Session）才默认在异常时回滚

手动实现`with`风格的事务管理器

最轻量且可控的方式是写一个上下文管理器，显式控制commit/rollback时机。

from contextlib import contextmanager
<p>@contextmanager
def transaction(conn):
try:
yield conn
conn.commit()
except Exception:
conn.rollback()
raise
</p>

使用时：

Galileo python sdk

Galileo AI 平台 Python SDK 完整参考，用于评估、监控和保护 GenAI 应用，适用于构建 Python 应用。

下载

with transaction(conn) as c:
    c.execute("INSERT INTO users(name) VALUES (?)", ("alice",))
    c.execute("UPDATE accounts SET balance = balance - 100 WHERE id = 1")

成功时调用 conn.commit()
任何异常（包括 KeyboardInterrupt）都会触发 conn.rollback() 并重新抛出
注意：该管理器不处理连接生命周期，只管事务边界

SQLAlchemy中`Session`的`rollback`行为细节

如果你用的是SQLAlchemy，Session确实在异常退出with块时自动回滚，但有几个关键前提和陷阱：

必须用 with Session(engine) as session:，而不是 session = Session() 后手动 with
回滚只清理Session内部的 pending/deleted/modified 状态，**不释放数据库锁、不终止正在执行的语句**
若在 session.execute() 中触发了数据库级错误（如唯一约束冲突），rollback() 仍会执行，但部分后端（如PostgreSQL）可能已隐式结束当前事务，此时再调 rollback() 是空操作
不要在 except 块里重复调 session.rollback() —— with 已做完

多语句批量操作中部分失败怎么回滚？

如果一组SQL需原子性执行（例如插入主表+关联子表），但中间某条失败，仅靠单个conn.rollback()是够的；难点在于**避免“半提交”状态被其他连接看到**。

真正要注意的是隔离级别和执行顺序：

确保所有操作在同一个Connection上完成（不能每个execute都新开游标或连接）
对MySQL，避免在AUTOCOMMIT=False下混用DDL（如CREATE TABLE），它会隐式提交当前事务
PostgreSQL中，若用psycopg2，注意cursor.execute()不自动开启事务，首次DML才开始事务；所以要在第一条DML前确认没意外commit()
测试时用SELECT ... FOR UPDATE类语句要格外小心——异常回滚后锁会释放，但业务逻辑可能已基于旧数据做判断