Python 操作mysql实现事务处理

一、应用场景

Python项目对MySQL数据库进行增、删、改操作时，有时会出现执行sql异常的情况。在批量提交数据的时候，如果其中一个事务提交错误，往往导致预期的整个数据链不完整。
例如银行转账数据，用户A向用户B转账：

• 步骤一：此时A的数据要存一份某个时间向B转账的数据。

• 步骤二：同时用户B也会存一份某个时间收到A转账的数据。

如果此时步骤一执行sql成功，步骤二出错并且未回滚事务，将会导致两边备份的数据校对错误。当引入事务时，步骤二出现错误，事务回滚后步骤一未执行，数据回归到出错前的情况，保证了数据的安全性。

二、使用方法

配合我们之前文章提到的try的使用（Python之异常处理（try的基本用法）），还有上篇文章学到的mysql的基本操作（Python MySQL数据库连接）实现捕捉异常并且回滚。

try:
    cursor.execute(sql_1)  
    cursor.execute(sql_2)  
    cursor.execute(sql_3)  
except Exception as e:
		# 事务回滚
    connect.rollback()  
    print('事务处理失败', e)
else:
		# 事务提交
    connect.commit()  
    print('事务处理成功', cursor.rowcount)

三、测试demo

事务出现错误demo

# -*- coding: utf-8 -*-

import pymysql
if __name__ == '__main__':
    db = pymysql.connect(host='localhost',
                         user='root',
                         password='root',
                         database='others')

    cursor = db.cursor(pymysql.cursors.DictCursor)
    select_sql = " select id,name from user_info "
    cursor.execute(select_sql)
    data_list = cursor.fetchall()
    print("执行更新前的数据-->")
    print(data_list)

    try:
        update_sql1 = " update user_info set name='1：code_space' where id = 1 "
        update_sql2 = " update user_info set name='2：code_space' where id = 2 "
        # 数据库的name长度是20，这里长度超过了，会报Data too long错误
        update_sql3 = " update user_info set name='测试测试测试测试测试测试测试测试测试测试测试测试' where id = 3 "
        
				cursor.execute(update_sql1)
        cursor.execute(update_sql2)
        cursor.execute(update_sql3)

    except Exception as e:
        # 事务回滚
        db.rollback()
        print('事务处理失败', e)
    else:
        # 事务提交
        db.commit()
        print('事务处理成功', cursor.rowcount)

    cursor.execute(select_sql)
    data_list = cursor.fetchall()
    print("执行更新后的数据-->")
    print(data_list)

    cursor.close()
    db.close()