前言

任何應用都離不開數據，所以在學習python的時候，當然也要學習一個如何用python操作數據庫了。MySQLdb就是python對mysql數據庫操作的模塊。今天寫了個工具，目的是把csv中的數據插入到數據庫中去。其中有一部分，是需要分別向兩張表中插入兩條數據，如果第二張表中的數據已經存在，那么第一張表中的數據也不需要插入。

然后通過百度查找發現，其實MySQLdb庫，自帶了事務處理的功能，pymysql庫也是一樣。

            
conn = MySQLdb.connect()返回一個連接對象

cursor = conn.cursor()返回一個游標對象
          

當我們使用 cursor.execute(SQL) 執行一條Insert/Update/Delete語句后，我們需要使用 conn.commit() 提交事務，否則，語句不會生效。

其實我們常用的 commit() ，就是包含了一種事務處理的概念，如果你在 commit() 之前，執行了多條語句，只有當 commit() 之后，才會全部生效。

那么如果像文章開始提到的那樣，執行兩條插入語句，我們可以把 conn.commit() 語句放在兩條插入語句執行之后，并放在try語句塊內，如果差生異常，我們可以使用：

            
conn.rollback()
          

這個語句可以使事務回滾，直到上一條 conn.commit() 執行之后的位置。

所以，我的代碼是這樣寫的，這里我就簡單的寫了，并沒有用類來封裝了：

            
try:
cursor = conn.cursor()
cursor.execute(insertSQL1)
cursor.execute(insertSQL2)
cursor.close()
conn.commit()
except Exception, e:
cursor.close()
conn.rollback()
          

總結

這就可以達到我們的目的了。其實更優雅的方法是，對事務處理進行一個封裝，將處理過程寫在自己的類的方法中，并接受1個SQL語句序列（可能包含1條或多條SQL語句）作為參數，返回事務執行狀態即可。以上就是這篇文章的全部內容，希望本文的內容對大家的學習或者工作能帶來一定的幫助。