python mysql 封装service类

原创

mob64ca12ddcacc 2024-11-20 08:32:29 ©著作权

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

Python MySQL 封装Service类

在现代应用程序中，数据库是数据存储和管理的关键部分。Python作为一种广泛使用的编程语言，提供了多种方式与MySQL数据库进行交互。为了提高代码的可维护性和复用性，我们可以通过封装Service类来简化数据库操作。本文将介绍如何实现一个基本的数据库Service类，并提供相关代码示例。

什么是Service类

Service类是一种封装逻辑的设计模式，通常用于处理与外部服务或资源的交互。在这个上下文中，我们将使用Service类来封装与MySQL数据库的交互。通过这种方式，我们可以把具体的数据库操作逻辑与应用程序的业务逻辑分离，提高代码的可读性。

环境准备

首先，确保您已经安装了MySQL数据库以及Python的mysql-connector-python库。如果还没有安装，可以通过以下命令进行安装：

pip install mysql-connector-python

Service类的实现

接下来，我们来实现一个简单的MySQL Service类。以下是一个基本的示例，包含了常见的CRUD（创建、读取、更新、删除）操作。

1. 创建数据库连接

我们首先需要创建一个用于连接MySQL数据库的类。以下是简单的连接配置：

import mysql.connector
from mysql.connector import Error

class Database:
    def __init__(self, host, user, password, database):
        self.connection = None
        try:
            self.connection = mysql.connector.connect(
                host=host,
                user=user,
                password=password,
                database=database,
            )
            print("连接到MySQL数据库成功")
        except Error as e:
            print(f"连接错误: {e}")

    def get_connection(self):
        return self.connection

    def close_connection(self):
        if self.connection:
            self.connection.close()
            print("数据库连接已关闭")

2. 创建Service类

通过创建Service类，我们可以在一个地方管理所有的数据库操作：

class UserService:
    def __init__(self, database):
        self.connection = database.get_connection()

    def create_user(self, username, email):
        cursor = self.connection.cursor(buffered=True)
        query = "INSERT INTO users (username, email) VALUES (%s, %s)"
        values = (username, email)
        cursor.execute(query, values)
        self.connection.commit()
        cursor.close()

    def get_user(self, user_id):
        cursor = self.connection.cursor(dictionary=True)
        query = "SELECT * FROM users WHERE id = %s"
        cursor.execute(query, (user_id,))
        user = cursor.fetchone()
        cursor.close()
        return user

    def update_user(self, user_id, username, email):
        cursor = self.connection.cursor()
        query = "UPDATE users SET username = %s, email = %s WHERE id = %s"
        values = (username, email, user_id)
        cursor.execute(query, values)
        self.connection.commit()
        cursor.close()

    def delete_user(self, user_id):
        cursor = self.connection.cursor()
        query = "DELETE FROM users WHERE id = %s"
        cursor.execute(query, (user_id,))
        self.connection.commit()
        cursor.close()

3. 使用Service类

现在，我们可以使用创建的UserService类来进行用户相关的数据库操作。下面是一个典型的用法示例：

if __name__ == "__main__":
    database = Database(host="localhost", user="root", password="password", database="test_db")
    user_service = UserService(database)

    # 创建一个新用户
    user_service.create_user("Alice", "alice@example.com")

    # 获取用户信息
    user = user_service.get_user(1)
    print(user)

    # 更新用户信息
    user_service.update_user(1, "Alice Smith", "alice.smith@example.com")

    # 删除用户
    user_service.delete_user(1)

    # 关闭连接
    database.close_connection()

4. 流程图

为了更好地理解用户操作的流程，我们可以使用Mermaid的Journey语法来展示这一过程：

journey
    title 用户管理的服务流程
    section 创建用户
      创建新用户: 5: 用户服务
    section 获取用户
      获取用户信息: 4: 用户服务
    section 更新用户
      更新用户信息: 4: 用户服务
    section 删除用户
      删除用户: 5: 用户服务