Python的輕量級ORM框架peewee

專欄作者：小小明

非常擅長解決各類復(fù)雜數(shù)據(jù)處理的邏輯，各類結(jié)構(gòu)化與非結(jié)構(gòu)化數(shù)據(jù)互轉(zhuǎn)，字符串解析匹配等等。至今已經(jīng)幫助很多數(shù)據(jù)從業(yè)者解決工作中的實際問題，如果你在數(shù)據(jù)處理上遇到什么困難，歡迎評論區(qū)與我交流。

在前面的《改變python對象規(guī)則的黑魔法metaclass》一文中，我介紹了使用metaclass自己編寫ORM框架的思路。

當然python庫中這類框架非常多，我們并沒有必要自己去實現(xiàn)。

ORM框架使用最廣泛的就是SQLAlchemy和Django自帶的ORM框架，但是SQLAlchemy的語法顯然相對Django的ORM框架麻煩一點。

而Django本身是一個web框架，比較重量級，僅僅為了使用Django的ORM框架的功能，而安裝Django有點導(dǎo)致系統(tǒng)臃腫。而peewee這個框架語法幾乎與Django的ORM框架一致，而又非常輕量。

它的安裝非常簡單：

pip install peewee

如果你在使用mysql數(shù)據(jù)庫的過程中報出如下錯誤：

peewee.ImproperlyConfigured: MySQL driver not installed!

則需要安裝一個mysql的驅(qū)動：

pip install pymysql

peewee的whl包是880kB，pymysql的whl包是51KB，非常輕量級。

peewee的官方文檔地址：http://docs.peewee-orm.com/en/latest/index.html

下面測試一下各項功能：

from peewee import *

db = MySQLDatabase('test', host="localhost", user='root', passwd='123456', port=3306)


# 定義Person
class Person(Model):
    name = CharField()
    birthday = DateField()
    is_relative = BooleanField()

    class Meta:
        database = db
        
def test_create():
    Person.create_table()
    # 創(chuàng)建多張表也可以這樣
    # database.create_tables([Person])


def test_insert():
    # 添加一條數(shù)據(jù)
    p = Person(name='小華', birthday=date(1996, 12, 20), is_relative=True)
    p.save()


def test_delete():
    # 刪除姓名為perter的數(shù)據(jù)
    Person.delete().where(Person.name == 'perter').execute()
    # 已經(jīng)實例化的數(shù)據(jù), 使用delete_instance
    p = Person(name='小華', birthday=date(1996, 12, 20), is_relative=False)
    p.id = 1
    p.save()
    p.delete_instance()


def test_update():
    # 已經(jīng)實例化的數(shù)據(jù),指定了id這個primary key,則此時保存就是更新數(shù)據(jù)
    p = Person(name='小華', birthday=date(1996, 12, 20), is_relative=False)
    p.id = 1
    p.save()

    # 更新birthday數(shù)據(jù)
    q = Person.update({Person.birthday: date(1983, 12, 21)}).where(Person.name == '小華')
    q.execute()


def test_query():
    # 查詢單條數(shù)據(jù)
    p = Person.get(Person.name == '小華')
    print(p.name, p.birthday, p.is_relative)

    # 使用where().get()查詢
    p = Person.select().where(Person.name == '小華').get()
    print(p.name, p.birthday, p.is_relative)

    # 查詢多條數(shù)據(jù)
    persons = Person.select().where(Person.is_relative == True)
    for p in persons:
        print(p.name, p.birthday, p.is_relative)

下面測試一個各個方法。

測試創(chuàng)建表：

if __name__=="__main__":
    Person.create_table()

執(zhí)行完畢，檢查數(shù)據(jù)庫成功創(chuàng)建下面這張表：

測試插入數(shù)據(jù)：

if __name__=="__main__":
    p = Person(name='小華', birthday=date(1996, 12, 20), is_relative=True)
    p.save()

執(zhí)行完畢后，表數(shù)據(jù)多了一行：

測試查詢數(shù)據(jù)：

if __name__=="__main__":
 p = Person.get(Person.name == '小華')
    print(p.name, p.birthday, p.is_relative)

結(jié)果：

小華 1996-12-20 True

測試刪除數(shù)據(jù)：

if __name__=="__main__":
    Person.delete().where(Person.name == '小華').execute()

執(zhí)行后，數(shù)據(jù)庫對應(yīng)的記錄被刪除：

測試修改數(shù)據(jù)：

if __name__ == "__main__":
    p = Person(name='小新', birthday=date(1995, 6, 20), is_relative=False)
    p.save()
    # 更新birthday數(shù)據(jù)
    q = Person.update({Person.birthday: date(1983, 5, 21)}).where(Person.name == '小新')
    q.execute()

執(zhí)行后：

測試批量查詢：

if __name__ == "__main__":
    for i in range(1, 5):
        p = Person(name=f'小張{i}', birthday=date(1995, 6, 20), is_relative=False)
        p.save()
    # 查詢多條數(shù)據(jù)
    persons = Person.select().where(Person.is_relative == False)
    for p in persons:
        print(p.name, p.birthday, p.is_relative)

執(zhí)行后：

結(jié)果：

小新 1983-05-21 False
小張1 1995-06-20 False
小張2 1995-06-20 False
小張3 1995-06-20 False
小張4 1995-06-20 False

更多用法可參考官方文檔。