Python Peewee 返回查询|极客笔记

Python Peewee 返回查询

Peewee 是一个轻量级的 Python ORM（对象关系映射）库，可以用来操作各种类型的数据库。在实际开发中，经常需要查询数据库中的数据，并且以某种方式返回查询结果。本文将详细介绍如何使用 Peewee 进行数据库查询，并返回查询结果。

安装 Peewee

要使用 Peewee，首先需要安装它。可以通过 pip 来安装 Peewee：

pip install peewee

安装完成后，就可以在 Python 代码中导入 Peewee 模块了：

from peewee import *

连接数据库

在进行查询之前，需要先连接数据库。Peewee 支持多种类型的数据库，包括 SQLite、MySQL、PostgreSQL 等。这里以 SQLite 为例进行演示。首先，创建一个 SQLite 数据库文件，并在代码中连接该数据库：

# 创建一个 SQLite 数据库文件
db = SqliteDatabase('my_database.db')

# 定义一个基础模型类，并指定数据库
class BaseModel(Model):
    class Meta:
        database = db

然后，使用 connect() 方法连接数据库：

# 连接数据库
db.connect()

定义模型

在 Peewee 中，可以通过定义模型类来映射数据库中的表。以下是一个示例模型类，用来表示一个学生信息表：

# 定义学生模型类
class Student(BaseModel):
    name = CharField()
    age = IntegerField()

在上面的代码中，定义了一个 Student 类，包含两个字段 name 和 age，分别表示学生的姓名和年龄。这些字段的数据类型可以根据实际需求选择。

进行查询

有了模型类之后，就可以利用 Peewee 进行数据库查询了。以下是一些查询示例：

查询所有数据

通过 select() 方法查询所有记录：

# 查询所有学生记录
students = Student.select()
for student in students:
    print(student.name, student.age)

条件查询

可以通过 where() 方法添加查询条件：

# 查询年龄大于等于 18 岁的学生记录
students = Student.select().where(Student.age >= 18)
for student in students:
    print(student.name, student.age)

查询单个记录

如果只需要查询单个记录，可以使用 get() 方法：

# 查询姓名为 Alice 的学生记录
student = Student.get(Student.name == 'Alice')
print(student.name, student.age)

返回查询结果

Peewee 查询返回的结果是一个可迭代对象，可以根据需要进行遍历或转换。以下是一些常见的处理方法：

转换为列表

可以使用 list() 函数将查询结果转换为列表：

students = list(Student.select())
for student in students:
    print(student.name, student.age)

转换为字典

如果需要将查询结果转换为字典，可以使用 dict() 函数：

students_dict = [{ 'name': student.name, 'age': student.age } for student in Student.select()]
for student in students_dict:
    print(student)

序列化为 JSON

Peewee 模型对象支持通过 to_json() 方法将对象序列化为 JSON 格式：

students = Student.select()
json_data = [student.to_json() for student in students]
print(json_data)

完整示例代码

下面是一个完整的示例代码，包括定义模型类、连接数据库、添加数据、进行查询和返回查询结果：

from peewee import *

# 连接数据库
db = SqliteDatabase('my_database.db')
db.connect()

# 定义基础模型类
class BaseModel(Model):
    class Meta:
        database = db

# 定义学生模型类
class Student(BaseModel):
    name = CharField()
    age = IntegerField()

# 创建表
db.create_tables([Student])

# 添加数据
Student.create(name='Alice', age=20)
Student.create(name='Bob', age=22)
Student.create(name='Cathy', age=18)

# 查询所有学生记录
students = Student.select()
for student in students:
    print(student.name, student.age)

# 查询年龄大于等于 18 岁的学生记录
students = Student.select().where(Student.age >= 18)
for student in students:
    print(student.name, student.age)

# 查询姓名为 Alice 的学生记录
student = Student.get(Student.name == 'Alice')
print(student.name, student.age)

# 将查询结果转换为列表
students = list(Student.select())
for student in students:
    print(student.name, student.age)

# 将查询结果转换为字典
students_dict = [{ 'name': student.name, 'age': student.age } for student in Student.select()]
for student in students_dict:
    print(student)

# 将查询结果序列化为 JSON
students = Student.select()
json_data = [student.to_json() for student in students]
print(json_data)

# 关闭数据库连接
db.close()

在上面的代码中，定义了一个 Student 模型类，连接了一个 SQLite 数据库，并使用 Peewee 进行了数据操作和查询。