Peewee SQL|极客教程

Peewee SQL

Peewee SQL是一个Python轻量级的ORM（对象关系映射）库，它让开发者可以用Python代码来操作数据库，而不需要直接编写SQL语句。Peewee支持SQLite、MySQL和PostgreSQL等多种数据库，并且使用简单，学习曲线平缓，适合快速搭建小型应用程序或原型。在本文中，我们将详细介绍Peewee SQL的基本用法和一些高级特性。

安装Peewee

首先，我们需要安装Peewee库。可以使用pip来安装：

pip install peewee

安装完成后，我们就可以开始使用Peewee来操作数据库了。

连接数据库

在使用Peewee之前，我们需要先连接数据库。以下是一个连接SQLite数据库的示例代码：

from peewee import SqliteDatabase, Model, CharField

# 创建一个SQLite数据库连接
db = SqliteDatabase('geek-docs.db')

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

# 连接数据库
db.connect()

在以上示例代码中，我们创建了一个SQLite数据库的连接，并定义了一个基类模型BaseModel，用来继承Peewee提供的Model类。

定义数据库模型

在Peewee中，我们可以通过定义模型类来映射数据库表。以下是一个示例代码，定义了一个User表：

class User(BaseModel):
    username = CharField(unique=True)
    password = CharField()

在以上示例中，我们定义了一个User类，继承自BaseModel。User表包含了两个字段：username和password，分别用于存储用户名和密码。

创建数据库表

在定义了模型类之后，我们需要通过create_table()方法来创建实际的数据库表。示例代码如下：

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

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

以上代码中，我们通过create_tables()方法创建了User表。需要注意的是，在操作完数据库后，一定要关闭数据库连接。

插入数据

接下来，我们可以通过模型类来插入新的数据。以下是一个插入数据的示例代码：

# 重新连接数据库
db.connect()

# 插入数据
user = User.create(username='geekdocs', password='password123')

在以上示例中，我们通过create()方法向User表中插入了一条数据。

查询数据

Peewee提供了灵活的查询方式，可以方便地获取数据库中的数据。以下是一个查询数据的示例代码：

# 查询数据
user = User.get(User.username == 'geekdocs')

# 打印查询结果
print(user.username, user.password)

以上代码中，我们使用get()方法根据用户名查询到了对应的用户，并打印了用户名和密码。

更新数据

除了新增和查询数据外，我们还可以通过Peewee更新数据库中的数据。以下是一个更新数据的示例代码：

# 更新数据
user.password = 'newpassword456'
user.save()

在以上示例中，我们修改了用户的密码，并通过save()方法将修改提交到数据库中。

删除数据

最后，我们还可以通过Peewee删除数据库中的数据。以下是一个删除数据的示例代码：

# 删除数据
user.delete_instance()

在以上示例中，我们通过delete_instance()方法删除了指定数据。

高级特性

除了基本的增删改查操作外，Peewee还提供了一些高级特性，例如事务、复杂查询、聚合函数等。以下是一个使用事务的示例代码：

# 开启事务
with db.atomic():
    # 插入数据
    User.create(username='geekdocs1', password='password123')
    User.create(username='geekdocs2', password='password456')

以上代码中，我们使用了db.atomic()来开启一个事务，确保两次插入操作要么同时成功，要么同时失败。