Python 进阶(二)：Python使用ORM框架peewee操作MySQL数据库_使用orm操作已有数据库表

前言

本文基于MySQL8.x版本的学习，python版本基于当前最新的3.x，windows操作系统下mysql的安装流程，可参考我之前的文章:
Windows MySQL8.0免安装版，配置多个安装实例

Peewee是一个简单而灵活的Python ORM（对象关系映射）框架，它提供了方便的接口和工具，用于在Python应用程序中进行数据库操作。Peewee支持多种数据库后端，包括MySQL、SQLite、PostgreSQL等，使得在不同数据库系统之间进行切换变得更加容易。

1. 安装Peewee库并初始化数据库

• 1.1 首先安装Peewee库，在terminal中输入命令：

  pip install peewee
  1

  

• 1.2 执行mysql脚本，创建mysql数据库和创建数据表

  安装好mysql数据库后，执行创建数据库命令

  CREATE DATABASE `py_conn_test` CHARACTER SET 'utf8mb4' COLLATE 'utf8mb4_general_ci';
  1

  执行创建数据表操作

  # 书目表
  CREATE TABLE `py_conn_test`.`book`  (
    `id` int UNSIGNED NOT NULL AUTO_INCREMENT,
    `title` varchar(255) NULL COMMENT '书名',
    `author` varchar(255) NULL COMMENT '作者',
    `price` float(10, 2) NULL COMMENT '书名价格',
    `edition` int NULL COMMENT '版次',
    PRIMARY KEY (`id`)
  );
  1
  2
  3
  4
  5
  6
  7
  8
  9

2. 创建数据库连接

在使用Peewee进行数据库操作之前创建一个数据库连接。输入连接参数：数据库名称、用户名、密码、主机和端口

from peewee import *

# 创建MySQL数据库连接
db = MySQLDatabase('py_conn_test', user='zhouquan', password='zhouquan@123',host='localhost', port=4706)
1
2
3
4

更好地组织代码，并使代码更易于维护和扩展。我们将数据库连接和映射类集中在一个地方，并在需要的时候导入并使用

新建database.py文件，数据库连接的代码写入此文件

3. 定义数据表模型类

使用Peewee进行数据库操作，需要定义数据表的模型类。每个模型类对应数据库中的一张表，并定义表中的字段以及其他相关属性。

class Book(Model):
    id = AutoField(primary_key=True)
    title = CharField()
    author = CharField()
    price = FloatField()
    edition = IntegerField()

    class Meta:
        database = db
1
2
3
4
5
6
7
8
9

在上述示例中，定义了一个名为User的模型类，它继承自Peewee的Model类。title 、author 、price 、edition是该数据表的字段。在Meta类中，指定了该模型类使用的数据库连接。

将映射类写入新建的database.py文件

4. 连接数据库并创建表

注意：这里仅做示例，在开发中数据表应该是创建好的！因此可以不执行此模块代码示例
在开始使用模型类进行数据库操作之前，需要先连接数据库并创建相关的表。

# 连接数据库
db.connect()

# 创建表
db.create_tables([Book])
1
2
3
4
5

5. 操作数据库

5.1 插入数据

使用Peewee插入数据非常简单，只需创建模型类的实例，并设置相应的属性值，然后调用save()方法保存到数据库中。

# 引入数据库连接模块和Book映射类
from database import db, Book

# 连接数据库
db.connect()

# 插入数据
book = Book(title="三体一", author="刘慈欣", price=59.5, edition=6)
book.save()

book = Book(title="1984", author="乔治奥威尔", price=47.9, edition=2)
book.save()

book = Book(title="万历十五年", author="黄仁宇", price=125.5, edition=3)
book.save()

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

5.2 查询数据

Peewee提供了丰富的查询API，可以灵活地进行条件查询、排序、限制返回结果等操作。

# 引入数据库连接模块和Book映射类
from database import db, Book

# 连接数据库
db.connect()

books = Book.select()
for book in books:
    print("全部查询：", book.to_string())

print("--------------------------------------------")

books = Book.select().where(Book.price < 100)
for book in books:
    print("价格小于100：", book.to_string())

print("--------------------------------------------")

# 排序查询
books = Book.select().order_by(Book.price.asc())
for book in books:
    print("价格升序排序：", book.to_string())

print("--------------------------------------------")

# 限制返回结果数量
books = Book.select().limit(2)
for book in books:
    print("限制返回两条结果：", book.to_string())

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

