今天寫了個tool,目的是把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語句)作為參數,返回事務執行狀態即可。