在 Python 的開(kāi)發(fā)世界中，數(shù)據(jù)庫(kù)操作是至關(guān)重要的一環(huán)。

今天介紹的 Peewee 作為一款簡(jiǎn)潔且功能強(qiáng)大的 ORM（對(duì)象關(guān)系映射）框架，為開(kāi)發(fā)者提供了高效便捷的數(shù)據(jù)庫(kù)交互方式。

1. Peewee概述

Peewee 是一個(gè)簡(jiǎn)單小巧的 ORM，它的概念簡(jiǎn)潔明了，易于學(xué)習(xí)和使用。

能夠與 SQLite、MySQL、MariaDB、PostgreSQL 等多種數(shù)據(jù)庫(kù)協(xié)同工作，擁有豐富的擴(kuò)展功能，其源代碼托管于 GitHub-peewee。

使用過(guò)Python的都知道，SQLAlchemy幾乎已經(jīng)是Python中的標(biāo)準(zhǔn)ORM框架了，功能強(qiáng)大，

為什么還要使用Peewee呢？

首先，Peewee 設(shè)計(jì)更為簡(jiǎn)潔，其 API 簡(jiǎn)單直觀，學(xué)習(xí)曲線平緩，新手能快速上手，而 SQLAlchemy 相對(duì)復(fù)雜，需要花費(fèi)更多時(shí)間去掌握。

其次，Peewee 代碼量較少，在一些簡(jiǎn)單項(xiàng)目中，其輕量級(jí)的特點(diǎn)能使項(xiàng)目結(jié)構(gòu)更清晰，開(kāi)發(fā)效率更高。

例如在小型數(shù)據(jù)庫(kù)應(yīng)用場(chǎng)景下，Peewee 能快速搭建起數(shù)據(jù)操作模塊。

再者，Peewee 的性能在特定場(chǎng)景下表現(xiàn)出色，如對(duì) SQLite 數(shù)據(jù)庫(kù)的操作，其資源占用相對(duì)較低，能為應(yīng)用帶來(lái)更好的運(yùn)行效果。

總之，如果項(xiàng)目規(guī)模不大，或者做一些小工具，那么Peewee 會(huì)更加趁手。

2. 快速上手

2.1. 初始化數(shù)據(jù)庫(kù)

針對(duì)不同的數(shù)據(jù)庫(kù)類型，有相應(yīng)的初始化方式。

下面我們選擇使用SQLite：

from peewee import SqliteDatabase

db = SqliteDatabase('my_database.db')

2.2. 模型定義

在 Peewee 中，通過(guò)定義類來(lái)創(chuàng)建模型，類的屬性對(duì)應(yīng)數(shù)據(jù)庫(kù)表中的字段。例如：

from peewee import Model, CharField, IntegerField

class User(Model):
    class Meta:
        database = db

    username = CharField(unique=True)
    age = IntegerField()

2.3. 創(chuàng)建數(shù)據(jù)庫(kù)和表

連接數(shù)據(jù)庫(kù)，然后就可通過(guò)SqliteDatabase來(lái)創(chuàng)建表。

if __name__ == "__main__":
    db.connect()
    db.create_tables([User])

執(zhí)行之后，就會(huì)發(fā)現(xiàn)創(chuàng)建了sqlite數(shù)據(jù)庫(kù)和表。

$  sqlite3.exe .\my_database.db
SQLite version 3.45.3 2024-04-15 13:34:05 (UTF-16 console I/O)
Enter ".help" for usage hints.
sqlite> .tables
user

db.create_tables反復(fù)執(zhí)行也沒(méi)關(guān)系，如果表已經(jīng)存在，不會(huì)重復(fù)創(chuàng)建。

2.4. 數(shù)據(jù)存儲(chǔ)與檢索

存儲(chǔ)數(shù)據(jù)時(shí)，先創(chuàng)建模型實(shí)例并賦值，然后調(diào)用 save 方法即可將數(shù)據(jù)保存到數(shù)據(jù)庫(kù)。

if __name__ == "__main__":
    user = User(username="Harry", age=23)
    user.save()

運(yùn)行之后，查詢數(shù)據(jù)庫(kù)，發(fā)現(xiàn)數(shù)據(jù)已經(jīng)寫入了數(shù)據(jù)庫(kù)。

sqlite> select * from user;
┌────┬──────────┬─────┐
│ id │ username │ age │
├────┼──────────┼─────┤
│ 1  │ Harry    │ 23  │
└────┴──────────┴─────┘

檢索數(shù)據(jù)可以使用各種查詢方法。如獲取單個(gè)記錄：

user = User.get(User.username == "Harry")
print(f"name: {user.username}, age: {user.age}")

# 運(yùn)行結(jié)果：
# name: Harry, age: 23

2.5. 更新記錄

更新記錄，比如將上面的年齡改為30。

User.update(age=30).where(User.username == 'Harry').execute()

