这是Python程序开发系列原创文章，我的第201篇原创文章。

一、背景知识

Peewee库：

Peewee 是一个 Python 编程语言中的轻量级 ORM（对象关系映射）库，用于简化与关系型数据库的交互。它提供了简洁的API，使得在 Python 中进行数据库操作更加方便和易于管理。Peewee 支持多种关系型数据库，包括 SQLite、MySQL、PostgreSQL 和更多。它提供了以下主要功能：

• 定义模型：使用 Peewee，你可以定义数据库中的表结构和字段。通过定义模型类，你可以将 Python 对象映射到数据库表，并使用类属性来表示表中的字段。

• 数据库连接和管理：Peewee 提供了用于连接和管理数据库的功能。你可以指定数据库引擎、主机、端口、用户名、密码等信息来连接到数据库。

• 数据库操作：Peewee 提供了用于执行 CRUD（创建、读取、更新、删除）操作的方法，如插入数据、查询数据、更新数据和删除数据。

• 查询构建器：Peewee 提供了一个强大的查询构建器，可以使用链式调用的方式构建复杂的数据库查询。你可以使用条件、排序、限制和连接等操作来精确地获取所需的数据。

SQLite：

SQLite 是一种嵌入式关系型数据库管理系统，它是一个零配置的数据库引擎，不需要单独的服务器进程，数据以文件形式存储在本地。SQLite 是一个轻量级的数据库，适用于小型应用程序或移动应用程序等场景。以下是一些关于 SQLite 数据库的重要特点：

• 零配置：SQLite 不需要单独的服务器进程或配置文件，数据以文件形式存储在本地。这使得它易于部署和使用。

• 单用户：SQLite 是一个单用户数据库，只能由一个进程访问。这意味着它不适用于多用户并发访问的场景。

• 轻量级：SQLite 是一个轻量级的数据库引擎，它的核心库非常小巧，运行时资源消耗较低。

• 支持标准 SQL：SQLite 支持标准的 SQL 语法和查询语句，包括常见的 SQL 操作和聚合函数。

• 事务支持：SQLite 支持事务操作，可以通过事务来确保数据的完整性和一致性。

• ACID 特性：SQLite 支持 ACID（原子性、一致性、隔离性和持久性）特性，保证数据的安全性和可靠性。

• 跨平台：SQLite 可以在多个操作系统上运行，包括 Windows、Mac、Linux 等。

二、实现过程

2.1 连接数据库

from peewee import *# 对于Mysql数据库，需要建立连接
# db = MySQLDatabase("spider", host="127.0.0.1", port=3306, user="root", password="123456")
# db.connect()
# 对于SQLite数据库，可以直接创建并连接
db = SqliteDatabase(dbDir + 'main.db')   # dbDir是数据库文件的位置
db.connect()class BaseModel(Model):class Meta:database = db     # 指定数据库连接

使用 SqliteDatabase 类在项目文件夹下来创建一个 SQLite 数据库文件。

2.2 设计表

#设计数据表的时候有几个重要点一定要注意
"""
char类型， 要设置最大长度
对于无法确定最大长度的字段，可以设置为Text
设计表的时候 采集到的数据要尽量先做格式化处理
default和null=True
"""# 创建模型类
class Person(Model):name = CharField()age = IntegerField()class Meta:table_name = "person"#不设置主键将自动生成一个默认主键id

设计了一个Person表，包含name、age两个字段，数据类型分别是Char和Integer。

2.3 创建表

db.create_tables([Person]) #括号内写表名，要加中号号（只有一个表也要加）
# db.create_tables([Person, XXX, XXXX])  # 创建多个表

2.4 将列表数据依次插入表中某个字段

names = ['a', 'b', 'c']
for name in names:Person.get(Person.name == name)

将列表里面的元素依次写入Person的name字段里面。

作者简介：

读研期间发表6篇SCI数据挖掘相关论文，现在某研究院从事数据算法相关科研工作，结合自身科研实践经历不定期分享关于Python、机器学习、深度学习、人工智能系列基础知识与应用案例。致力于只做原创，以最简单的方式理解和学习，关注我一起交流成长。需要数据集和源码的小伙伴可以关注底部公众号添加作者微信！