封装python 操作mysql 工具类

原创

mob64ca12d3dbd9 2024-12-25 03:48:39 ©著作权

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

封装Python操作MySQL工具类

在这篇文章中，我将指导你如何实现一个封装Python操作MySQL的工具类。这个工具类的主要目的是简化与MySQL数据库的交互，提供常用的增删改查（CRUD）功能。我们将按照以下步骤进行：

步骤	描述
1	安装必要的库
2	创建工具类结构
3	实现数据库连接
4	实现增删改查方法
5	进行测试

1. 安装必要的库

首先，我们需要安装mysql-connector-python库，这是一个用于Python与MySQL连接的库。可以使用以下命令安装：

pip install mysql-connector-python

2. 创建工具类结构

接下来，我们创建一个名为MySQLHelper的工具类。这个类将封装与数据库的所有交互。

class MySQLHelper:
    def __init__(self, host, user, password, database):
        """
        初始化数据库连接信息
        :param host: 数据库主机地址
        :param user: 数据库用户名
        :param password: 数据库密码
        :param database: 数据库名称
        """
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.connection = None

3. 实现数据库连接

在工具类中，我们需要实现一个方法来连接数据库。

    def connect(self):
        """连接到MySQL数据库"""
        import mysql.connector
        self.connection = mysql.connector.connect(
            host=self.host,
            user=self.user,
            password=self.password,
            database=self.database
        )
        print("数据库连接成功！")

4. 实现增删改查方法

增、删、改、查是与数据库交互的基本操作，我们将为每个操作实现对应的方法。

增加数据

    def insert(self, query, data):
        """
        插入数据到数据库
        :param query: SQL插入语句
        :param data: 数据元组
        """
        cursor = self.connection.cursor()
        cursor.execute(query, data)
        self.connection.commit()
        cursor.close()
        print("数据插入成功！")

查询数据

    def fetch(self, query, data=None):
        """
        查询数据库
        :param query: SQL查询语句
        :param data: 查询参数
        :return: 查询结果
        """
        cursor = self.connection.cursor()
        cursor.execute(query, data)
        result = cursor.fetchall()
        cursor.close()
        return result

更新数据

    def update(self, query, data):
        """
        更新数据库中的数据
        :param query: SQL更新语句
        :param data: 更新数据
        """
        cursor = self.connection.cursor()
        cursor.execute(query, data)
        self.connection.commit()
        cursor.close()
        print("数据更新成功！")

删除数据

    def delete(self, query, data):
        """
        删除数据库中的数据
        :param query: SQL删除语句
        :param data: 删除条件
        """
        cursor = self.connection.cursor()
        cursor.execute(query, data)
        self.connection.commit()
        cursor.close()
        print("数据删除成功！")

5. 进行测试

最后，我们可以测试这个工具类。

if __name__ == "__main__":
    db = MySQLHelper(host='localhost', user='root', password='yourpassword', database='yourdatabase')
    db.connect()
    
    insert_query = "INSERT INTO users (name, age) VALUES (%s, %s)"
    db.insert(insert_query, ("Alice", 30))
    
    select_query = "SELECT * FROM users"
    results = db.fetch(select_query)
    print(results)
    
    update_query = "UPDATE users SET age=%s WHERE name=%s"
    db.update(update_query, (31, "Alice"))
    
    delete_query = "DELETE FROM users WHERE name=%s"
    db.delete(delete_query, ("Alice",))

classDiagram
    class MySQLHelper {
        +__init__(host, user, password, database)
        +connect()
        +insert(query, data)
        +fetch(query, data)
        +update(query, data)
        +delete(query, data)
    }