Peewee 使用（二）——增删改查更详细使用

Peewee系列：
Peewee 使用
Peewee使用之事务
Peewee批量插入数据
Peewee 使用（二）——增删改查更详细使用

---

在四月份刚接触Peewee的时候，写过一篇Peewee 使用。而后，在使用的过程中，发现很多常用的内容需要搜索查阅，今天就在这里整理一下。

一、插入数据

插入数据，我们可以实例化一个Model，然后再使用save()的方法插入到数据库中。如下：

# 插入一条数据
p = Person(name='liuchungui', birthday=date(1990, 12, 20), is_relative=True)
p.save()

除了上面，我最常用的是insert()方法直接插入数据，它会返回新插入数据的主键给我们。

# 插入一条数据
p_id = Person.insert({
    'name': 'liuchungui'
}).execute()
 # 打印出新插入数据的id
print(p_id)

上面都是插入一条数据，若是有很多数据需要插入，例如几万条数据，为了性能，这时就需要使用insert_many()，如下：

NUM = 10000
data = [{
            'name': '123'
        } for i in range(NUM)]

with database.atomic():
    for i in range(0, NUM, 100):
        # 每次批量插入100条，分成多次插入
        Person.insert_many(data[i:i + 100]).execute()

至于为啥要使用insert_many()，可以看看我前面写的Peewee批量插入数据。

二、查询数据

1、查询单条数据

我们可以直接使用get()获取单条数据，在参数中传递查询条件。

# 查询name为liuchungui的Person
p = Person.get(Person.name == 'liuchungui')
print(p.name) # 打印出liuchungui

2、查询多条数据

使用select()查询，后面不添加where()是查询整个表的内容。

# 查询Person整张表的数据
persons = Person.select()
# 遍历数据
for p in persons:
    print(p.name, p.birthday, p.is_relative)

我们可以在select()后面添加where()当做查询条件

# 获取is_relative为True的数据
persons = Person.select().where(Person.is_relative == True)
for p in persons:
    print(p.name, p.birthday, p.is_relative)

我们可以通过sql()方法转换为SQL语句进行查看理解

persons = Person.select().where(Person.is_relative == True)
# 打印出的结果为：('SELECT `t1`.`id`, `t1`.`name`, `t1`.`is_relative` FROM `Person` AS `t1` WHERE (`t1`.`is_relative` = %s)', [True])
print(persons.sql())

3、查询数据条数、排序、Limit

查询数据条数，直接在后面加上count()就行了

# 查询整张表的数据条数
total_num = Person.select().count()

# 查询name为liuchungui的Person数量, 返回数量为1
num = Person.select().where(Person.name == 'liuchungui').count()

排序，使用的是order_by()，参数内加上按对应字段进行排序

# 按照创建时间降序排序
persons = Person.select().order_by(Person.create_time.desc())

# 按照创建时间升序排序
persons = Person.select().order_by(Person.create_time.asc())

Limit是使用limit()，传递一个数字，例如2就是获取前两条数据，它可以搭配offset()一起使用

# 相当于sql语句: select * from person order by create_time desc limit 5
persons = Person.select().order_by(Person.create_time.asc()).limit(5)

# 相当于sql语句中：select * from person order by create_time desc limit 2, 5
persons = Person.select().order_by(Person.create_time.asc()).limit(5).offset(2)

三、更新数据

当一个Model实例拥有主键时，此时使用save()就是修改数据

# 已经实例化的数据,指定了id这个primary key,则此时保存就是更新数据
p = Person(name='liuchungui', birthday=date(1990, 12, 20), is_relative=False)
p.id = 1
p.save()

也可以使用update()来更新数据，一般都会搭配where()使用

# 更新birthday数据
q = Person.update({Person.height: 1.75}).where(Person.name == 'Jack')
q.execute()

当然，除了使用Model的属性，我们可以直接使用字典结构来更新数据

q = Person.update({
    'height': 1.75
}).where(Person.name == 'Jack')
q.execute()

四、查询操作符

在查询、更新、删除数据的时候，经常会带有Where条件语句。而Peewee支持以下类型比较符：

image.png

其中，==、<、<=、>、>=、!=是很容易理解的，重点提下<<、>>和%。用示例说明：

# <<使用，查询省份属于湖北和湖南的，对应sql语句：select * from person where province in ('湖南', '湖北')
persons = Person.select().where(Person.province << ['湖南', '湖北'])

# >>使用，查询省份为空的，sql语句: select * from person where province is Null
persons = Person.select().where(Person.province >> None)

# %使用，查询省份中含有 湖 字，sql语句：select * from person where province like '%湖%'
persons = Person.select().where(Person.province % '%湖%')

有时，我们查询条件不止一个，需要使用逻辑运算符连接，而Python中的and、or在Peewee是不支持的，此时我们需要使用Peewee封装好的运算符，如下：

image.png

使用示例如下：

# 查询湖南和湖北的, 注意需要用()将Person.province == '湖南'包一层
persons = Person.select().where((Person.province == '湖南') | (Person.province == '湖北'))

# 查询湖南和身高1.75
persons = Person.select().where((Person.province == '湖南') & (Person.height == 1.75))

注意：使用的时候，需要内部还使用()将Person.province == '湖南'包起来，否则不会生效。示例：persons = Person.select().where((Person.province == '湖南') | (Person.province == '湖北'))

除了上面的操作符以外，Peewee还有更多没有重载的操作符，如下：

image.png

五、联表查询

有时，我们需要查询两个表中的数据，在Peewee中也可以实现，官方示例如下：

query = (Tweet
         .select(Tweet.content, Tweet.timestamp, User.username)
         .join(User, on=(User.id == Tweet.user_id))
         .order_by(Tweet.timestamp.desc()))

上面查询的结果，会在Tweet的Model中添加一个属性user，此时我们可以通过user来访问到查询到的User表信息，如下：

for tweet in query:
    print(tweet.content, tweet.timestamp, tweet.user.username)

六、事务

Peewee实现事务最常用的方法是Database.atomic()方法，使用起来非常简单，如下：

from xModels import XUser, database

with database.atomic() as transaction:
    XUser.create(phone='184738373833', password='123456')
    XUser.create(phone='184738373833332323232', password='123456')

更多可参考我前面写的Peewee使用之事务

参考

Peewee querying
peewee 查询