python - 在 "with"关键字打开资源期间捕获异常

Question

举个具体的例子，以psycopg2 for Postgres为例(我知道mysql.connection有类似的API):

try:
    with closing(connection.cursor()) as cursor:
        ... # run a transaction with cursor, like insert
except IntegrityError as e:
    cursor.rollback()

在这种情况下游标是否能够回滚，或者在这种情况下游标资源是否会在回滚发生之前关闭？

Answer 1

你能交换 try: 和 with ... 吗？

with closing(connection.cursor()) as cursor:
    try:
        ... # run a transaction with cursor, like insert
    except IntegrityError as e:
        cursor.rollback()

这样，当到达except ...: 时，光标将在当前范围内。

使用您的代码，当到达 except ...: 时， cursor 将为 None。