淺析Python中MySQLdb的交易處理功能,pythonmysqldb
前言
任何應用都離不開資料,所以在學習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()執行之後的位置。
所以,My Code是這樣寫的,這裡我就簡單的寫了,並沒有用類來封裝了:
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語句)作為參數,返回事務執行狀態即可。以上就是這篇文章的全部內容,希望本文的內容對大家的學習或者工作能帶來一定的協助。