Python的輕量級ORM框架peewee使用教程

Posted on by

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

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

它的安裝非常簡單：

pip install peewee

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

peewee.ImproperlyConfigured: MySQL driver not installed!

則需要安裝一個mysql的驅動：

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()
  # 創建多張表也可以這樣
  # database.create_tables([Person])


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


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


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

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


def test_query():
  # 查詢單條數據
  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)

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

下面測試一個各個方法。

測試創建表：

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

執行完畢，檢查數據庫成功創建下面這張表：

測試插入數據：

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

執行完畢後，表數據多瞭一行：

測試查詢數據：

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

結果：

小華 1996-12-20 True

測試刪除數據：

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

執行後，數據庫對應的記錄被刪除：

測試修改數據：

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

測試批量查詢：

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

以上就是Python的輕量級ORM框架peewee使用教程的詳細內容，更多關於Python的輕量級ORM框架peewee的資料請關註WalkonNet其它相關文章！

推薦閱讀：