基于pymysql写一个数据库回滚的python脚本
时间: 2024-03-29 15:40:10 浏览: 123
好的,以下是一个基于pymysql写的数据库回滚的Python脚本示例:
```python
import pymysql
# 连接数据库
conn = pymysql.connect(host='localhost', port=3306, user='root', password='password', db='test')
try:
# 开始事务
with conn.cursor() as cursor:
cursor.execute('START TRANSACTION')
# 执行需要回滚的SQL语句
with conn.cursor() as cursor:
cursor.execute('INSERT INTO users (id, name, age) VALUES (1, "Tom", 20)')
cursor.execute('INSERT INTO users (id, name, age) VALUES (2, "Jerry", 22)')
# 回滚事务
with conn.cursor() as cursor:
cursor.execute('ROLLBACK')
# 提交事务
conn.commit()
except:
# 发生异常时回滚事务
conn.rollback()
finally:
# 关闭数据库连接
conn.close()
```
以上示例中,首先连接数据库,然后使用`with`语句开启事务,并执行需要回滚的SQL语句。如果执行过程中出现异常,则回滚事务,否则提交事务。最后关闭数据库连接。
阅读全文