前言

任何應(yīng)用都離不開數(shù)據(jù)，所以在學(xué)習(xí)python的時(shí)候，當(dāng)然也要學(xué)習(xí)一個(gè)如何用python操作數(shù)據(jù)庫(kù)了。MySQLdb就是python對(duì)mysql數(shù)據(jù)庫(kù)操作的模塊。今天寫了個(gè)工具，目的是把csv中的數(shù)據(jù)插入到數(shù)據(jù)庫(kù)中去。其中有一部分，是需要分別向兩張表中插入兩條數(shù)據(jù)，如果第二張表中的數(shù)據(jù)已經(jīng)存在，那么第一張表中的數(shù)據(jù)也不需要插入。

然后通過百度查找發(fā)現(xiàn)，其實(shí)MySQLdb庫(kù)，自帶了事務(wù)處理的功能，pymysql庫(kù)也是一樣。

            
conn = MySQLdb.connect()返回一個(gè)連接對(duì)象

cursor = conn.cursor()返回一個(gè)游標(biāo)對(duì)象
          

當(dāng)我們使用 cursor.execute(SQL) 執(zhí)行一條Insert/Update/Delete語(yǔ)句后，我們需要使用 conn.commit() 提交事務(wù)，否則，語(yǔ)句不會(huì)生效。

其實(shí)我們常用的 commit() ，就是包含了一種事務(wù)處理的概念，如果你在 commit() 之前，執(zhí)行了多條語(yǔ)句，只有當(dāng) commit() 之后，才會(huì)全部生效。

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

            
conn.rollback()
          

這個(gè)語(yǔ)句可以使事務(wù)回滾，直到上一條 conn.commit() 執(zhí)行之后的位置。

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

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

總結(jié)

這就可以達(dá)到我們的目的了。其實(shí)更優(yōu)雅的方法是，對(duì)事務(wù)處理進(jìn)行一個(gè)封裝，將處理過程寫在自己的類的方法中，并接受1個(gè)SQL語(yǔ)句序列（可能包含1條或多條SQL語(yǔ)句）作為參數(shù)，返回事務(wù)執(zhí)行狀態(tài)即可。以上就是這篇文章的全部?jī)?nèi)容，希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作能帶來一定的幫助。