Python封装MySQL操作工具类

pip install mysql-connector-python

import mysql.connector
from mysql.connector import Error
from mysql.connector import pooling

class MySQLDatabase:
    def __init__(self, host, user, password, database, pool_size=5):
        """
        初始化 MySQL 数据库操作类
        :param host: 数据库地址
        :param user: 用户名
        :param password: 密码
        :param database: 数据库名
        :param pool_size: 连接池大小
        """
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.pool_size = pool_size

        # 初始化连接池
        self.pool = pooling.MySQLConnectionPool(
            pool_name="mysql_pool",
            pool_size=self.pool_size,
            host=self.host,
            user=self.user,
            password=self.password,
            database=self.database
        )

    def _get_connection(self):
        """获取连接池中的连接"""
        return self.pool.get_connection()

    def _execute_query(self, query, params=None, fetchone=False):
        """执行查询操作（查询、插入、更新等）"""
        connection = None
        cursor = None
        try:
            connection = self._get_connection()
            cursor = connection.cursor(dictionary=True)
            cursor.execute(query, params)
            connection.commit()  # 提交事务

            if fetchone:
                return cursor.fetchone()  # 获取单条记录
            return cursor.fetchall()  # 获取所有记录
        except Error as e:
            print(f"Error: {e}")
            return None
        finally:
            if cursor:
                cursor.close()
            if connection:
                connection.close()

    def query(self, query, params=None):
        """执行查询操作，返回查询结果"""
        return self._execute_query(query, params)

    def insert(self, query, params):
        """执行插入操作"""
        return self._execute_query(query, params)

    def update(self, query, params):
        """执行更新操作"""
        return self._execute_query(query, params)

    def delete(self, query, params):
        """执行删除操作"""
        return self._execute_query(query, params)

    def execute(self, query, params=None):
        """执行任意SQL语句"""
        return self._execute_query(query, params)

if __name__ == "__main__":
    # 创建MySQL数据库操作类的实例
    db = MySQLDatabase(host="localhost", user="root", password="password", database="testdb")

    # 执行查询操作
    query = "SELECT * FROM users WHERE age = %s"
    params = (25,)
    results = db.query(query, params)
    print("Query Result:", results)

    # 执行插入操作
    insert_query = "INSERT INTO users (name, age) VALUES (%s, %s)"
    insert_params = ("Alice", 30)
    db.insert(insert_query, insert_params)
    print("Inserted successfully!")

    # 执行更新操作
    update_query = "UPDATE users SET age = %s WHERE name = %s"
    update_params = (31, "Alice")
    db.update(update_query, update_params)
    print("Updated successfully!")

    # 执行删除操作
    delete_query = "DELETE FROM users WHERE name = %s"
    delete_params = ("Alice",)
    db.delete(delete_query, delete_params)
    print("Deleted successfully!")

def begin_transaction(self):
    connection = self._get_connection()
    connection.start_transaction()
    return connection

def commit_transaction(self, connection):
    connection.commit()

def rollback_transaction(self, connection):
    connection.rollback()