pymysql mysql 封装实战

原创

mob64ca12e7f20c 2024-10-04 04:06:20 ©著作权

文章标签 Database 数据库操作封装 文章分类 MySQL 数据库

©著作权归作者所有：来自51CTO博客作者mob64ca12e7f20c的原创作品，请联系作者获取转载授权，否则将追究法律责任

PyMySQL与MySQL的封装实战

在日常的开发过程中，数据的存储和管理是非常重要的，而MySQL作为一种流行的关系型数据库，广泛应用于后端开发中。在Python中，PyMySQL 是一个非常常用的MySQL客户端库，今天我们将探讨如何封装 PyMySQL ，以实现更高效的数据库操作。

1. 引入PyMySQL

首先，你需要安装 PyMySQL 。可以使用以下命令进行安装：

pip install pymysql

2. 创建数据库连接类

为了方便后续的数据库操作，我们可以创建一个 Database 类。这个类将封装常用的数据库操作，包括连接数据库、执行查询和关闭连接等。

import pymysql

class Database:
    def __init__(self, host, user, password, database):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.connection = None

    def connect(self):
        self.connection = pymysql.connect(
            host=self.host,
            user=self.user,
            password=self.password,
            database=self.database,
            charset='utf8mb4',
            cursorclass=pymysql.cursors.DictCursor
        )

    def execute_query(self, query, params=None):
        with self.connection.cursor() as cursor:
            cursor.execute(query, params)
            return cursor.fetchall()

    def execute_update(self, query, params=None):
        with self.connection.cursor() as cursor:
            cursor.execute(query, params)
            self.connection.commit()

    def close(self):
        if self.connection:
            self.connection.close()

3. 使用示例

我们可以通过实例化 Database 类来使用封装好的数据库操作。

if __name__ == '__main__':
    db = Database(host='localhost', user='root', password='your_password', database='your_database')
    
    try:
        db.connect()
        
        # 查询数据
        results = db.execute_query("SELECT * FROM your_table WHERE id=%s", (1,))
        print(results)

        # 更新数据
        db.execute_update("UPDATE your_table SET name=%s WHERE id=%s", ('new_name', 1))
    finally:
        db.close()

4. 状态图

在进行数据库操作的过程中，我们可以通过状态图来描述连接的各个状态，如下所示：

stateDiagram
    [*] --> 初始状态
    初始状态 --> 连接成功 : 调用 connect()
    连接成功 --> 查询执行中 : 调用 execute_query()
    连接成功 --> 更新执行中 : 调用 execute_update()
    查询执行中 --> 连接关闭 : 调用 close()
    更新执行中 --> 连接关闭 : 调用 close()
    连接关闭 --> [*]

5. 序列图

为了更直观地展示数据库操作的流程，我们使用序列图展示 Database 类的操作过程：

sequenceDiagram
    participant User
    participant Database

    User->>Database: connect()
    Database-->>User: 连接成功

    User->>Database: execute_query("SELECT * FROM your_table WHERE id=1")
    Database-->>User: 返回查询结果

    User->>Database: execute_update("UPDATE your_table SET name='new_name' WHERE id=1")
    Database-->>User: 数据更新成功

    User->>Database: close()
    Database-->>User: 连接已关闭