Python中如何正确的使用事务

在 Python 中，如果要使用事务，通常需要使用数据库连接和事务管理模块。下面以 MySQL 数据库为例，演示如何使用 PyMySQL 模块和事务管理器来完成事务操作。

首先，需要安装 PyMySQL 模块。可以使用 pip install pymysql 命令进行安装。

接下来，可以使用以下代码创建一个数据库连接对象，并开启自动提交（autocommit）：

import pymysql

# 创建数据库连接对象
connection = pymysql.connect(
    host="localhost",
    user="username",
    password="password",
    database="database_name",
    cursorclass=pymysql.cursors.DictCursor,
    autocommit=True
)

注意，开启自动提交后，每个 SQL 语句都将作为一个单独的事务提交，因此无法使用事务回滚。

如果要使用事务，可以使用 connection.begin() 方法开启一个事务，使用 connection.commit() 提交事务，使用 connection.rollback() 回滚事务。事务的使用方法示例代码如下：

import pymysql

# 创建数据库连接对象
connection = pymysql.connect(
    host="localhost",
    user="username",
    password="password",
    database="database_name",
    cursorclass=pymysql.cursors.DictCursor,
    autocommit=False
)

try:
    with connection.cursor() as cursor:
        # 开始事务
        connection.begin()

        # 执行 SQL 语句
        cursor.execute("INSERT INTO table_name (column1, column2) VALUES (%s, %s)", ("value1", "value2"))

        # 提交事务
        connection.commit()

except:
    # 出现异常，回滚事务
    connection.rollback()

finally:
    # 关闭数据库连接
    connection.close()

注意，在使用事务时，需要在 connection 对象上调用 begin() 方法来开启事务，在执行 SQL 语句时，使用 with connection.cursor() as cursor: 语句块创建游标对象，并在语句块内执行 SQL 语句。如果出现异常，需要在 except 语句块内调用 rollback() 方法来回滚事务。最后，在 finally 语句块内关闭数据库连接。

需要注意的是，在使用事务时，应该尽量减少事务的持续时间，避免长时间占用数据库连接和资源。此外，为了保证事务的正确性，应该在代码中添加必要的异常处理和错误处理逻辑。