基于面向对象封装pymysql工具类

在工作中,手写SQL语句基于execute执行,这种方式,性能是最好的!但是往往开发中,如果我们编写大量固定的SQL语句,也会导致程序的维护成本提升,同时将来如果应用程序要切换底层数据库的话,将会带来巨大的兼容问题。

import pymysql.cursors


class DB(object):
    """数据库工具类"""
    connected = False  # 数据库连接状态,False表示没有连接或连接失败
    __conn = None  # 数据库连接对象

    class DBError(Exception):
        """数据库异常基类"""
        pass

    class DBConfigError(DBError):
        """数据库配置异常类"""
        pass

    class ConnectionError(DBError):
        """数据库连接异常类"""
        pass

    class ExecuteError(DBError):
        """数据库执行异常类"""
        pass

    def __init__(self, conf):
        if type(conf) is not dict:
            raise self.DBConfigError("错误: 数据库连接参数必须是字典类型!")

        for key in ["host", "user", "password", "database"]:
            if key not in conf.keys():
                raise self.DBConfigError(f"错误: 数据库连接缺少'{key}'参数")

        if not conf.get("port", None):
            conf["port"] = 3306

        if 'charset' not in conf.keys():
            conf["charset"] = "utf8mb4"

        try:
            self.__conn = pymysql.connect(
                host=conf['host'],
                port=conf['port'],
                user=conf['user'],
                password=conf['password'],
                database=conf['database'],
                charset=conf['charset'],
                cursorclass=pymysql.cursors.DictCursor
            )
            self.connected = True
        except pymysql.Error as e:
            raise self.ConnectionError(f"数据库连接失败: {e}")
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您好!关于 pymysql 工具类,可以提供一些基本的信息。pymysql 是一个用于 Python 的第三方库,用于连接和操作 MySQL 数据库。它提供了一组简单而强大的 API,使得在 Python 中进行数据库操作变得更加方便和高效。 在使用 pymysql 进行数据库操作时,可以创建一个工具类封装常用的功能,以便在项目中重复使用。下面是一个简单的示例: ```python import pymysql class MySQLUtil: def __init__(self, host, user, password, database): self.connection = pymysql.connect(host=host, user=user, password=password, database=database) self.cursor = self.connection.cursor() def execute_query(self, query): self.cursor.execute(query) result = self.cursor.fetchall() return result def execute_update(self, query): self.cursor.execute(query) self.connection.commit() def close(self): self.cursor.close() self.connection.close() ``` 以上示例中,MySQLUtil 类包含了连接数据库、执行查询和更新操作、关闭连接等方法。您可以根据实际需要进行扩展和修改。 使用该工具类时,可以先实例化一个 MySQLUtil 对象,并传入数据库的相关参数(主机名、用户名、密码、数据库名)。然后就可以调用其中的方法来执行数据库操作,例如执行查询语句 `execute_query` 或执行更新语句 `execute_update`。 注意,为了安全起见,建议将数据库的敏感信息存储在配置文件中,并在创建 MySQLUtil 对象时从配置文件中读取。 希望以上信息对您有所帮助!如果还有其他问题,请随时提问。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值