之前在學Django時,發現它的模型層非常好用,把對數據庫的操作映射成對類、對象的操作,避免了我們直接寫在Web項目中SQL語句,當時想,如果這個模型層可以獨立出來使用就好了,那我們平臺操作數據庫也可以這麼玩了,我不喜歡寫SQL語句。
後來才知道,原來這個叫ORM(Object Relational Mapping,對象關係映射),在Python下面有很多這樣的類庫,如SQLObject、Storm、peewee和SQLAlchemy。
這裏就給你們介紹一下Peewee的基本使用,因爲它非常的輕量級,最主要是和Django的ORM 操作很像,如果你學過Django那麼很容易上手。
這篇博客承接上一篇,以操作MySQL爲例。前提條件:
Python3.x 編程語言
MySQL數據庫
PyMySQL驅動
一、安裝peewee
安裝非常簡單,推薦使用pip命令。
> pip install peewee
另外,你也可以通過下載包安裝:https://pypi.python.org/pypi/peewee
二、創建表
from peewee import *import datetime db = MySQLDatabase("guest", host="127.0.0.1", port=3306, user="root", passwd="123456") db.connect()class BaseModel(Model): class Meta: database = dbclass User(BaseModel): username = CharField(unique=True)class Tweet(BaseModel): user = ForeignKeyField(User, related_name='tweets') message = TextField() created_date = DateTimeField(default=datetime.datetime.now) is_published = BooleanField(default=True)if __name__ == "__main__": # 創建表 User.create_table() # 創建User表 Tweet.create_table() # 創建Tweet表
這裏麪包含不了少知識點,我們來一一解釋。
首先,導入peewee庫下面的所有方法,這個當然需要。
然後,通過MySQLDatabase連接數據庫,把數據連接的幾個必要參數一一填寫。通過connect()方法與MySQL數據庫建立鏈接。
接下來就是表的創建,創建BaseModel類,在該類下創建子類Meta,Meta是一個內部類,它用於定義peewee的Model類的行爲特性。指定dabatase 爲 前面定義的db。
再接下來就是表的創建了,我們在SQL語句創建表時一般需要知道以下信息。表的名字,表有哪些字段?這些字段分別是什麼類型?是否允許爲空,或自增?哪個字段是主鍵是?哪個是外鍵?
ORM用編程語言裏的概念幫我們映射了這些東西。
創建 User 和 Tweet類做爲表名。在類下面定義的變量爲字段名,如username、message、created_date等。通過CharField、DateTimeField、BooleanField表示字段的類型。ForeignKeyField 建立外鍵。 主鍵呢? 建表時不都要有個主鍵id嘛,不用!peewee默認已經爲我們加上這個id了。
最後,執行create_table()方法創建兩張表。
通過數據庫工具,查看生成的兩張表。
三、插入數據
要想操作表數據,首先表裏得有數據。先來看看如何添加數據。
if __name__ == "__main__": # ... user = User.create(username='tom') Tweet.create(user=user, message="這是一段文字")
這樣就要User表裏添加一個tom的用戶,這用戶發了一條Tweet,在Tweet表裏。但這個用戶興致來了,想繼續發第二條Tweet。於是:
if __name__ == "__main__": # ... Tweet.create(user_id=1, message="這是第二段文字")
咦~!?不對,我們沒有創建user_id字段啊!但是,如果你查詢Tweet表,就會發現有這個字段,用它來關聯User表的id。
四、查詢數據
接下來,查詢數據。
if __name__ == "__main__": # ... # 查詢 1 t = Tweet.get(message="這是一段文字") print(t.user_id) print(t.created_date) print(t.is_published)
查詢結果:
1 2017-05-19 15:44:32True
不過,get()方法只能查詢一條,且是唯一的一條數據;通過查詢條件不能查詢出多條,也不能查詢出0條。
if __name__ == "__main__": # ... # 查詢 2 ts = Tweet.filter(user_id=1) for t in ts: print(t.message)
運行結果:
這是一段文字 這是第二段文字
而,filter()方法,就會更加靈活,可以查詢多條結果,並把結果循環輸出。
五、Playhouse庫
Playhouse庫中提供了一些比較有意思的方法。當我們在安裝peewee時,也會捎帶着把它裝上。
這裏只介紹它裏面的model_to_dict和dict_to_model兩個方法。
model_to_dict方法用法:
from playhouse.shortcuts import model_to_dict, dict_to_model# ……if __name__ == "__main__": # …… user = User.create(username='jack') # 把數據對象轉成字典 u = model_to_dict(user) print(u)
運行結果:
{'id': 7, 'username': 'jack'}
dict_to_model方法用法:
from playhouse.shortcuts import model_to_dict, dict_to_model# ……if __name__ == "__main__": # …… # 把字典轉成數據對象 user_data = {'id': 2, 'username': 'charlie'} user = dict_to_model(User, user_data) print(user.username)
運行結果:
charlie