基础类封装-pymysql库操作mysql封装

import pymysql
from lib.logger import logger
from warnings import filterwarnings
filterwarnings("ignore",category=pymysql.Warning) # 忽略mysql警告信息

class MysqlDb():

    logger = logger

    def __init__(self, host, port, user, passwd):
        # 建立数据库连接
        self.conn = pymysql.connect(
            host=host,
            port=port,
            user=user,
            passwd=passwd,
            charset='utf8'
        )
        # 通过 cursor() 创建游标对象，并让查询结果以字典格式输出
        self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

    def __del__(self): # 对象资源被释放时触发，在对象即将被删除时的最后操作
        # 关闭游标
        self.cur.close()
        # 关闭数据库连接
        self.conn.close()

    def select_db(self,db_name):
        """
        选择数据库
        :return:
        """
        try:
            self.conn.select_db(db_name)
            self.logger.logger.info("选择 {0} 执行成功".format(db_name))
        except Exception as e:
            self.logger.logger.exception("操作出现错误：{0}".format(e))

    def query_db(self, sql,state="all"):
        """
        查询
        :param sql:sql语句
        :param state:all查询所有，outhor查询单条
        :return:
        """
        # 使用 execute() 执行sql
        self.logger.logger.info(sql)
        self.cur.execute(sql)
        if state == "all":
            # 使用 fetchall() 获取查询结果
            data = self.cur.fetchall()
        else:
            data = self.cur.fetchone()
        return data

    def execute_db(self, sql):
        """更新/插入/删除"""
        try:
            # 使用 execute() 执行sql
            self.cur.execute(sql)
            # 提交事务
            self.conn.commit()
            self.logger.logger.info("{0} 执行成功".format(sql))
        except Exception as e:
            self.logger.logger.exception("操作出现错误：{}".format(e))
            # 回滚所有更改
            self.conn.rollback()

if __name__ == '__main__':
    pass