轉載自：http://www.cnblogs.com/liwenzhou/articles/8283687.html

我們之前使用pymysql操作數據庫的操作都是寫死在視圖函數中的，并且很多都是重復的代碼。

我們可以優化一下，把重復的代碼提取出來，寫成函數：

importpymysql#定義一個數據庫相關的配置項

DB_CONFIG ={"host": "127.0.0.1","port": 3306,"user": "root","passwd": "root1234","db": "mysite","charset": "utf8"}#查詢多條數據函數

def get_list(sql, args=None):

conn=pymysql.connect(

host=DB_CONFIG["host"],

port=DB_CONFIG["port"],

user=DB_CONFIG["user"],

passwd=DB_CONFIG["passwd"],

db=DB_CONFIG["db"],

charset=DB_CONFIG["charset"]

)

cursor= conn.cursor(cursor=pymysql.cursors.DictCursor)

cursor.execute(sql, args)

result=cursor.fetchall()

cursor.close()

conn.close()returnresult#查詢單條數據函數

def get_one(sql, args=None):

conn=pymysql.connect(

host=DB_CONFIG["host"],

port=DB_CONFIG["port"],

user=DB_CONFIG["user"],

passwd=DB_CONFIG["passwd"],

db=DB_CONFIG["db"],

charset=DB_CONFIG["charset"]

)

cursor= conn.cursor(cursor=pymysql.cursors.DictCursor)

cursor.execute(sql, args)

result=cursor.fetchone()

cursor.close()

conn.close()returnresult#修改記錄

def modify(sql, args=None):

conn=pymysql.connect(

host=DB_CONFIG["host"],

port=DB_CONFIG["port"],

user=DB_CONFIG["user"],

passwd=DB_CONFIG["passwd"],

db=DB_CONFIG["db"],

charset=DB_CONFIG["charset"]

)

cursor= conn.cursor(cursor=pymysql.cursors.DictCursor)

cursor.execute(sql, args)

conn.commit()

cursor.close()

conn.close()#創建記錄

def create(sql, args=None):

conn=pymysql.connect(

host=DB_CONFIG["host"],

port=DB_CONFIG["port"],

user=DB_CONFIG["user"],

passwd=DB_CONFIG["passwd"],

db=DB_CONFIG["db"],

charset=DB_CONFIG["charset"]

)

cursor= conn.cursor(cursor=pymysql.cursors.DictCursor)

cursor.execute(sql, args)

conn.commit()#返回剛才創建的那條數據的ID

last_id =cursor.lastrowid

cursor.close()

conn.close()return last_id

這樣只要在需要連接數據庫做操作的時候，只需要調用我們上面定義好的函數就可以了。

但是這樣還是有問題，當我要大批量創建數據的時候，就需要多次調用create方法了，相當于多次連接多次提交。

可以繼續優化下，把數據庫的連接重用，做到只需一次連接就可執行多次操作。

classSQLManager(object):#初始化實例方法

def __init__(self):

self.conn=None

self.cursor=None

self.connect()#連接數據庫

defconnect(self):

self.conn=pymysql.connect(

host=DB_CONFIG["host"],

port=DB_CONFIG["port"],

user=DB_CONFIG["user"],

passwd=DB_CONFIG["passwd"],

db=DB_CONFIG["db"],

charset=DB_CONFIG["charset"]

)

self.cursor= self.conn.cursor(cursor=pymysql.cursors.DictCursor)#查詢多條數據

def get_list(self, sql, args=None):

self.cursor.execute(sql, args)

result=self.cursor.fetchall()returnresult#查詢單條數據

def get_one(self, sql, args=None):

self.cursor.execute(sql, args)

result=self.cursor.fetchone()returnresult#執行單條SQL語句

def moddify(self, sql, args=None):

self.cursor.execute(sql, args)

self.conn.commit()#創建單條記錄的語句

def create(self, sql, args=None):

self.cursor.execute(sql, args)

self.conn.commit()

last_id=self.cursor.lastrowidreturnlast_id#關閉數據庫cursor和連接

defclose(self):

self.cursor.close()

self.conn.close()

我們把我們數據庫的相關操作都封裝成一個類，在用到的時候，只需要生成一個實例，并對實例調用相應的操作方法就可以了。

db =SQLManager()

class_list= db.get_list("select id, name from class")

teacher_info= db.get_list("SELECT teacher.id, teacher.name, teacher2class.class_id FROM teacher LEFT JOIN teacher2class ON teacher.id = teacher2class.teacher_id WHERE teacher.id=%s;", [teacher_id])

db.close()

但是，我如果要批量執行多個創建操作，雖然只建立了一次數據庫連接但是還是會多次提交，可不可以改成一次連接，一次提交呢？

可以，只需要用上pymysql的executemany()方法就可以了。

給我們的 SQLManager類添加一個批量執行的?multi_modify()方法就可以了。

#執行多條SQL語句

def multi_modify(self, sql, args=None):

self.cursor.executemany(sql, args)

self.conn.commit()

現在我們如果一次執行多個創建操作的話就可以使用multi_modify()方法，實現一次連接一次提交了。

最后，我們每次操作完數據庫之后都要手動關閉，可不可以寫成自動關閉的呢？

聯想到我們之前學過的文件操作，使用with語句可以實現縮進結束自動關閉文件句柄的例子。

我們來把我們的數據庫連接類SQLManager類再優化下，使其支持with語句操作。

classSQLManager(object):#初始化實例方法

def __init__(self):

self.conn=None

self.cursor=None

self.connect()#連接數據庫

defconnect(self):

self.conn=pymysql.connect(

host=DB_CONFIG["host"],

port=DB_CONFIG["port"],

user=DB_CONFIG["user"],

passwd=DB_CONFIG["passwd"],

db=DB_CONFIG["db"],

charset=DB_CONFIG["charset"]

)

self.cursor= self.conn.cursor(cursor=pymysql.cursors.DictCursor)#查詢多條數據

def get_list(self, sql, args=None):

self.cursor.execute(sql, args)

result=self.cursor.fetchall()returnresult#查詢單條數據

def get_one(self, sql, args=None):

self.cursor.execute(sql, args)

result=self.cursor.fetchone()returnresult#執行單條SQL語句

def moddify(self, sql, args=None):

self.cursor.execute(sql, args)

self.conn.commit()#執行多條SQL語句

def multi_modify(self, sql, args=None):

self.cursor.executemany(sql, args)

self.conn.commit()#創建單條記錄的語句

def create(self, sql, args=None):

self.cursor.execute(sql, args)

self.conn.commit()

last_id=self.cursor.lastrowidreturnlast_id#關閉數據庫cursor和連接

defclose(self):

self.cursor.close()

self.conn.close()#進入with語句自動執行

def __enter__(self):returnself#退出with語句塊自動執行

def __exit__(self, exc_type, exc_val, exc_tb):

self.close()

現階段，我們只需要優化到這一步就可以，后面的項目實戰中會繼續優化。如使用數據庫連接池等。

總結

以上是生活随笔為你收集整理的python数据库优化_Python学习（二十九）—— pymysql操作数据库优化的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。