使用 Python 操作 MySQL 数据库的实用工具类:MySQLHandler

操作数据库是非常常见的需求,使用 Python 和 pymysql 库封装一个通用的 MySQL 数据库操作工具类,并通过示例演示如何使用这个工具类高效地管理数据库。

工具类的核心代码解析

MySQLHandler 类简介

MySQLHandler 是一个 Python 类,用于简化对 MySQL 数据库的常见操作,如连接、查询、插入、更新和删除等。以下是该类的主要功能:

连接数据库

使用 pymysql.connect 方法建立与数据库的连接。

执行查询

提供 execute_query 方法执行 SELECT 查询,返回查询结果。

数据插入

使用 insert_data 方法将数据插入指定的表。

数据更新与删除

提供 update_data 和 delete_data 方法分别用于更新和删除数据。

代码详解

以下是 mysql_utils.py 的完整代码:

python 复制代码
import pymysql

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

    def connect(self):
        """建立数据库连接"""
        try:
            self.connection = pymysql.connect(
                host=self.host,
                user=self.user,
                password=self.password,
                database=self.database,
                port=self.port,
                charset='utf8mb4'
            )
        except Exception as e:
            print(f"连接数据库失败: {e}")
            raise

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

    def execute_query(self, query, params=None):
        """执行查询语句"""
        try:
            with self.connection.cursor() as cursor:
                cursor.execute(query, params)
                result = cursor.fetchall()
                return result
        except Exception as e:
            print(f"查询失败: {e}")
            raise

    def execute_update(self, query, params=None):
        """执行更新语句"""
        try:
            with self.connection.cursor() as cursor:
                rows_affected = cursor.execute(query, params)
                self.connection.commit()
                return rows_affected
        except Exception as e:
            print(f"更新失败: {e}")
            self.connection.rollback()
            raise

    def insert_data(self, table, data):
        """插入数据"""
        try:
            keys = ', '.join(data.keys())
            values = ', '.join(['%s'] * len(data))
            query = f"INSERT INTO {table} ({keys}) VALUES ({values})"
            self.execute_update(query, list(data.values()))
        except Exception as e:
            print(f"插入数据失败: {e}")
            raise

    def delete_data(self, table, condition, params):
        """删除数据"""
        try:
            query = f"DELETE FROM {table} WHERE {condition}"
            self.execute_update(query, params)
        except Exception as e:
            print(f"删除数据失败: {e}")
            raise

    def update_data(self, table, updates, condition, params):
        """更新数据"""
        try:
            update_str = ', '.join([f"{key}=%s" for key in updates.keys()])
            query = f"UPDATE {table} SET {update_str} WHERE {condition}"
            self.execute_update(query, list(updates.values()) + params)
        except Exception as e:
            print(f"更新数据失败: {e}")
            raise

使用示例

main.py 文件提供了对 MySQLHandler 的使用示例,包含常见的数据库操作流程。

python 复制代码
from mysql_utils import MySQLHandler

def main():
    # 创建 MySQLHandler 实例
    db = MySQLHandler(host="localhost", user="root", password="root", database="testdb")
    
    try:
        # 连接数据库
        db.connect()
        
        # 执行查询示例
        result = db.execute_query("SELECT * FROM users")
        print("查询结果:", result)
        
        # 插入数据示例
        db.insert_data("users", {"name": "Alice", "age": 25})
        print("数据插入成功")
        
        # 更新数据示例
        db.update_data("users", {"age": 26}, "name=%s", ["Alice"])
        print("数据更新成功")
        
        # 删除数据示例
        db.delete_data("users", "name=%s", ["Alice"])
        print("数据删除成功")
    except Exception as e:
        print(f"发生错误: {e}")
    finally:
        # 关闭数据库连接
        db.close()

if __name__ == "__main__":
    main()

MySQLHandler 提供了一个简单、灵活的方式来管理 MySQL 数据库操作。它封装了连接、查询、插入、更新和删除的逻辑,可以减少重复代码,提高开发效率。在实际使用中,还可以根据项目需求扩展该类的功能,例如增加事务管理或日志记录。

相关推荐
Code季风5 分钟前
将 gRPC 服务注册到 Consul:从配置到服务发现的完整实践(上)
数据库·微服务·go·json·服务发现·consul
Boilermaker199213 分钟前
【Java EE】SpringIoC
前端·数据库·spring
霸王龙的小胳膊27 分钟前
泛微虚拟视图-数据虚拟化集成
数据库
电商数据girl29 分钟前
有哪些常用的自动化工具可以帮助处理电商API接口返回的异常数据?【知识分享】
大数据·分布式·爬虫·python·系统架构
CoooLuckly32 分钟前
numpy数据分析知识总结
python·numpy
灵犀学长1 小时前
解锁Spring Boot多项目共享Redis:优雅Key命名结构指南
数据库·redis
超龄超能程序猿1 小时前
(六)PS识别:源数据分析- 挖掘图像的 “元语言”技术实现
python·组合模式
轩情吖1 小时前
Qt的信号与槽(二)
数据库·c++·qt·信号·connect·信号槽·
ZeroNews内网穿透1 小时前
服装零售企业跨区域运营难题破解方案
java·大数据·运维·服务器·数据库·tcp/ip·零售
可观测性用观测云1 小时前
达梦数据库监控观测最佳实践
数据库