在一個Web App中,所有數(shù)據(jù),包括用戶信息、發(fā)布的日志、評論等,都存儲在數(shù)據(jù)庫中。在awesome-python-app中,我們選擇MySQL作為數(shù)據(jù)庫。
Web App里面有很多地方都要訪問數(shù)據(jù)庫。訪問數(shù)據(jù)庫需要創(chuàng)建數(shù)據(jù)庫連接、游標(biāo)對象,然后執(zhí)行SQL語句,最后處理異常,清理資源。這些訪問數(shù)據(jù)庫的代碼如果分散到各個函數(shù)中,勢必?zé)o法維護(hù),也不利于代碼復(fù)用。
此外,在一個Web App中,有多個用戶會同時訪問,系統(tǒng)以多進(jìn)程或多線程模式來處理每個用戶的請求。假設(shè)以多線程為例,每個線程在訪問數(shù)據(jù)庫時,都必須創(chuàng)建僅屬于自身的連接,對別的線程不可見,否則,就會造成數(shù)據(jù)庫操作混亂。
所以,我們還要創(chuàng)建一個簡單可靠的數(shù)據(jù)庫訪問模型,在一個線程中,能既安全又簡單地操作數(shù)據(jù)庫。
為什么不選擇SQLAlchemy?SQLAlchemy太龐大,過度地面向?qū)ο笤O(shè)計導(dǎo)致API太復(fù)雜。
所以我們決定自己設(shè)計一個封裝基本的SELECT、INSERT、UPDATE和DELETE操作的db模塊:transwarp.db。
設(shè)計db接口
設(shè)計底層模塊的原則是,根據(jù)上層調(diào)用者設(shè)計簡單易用的API接口,然后,實現(xiàn)模塊內(nèi)部代碼。
假設(shè)transwarp.db模塊已經(jīng)編寫完畢,我們希望以這樣的方式來調(diào)用它:
首先,初始化數(shù)據(jù)庫連接信息,通過create_engine()函數(shù):
from transwarp import dbdb.create_engine(user='root', password='password', database='test', host='127.0.0.1', port=3306)
然后,就可以直接操作SQL了。
如果需要做一個查詢,可以直接調(diào)用select()方法,返回的是list,每一個元素是用dict表示的對應(yīng)的行:
users = db.select('select * from user')# users =># [# { "id": 1, "name": "Michael"},# { "id": 2, "name": "Bob"},# { "id": 3, "name": "Adam"}# ]
如果要執(zhí)行INSERT、UPDATE或DELETE操作,執(zhí)行update()方法,返回受影響的行數(shù):
n = db.update('insert into user(id, name) values(?, ?)', 4, 'Jack')
update()函數(shù)簽名為:
update(sql, *args)
統(tǒng)一用?作為占位符,并傳入可變參數(shù)來綁定,從根本上避免SQL注入攻擊。
每個select()或update()調(diào)用,都隱含地自動打開并關(guān)閉了數(shù)據(jù)庫連接,這樣,上層調(diào)用者就完全不必關(guān)心數(shù)據(jù)庫底層連接。
但是,如果要在一個數(shù)據(jù)庫連接里執(zhí)行多個SQL語句怎么辦?我們用一個with語句實現(xiàn):
with db.connection(): db.select('...') db.update('...') db.update('...')
如果要在一個數(shù)據(jù)庫事務(wù)中執(zhí)行多個SQL語句怎么辦?我們還是用一個with語句實現(xiàn):
with db.transaction(): db.select('...') db.update('...') db.update('...')
新聞熱點
疑難解答
圖片精選