[转] 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中的andor在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使用之事务

参考

Peewee querying
peewee 查询



作者:liuchungui
链接:https://www.jianshu.com/p/ba8a27cf7da1
来源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。




评论

支持上传图片(拖动图片或者截图粘贴)

0 评论