python 数据库mysql查询工具类

1.工具类代码

#新建 tz_mysql.py
import pymysql
from pymysql.cursors import DictCursor


class HandleSql:

    def __init__(self, db_config):
        # 创建数据库连接
        # 根据不同的数据库，创建不同的链接
        # 如果engine没有值默认返回mysql
        engine = db_config.pop('engine', 'mysql')
        if engine.lower() == 'mysql':
            self.conn = pymysql.connect(**db_config)
        else:
            pass  # 此处可以配置其他的数据库

    def get_fetchone(self, sql, res_type='t'):
        """
        返回一条数据
        :param sql:
        :param size: 默认返回元组
        :return:
        """
        # 创建游标
        # 返回元组类型的数据
        if res_type == 't':
            with self.conn.cursor() as cursor:
                cursor.execute(sql)
                return cursor.fetchone()
        # 返回字典类型的数据
        else:
            with self.conn.cursor(DictCursor) as cursor:
                cursor.execute(sql)
                return cursor.fetchone()

    # 获取多个数据
    def get_fetchmany(self, sql, size, res_type='t'):
        if res_type == 't':
            with self.conn.cursor() as cursor:
                cursor.execute(sql)
                return cursor.fetchmany(size)
        # 返回字典类型的数据
        else:
            with self.conn.cursor(DictCursor) as cursor:
                cursor.execute(sql)
                return cursor.fetchmany(size)

    # 获取全部数据
    def get_fetchall(self, sql, res_type='t'):
        if res_type == 't':
            with self.conn.cursor() as cursor:
                cursor.execute(sql)
                return cursor.fetchall()
        # 返回字典类型的数据
        else:
            with self.conn.cursor(DictCursor) as cursor:
                cursor.execute(sql)
                return cursor.fetchall()

    def exist_sql(self, sql):
        if self.get_fetchone(sql):
            return True
        else:
            return False

    def __del__(self):
        self.conn.close()

2.调用方

from xxxx.tz_mysql import HandleSql

# 配置文件中db_config
db_config = {
    'host': '192.168.1.xx',  # 主机
    'user': 'xx',  # 用户名
    'password': 'xxx',  # 密码
    'port': 3306,  # 端口 3306
    'database': 'xxx'  # 数据库名
}

HandleSql(db_config).get_fetchall(sql='select * from xxx',res_type='')