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 查询

from peewee import * 
db = SqliteDatabase('sampleDB.db')
class BaseModel(Model):
    class Meta:
        database = db  
class Course(BaseModel):
    id = PrimaryKeyField()
    title = CharField(null=False)
    period = IntegerField()
    description = CharField()
    class Meta:
        order_by = ('title',)
        db_table = 'course'
class Teacher(BaseModel):
    id = PrimaryKeyField()
    name = CharField(null=False)
    gender = BooleanField()
    address = CharField()
    course_id = ForeignKeyField(Course,to_field='id',related_name="course")
    class Meta:
        order_by = ('name',)
        db_table = 'teacher'
#创建表
Course.create_table()
Teacher.create_table()
#新增行
Course.create(id=1,title="经济学",period=320,description="文理科学生均可选修")
Course.create(id=2,title="大学英语",period=300,description="大一学生必修")
Course.create(id=3,title="哲学",period=200,description="课必修")
Teacher.create(name="张雯雯",gender=False,address="..",course_id=2)
#查询一行
record =  Course.get(title="大学英语")
print("课程:%s,学时:%d"%(record.title,record.period))
#更新
record.period = 200 
record.save() 
#删除
record.delete_instance()
#查询所有记录
courses = Course.select() 
#带条件查询，并将结果按period字段倒序排列
courses = Course.select().where(Course.id<10).order_by(Course.period.desc())
for item in courses:
    print(item)
#统计所有课程的平均学时
total = Course.select(fn.Avg(Course.period).alias('avg_period'))

#更新多个记录
Course.update(period=300).where(Course.id>100).execute()
#多表连接操作，Peewee 会自动根据ForeignKeyField的外键定义进行连接
Record = Course.select().join(Teacher).where(Teacher.gender==True)