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支持以下类型比较符:
其中,==、<、<=、>、>=、!=是很容易理解的,重点提下<<
、>>
和%
。用示例说明:
# <<使用,查询省份属于湖北和湖南的,对应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封装好的运算符,如下:
使用示例如下:
# 查询湖南和湖北的, 注意需要用()将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还有更多没有重载的操作符,如下:
五、联表查询
有时,我们需要查询两个表中的数据,在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使用之事务