程序员求职经验分享与学习资料整理平台

网站首页 > 文章精选 正文

Python 操作mysql实现事务处理 python+操作mysql实现事务处理功能

balukai 2024-12-23 16:52:57 文章精选 49 ℃

一、应用场景

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

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

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

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

二、使用方法

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 -*-
"""
@Time : 2022/1/27 17:34
@Auth : 技术空间
@File :rollback_demo.py
@IDE :PyCharm
@Motto:技术总是要日积月累的

"""
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()
  • 事务正确执行demo
# -*- coding: utf-8 -*-
"""
@Time : 2022/1/27 17:34
@Auth : 技术空间
@File :rollback_demo.py
@IDE :PyCharm
@Motto:技术总是要日积月累的

"""
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 "
        update_sql3 = " update user_info set name='3我的头条号:code_space' 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()

四、拓展

在处理事务过程中,往往涉及到批量执行sql语句的情况,比如批量提交订单、批量删除订单等。下篇文章我们将会讲解关于批量执行sql语句的代码,涉及到代码复用的重要操作。还请多多关注。

关注我,坚持每日积累一个技巧,长期坚持,我们将会不断进步。

#程序员##python##数据库##教育听我说##请回答,你的年度知识点#

最近发表
标签列表