1. 程式人生 > >Python ORM框架之 Peewee入門

Python ORM框架之 Peewee入門

lob shortcuts pymysql 主鍵 也會 roo username fault 有意思

  之前在學Django時,發現它的模型層非常好用,把對數據庫的操作映射成對類、對象的操作,避免了我們直接寫在Web項目中SQL語句,當時想,如果這個模型層可以獨立出來使用就好了,那我們平臺操作數據庫也可以這麽玩了,我不喜歡寫SQL語句。

  後來才知道,原來這個叫ORMObject Relational Mapping,對象關系映射),在Python下面有很多這樣的類庫,如SQLObjectStormpeeweeSQLAlchemy

  這裏就給你們介紹一下Peewee的基本使用,因為它非常的輕量級,最主要是和DjangoORM 操作很像,如果你學過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入門