Python Peewee

轉載：https://www.jianshu.com/p/8d1bdd7f4ff5

Peewee系列：

• Peewee 使用
• Peewee使用之事務
• Peewee批量插入數據
• Peewee 使用（二）——增刪改查更詳細使用

---

Peewee是一個簡單小巧的Python ORM，它非常容易學習，並且使用起來很直觀。

如果想快速入門，請參考官方的Quckstart。

本文，只是寫今天在使用過程中的一些記錄。

基本知識

在官方的Quckstart中，我了解到，Peewee中Model類、fields和model實例與數據庫的映射關系如下：

 

 

也就是說，一個Model類代表一個數據庫的表，一個Field字段代表數據庫中的一個字段，而一個model類實例化對象則代表數據庫中的一行。

至於Peewee的實現原理，我暫時沒有看源代碼，但覺得和廖雪峰老師的使用元類這個文章的例子實現類似。

實踐

而使用過程，分成兩步：

定義Model，建立數據庫

在使用的時候，根據需求先定義好Model，然后可以通過create_tables()創建表，若是已經創建好數據庫表了，可以通過python -m pwiz腳本工具直接創建Model。

第一種方式：

先定義Model，然后通過db.create_tables()創建或Model.create_table()創建表。

例如，我們需要建一個Person表，里面有name、birthday和is_relative三個字段，我們定義的Model如下：

from peewee import *

# 連接數據庫
database = MySQLDatabase('test', user='root', host='localhost', port=3306)

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

    class Meta:
        database = database

然后，我們就可以創建表了

 

# 創建表
Person.create_table()

# 創建表也可以這樣, 可以創建多個
# database.create_tables([Person])

其中，CharField、DateField、BooleanField等這些類型與數據庫中的數據類型一一對應，我們直接使用它就行，至於CharField => varchar(255)這種轉換Peewee已經為我們做好了 。

第二種方式：

已經存在過數據庫，則直接通過python -m pwiz批量創建Model。
例如，上面我已經創建好了test庫，並且創建了Person表，表中擁有id、name、birthday和is_relative字段。那么，我可以使用下面命令：

# 指定mysql，用戶為root，host為localhost，數據庫為test
python -m pwiz -e mysql -u root -H localhost --password test > testModel.py

然后，輸入密碼，pwiz腳本會自動創建Model，內容如下：

 

from peewee import *

database = MySQLDatabase('test', **{'charset': 'utf8', 'use_unicode': True, 'host': 'localhost', 'user': 'root', 'password': ''})

class UnknownField(object):
    def __init__(self, *_, **__): pass

class BaseModel(Model):
    class Meta:
        database = database

class Person(BaseModel):
    birthday = DateField()
    is_relative = IntegerField()
    name = CharField()

    class Meta:
        table_name = 'person'

操作數據庫

操作數據庫，就是增、刪、改和查。

一、增

直接創建示例，然后使用save()就添加了一條新數據

# 添加一條數據
p = Person(name='liuchungui', birthday=date(1990, 12, 20), is_relative=True)
p.save()

二、刪

使用delete().where().execute()進行刪除，where()是條件，execute()負責執行語句。若是已經查詢出來的實例，則直接使用delete_instance()刪除。

# 刪除姓名為perter的數據
Person.delete().where(Person.name == 'perter').execute()

# 已經實例化的數據, 使用delete_instance
p = Person(name='liuchungui', birthday=date(1990, 12, 20), is_relative=False)
p.id = 1
p.save()
p.delete_instance()

三、改

若是，已經添加過數據的的實例或查詢到的數據實例，且表擁有primary key時，此時使用save()就是修改數據；若是未擁有實例，則使用update().where()進行更新數據。

# 已經實例化的數據,指定了id這個primary key,則此時保存就是更新數據
p = Person(name='liuchungui', birthday=date(1990, 12, 20), is_relative=False)
p.id = 1
p.save()

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

四、查

單條數據使用Person.get()就行了，也可以使用Person.select().where().get()。若是查詢多條數據，則使用Person.select().where()，去掉get()就行了。語法很直觀，select()就是查詢，where是條件，get是獲取第一條數據。

# 查詢單條數據
p = Person.get(Person.name == 'liuchungui')
print(p.name, p.birthday, p.is_relative)

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

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