pyhton 操作mysql数据库的事务怎么写？

Question

我有一个需求：
1、导入新数据前trunk清空mysql中A表的数据；
2、导入A表新数据。
2个操作要么一起成功，要么一起失败，是不是应当用事务来做？怎么写？请教，谢谢！

Answer 1

import mysql.connector

try:
    conn = mysql.connector.connect(host='localhost',
                                   database='数据库名字',
                                   user='用户',
                                   password='密码')

    conn.autocommit = False
    cursor = conn.cursor()
    # 清空A表数据 
    sql_update_query = """TRUNCATE TABLE A;"""
    cursor.execute(sql_update_query)

    # 导入数据
    sql_update_query = """insert into table_name values();"""
    cursor.execute(sql_update_query)
    print("success")

    # 提交
    conn.commit()

except mysql.connector.Error as error:
    print("fail, rollback: {}".format(error))
    # 如果失败则回滚
    conn.rollback()
finally:
    # 关闭数据库连接
    if conn.is_connected():
        cursor.close()
        conn.close()
        print("connection is closed")