前言
任何應用都離不開數據,所以在學習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語句)作為參數,返回事務執行狀態即可。以上就是這篇文章的全部內容,希望本文的內容對大家的學習或者工作能帶來一定的幫助。
新聞熱點
疑難解答