運(yùn)行之后：

sqlite> select * from user;
┌────┬──────────┬─────┐
│ id │ username │ age │
├────┼──────────┼─────┤
│ 1  │ Harry    │ 30  │
└────┴──────────┴─────┘

2.6. 刪除記錄

刪除記錄也很簡(jiǎn)單：

User.delete().where(User.username == 'Harry').execute()

運(yùn)行之后：

sqlite> select * from user;
sqlite> select count(1) from user;
┌──────────┐
│ count(1) │
├──────────┤
│ 0        │
└──────────┘

3. 高級(jí)查詢功能

高級(jí)的查詢功能包括多條件過(guò)濾，排序以及分頁(yè)查詢等等。

3.1. 批量插入數(shù)據(jù)

為了演示高級(jí)查詢功能，先批量插入一批數(shù)據(jù)。

User.insert_many(users, fields=[User.username, User.age]).execute()

運(yùn)行結(jié)果：

sqlite> select * from user;
┌────┬──────────┬─────┐
│ id │ username │ age │
├────┼──────────┼─────┤
│ 1  │ harry    │ 23  │
│ 2  │ lily     │ 20  │
│ 3  │ tom      │ 35  │
│ 4  │ jerry    │ 12  │
│ 5  │ kate     │ 42  │
└────┴──────────┴─────┘

3.2. 多條件查詢

多個(gè)條件的交集，比如id>2并且age>30的數(shù)據(jù)：

users = User.select().where((User.id > 2) & (User.age > 30)).execute()

print("滿足條件的用戶：")
for u in users:
    print(f"{u.username}: {u.age}")

運(yùn)行結(jié)果：

$  python.exe .\main.py
滿足條件的用戶：
tom: 35
kate: 42

多個(gè)條件的并集，比如id>4或者age>20的數(shù)據(jù)：

users = User.select().where((User.id > 4) | (User.age > 20)).execute()

運(yùn)行結(jié)果：

$  python.exe .\main.py
滿足條件的用戶：
harry: 23
tom: 35
kate: 42

3.3. 排序

按照年齡增長(zhǎng)排序：

users = User.select().order_by(User.age)

運(yùn)行結(jié)果：

$  python.exe .\main.py
按照年齡增長(zhǎng)排序：
jerry: 12
lily: 20
harry: 23
tom: 35
kate: 42

按照年齡減少方向排序：

users = User.select().order_by(User.age.desc())

運(yùn)行結(jié)果：

$  python.exe .\main.py
按照年齡減少排序：
kate: 42
tom: 35
harry: 23
lily: 20
jerry: 12

3.4. 分頁(yè)查詢

最后，再來(lái)看看分頁(yè)查詢，這在前端展示大規(guī)模數(shù)據(jù)時(shí)非常有用。

一般的ORM會(huì)通過(guò)SQL語(yǔ)句中的limit和offset來(lái)實(shí)現(xiàn)分頁(yè)查詢，而Peewee直接提供了分頁(yè)查詢的API。

page_number = 1 # 頁(yè)序號(hào)，從1開(kāi)始
page_size = 3  # 每頁(yè)數(shù)據(jù)的數(shù)量

users = User.select().paginate(page_number, page_size)
print(f"第{page_number}頁(yè)數(shù)據(jù)：")
for u in users:
    print(f"{u.username}: {u.age}")

運(yùn)行結(jié)果：

$  python.exe .\main.py
第1頁(yè)數(shù)據(jù)：
harry: 23
lily: 20
tom: 35

這樣就顯示了前3個(gè)數(shù)據(jù)，如果把上面的page_numberg=2，那么會(huì)返回剩下的2條數(shù)據(jù)。

4. 總結(jié)

Peewee 還擁有眾多擴(kuò)展，如 Playhouse 提供了更多高級(jí)功能，包括對(duì)不同數(shù)據(jù)庫(kù)的特定擴(kuò)展（如 SQLite 的擴(kuò)展函數(shù)）、模型生成工具、數(shù)據(jù)庫(kù)遷移工具、反射功能等，大大增強(qiáng)了 Peewee 的實(shí)用性和靈活性。

本篇介紹的是最基本的使用方法，其他還有多表之間關(guān)系的建立和查詢，請(qǐng)參考官方的文檔。

總之，Peewee 以其簡(jiǎn)潔的語(yǔ)法、豐富的功能和良好的擴(kuò)展性，成為 Python 開(kāi)發(fā)者在數(shù)據(jù)庫(kù)操作方面的有力工具，無(wú)論是小型項(xiàng)目還是大型應(yīng)用，都能提供高效可靠的數(shù)據(jù)庫(kù)交互支持。

到此這篇關(guān)于Python簡(jiǎn)潔強(qiáng)大的ORM框架Peewee的用法詳解的文章就介紹到這了,更多相關(guān)Python Peewee內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論