Python ORM框架之 Peewee入门

悠悠小仙仙 · 发表于 2017-6-21 14:24:39

之前在学Django时，发现它的模型层非常好用，把对数据库的操作映射成对类、对象的操作，避免了我们直接写在Web项目中SQL语句，当时想，如果这个模型层可以独立出来使用就好了，那我们平台操作数据库也可以这么玩了，我不喜欢写SQL语句。

　　后来才知道，原来这个叫ORM（Object Relational Mapping，对象关系映射），在Python下面有很多这样的类库，如SQLObject、Storm、peewee和SQLAlchemy。

　　这里就给你们介绍一下Peewee的基本使用，因为它非常的轻量级，最主要是和Django的ORM 操作很像，如果你学过Django那么很容易上手。
         以操作MySQL为例。前提条件：
      Python3.x 编程语言
      MySQL数据库
      PyMySQL驱动

一、安装peewee
安装非常简单，推荐使用pip命令。

> pip install 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方法用法：