Python ORM框架之 Peewee入門
之前在學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 = db
class 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:32
True
不過,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
Python ORM框架之 Peewee入門