查询结果如下：

全部查询： id=5,title=三体一,author=刘慈欣,price=59.5,edition=6
全部查询： id=6,title=1984,author=乔治奥威尔,price=47.9,edition=2
全部查询： id=7,title=万历十五年,author=黄仁宇,price=125.5,edition=3
--------------------------------------------
价格小于100： id=5,title=三体一,author=刘慈欣,price=59.5,edition=6
价格小于100： id=6,title=1984,author=乔治奥威尔,price=47.9,edition=2
--------------------------------------------
价格升序排序： id=6,title=1984,author=乔治奥威尔,price=47.9,edition=2
价格升序排序： id=5,title=三体一,author=刘慈欣,price=59.5,edition=6
价格升序排序： id=7,title=万历十五年,author=黄仁宇,price=125.5,edition=3
--------------------------------------------
限制返回两条结果： id=5,title=三体一,author=刘慈欣,price=59.5,edition=6
限制返回两条结果： id=6,title=1984,author=乔治奥威尔,price=47.9,edition=2
1
2
3
4
5
6
7
8
9
10
11
12
13

5.3 更新数据

使用Peewee更新数据也非常简单，可以直接修改模型实例的属性，并调用save()方法保存修改后的数据到数据库中。

# 引入数据库连接模块和Book映射类
from database import db, Book

# 连接数据库
db.connect()

book = Book.get_by_id(5)
print("修改前：", book.to_string())

book.price = 24.5
book.save()
print("修改后：", book.to_string())

1
2
3
4
5
6
7
8
9
10
11
12
13

5.4 删除数据

使用Peewee删除数据可以通过调用模型实例

的delete_instance()方法或使用delete()方法进行删除操作。

# 引入数据库连接模块和Book映射类
from database import db, Book

# 连接数据库
db.connect()

print("删除前数据：")
books = Book.select()
for book in books:
    print(book.to_string())

book = Book.get(Book.title == '1984')
book.delete_instance()

# 使用where条件删除数据
Book.delete().where(Book.edition == 3).execute()

print("删除后数据：")
books = Book.select()
for book in books:
    print(book.to_string())

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

6. 聚合查询

# 引入数据库连接模块和Book映射类
from peewee import fn

from database import db, Book

# 连接数据库
db.connect()

'''
# 清空book表
delete from book;

# 执行聚合前先进行mysql的插入,插入数据用于聚合
INSERT INTO `py_conn_test`.`book`( `title`, `author`, `price`, `edition`) VALUES ( '三体二', '刘慈欣', 24.5, 6);
INSERT INTO `py_conn_test`.`book`( `title`, `author`, `price`, `edition`) VALUES ( '三体三', '刘慈欣', 24.5, 6);
INSERT INTO `py_conn_test`.`book`( `title`, `author`, `price`, `edition`) VALUES ( '1984', '乔治奥威尔', 47.9, 2);
INSERT INTO `py_conn_test`.`book`( `title`, `author`, `price`, `edition`) VALUES ( '动物农场', '乔治奥威尔', 47.9, 2);
INSERT INTO `py_conn_test`.`book`( `title`, `author`, `price`, `edition`) VALUES ( '伊豆舞女', '川端康成', 47.9, 2);
'''

# 聚合出数据库中书目大于等于2的作者信息
group_query = (Book
               .select(Book.author, fn.Count(Book.id).alias('count'))
               .group_by(Book.author)
               .having(fn.Count(Book.id) >= 2))

print("查询SQL打印：", group_query)

# 打印聚合结果
for book in group_query:
    print(f"标题: {book.author}, 计数: {book.count}")
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

查询结果：

查询SQL打印： SELECT `t1`.`author`, Count(`t1`.`id`) AS `count` FROM `book` AS `t1` GROUP BY `t1`.`author` HAVING (Count(`t1`.`id`) >= 2)
标题: 刘慈欣, 计数: 2
标题: 乔治奥威尔, 计数: 2
1
2
3

示例中展示了一些Peewee的基本用法和常见操作，但Peewee还有更多的功能和选项可供探索，例如事务管理、连接池、复杂查询等。可以参考Peewee的官方文档（http://docs.peewee-orm.com/）以了解更多详细信息和示例。

以上为使用Peewee对mysql的基础的应用，至于复杂的语句查询如果有人需要或者看的话再写吧