python 对数据库的方法封装

一、做接口测试的时候，很多时候都需要查询数据库来验证结果

以下是对数据库操作的方法封装

import pymysql
import json

"""
通用封装查询
param sql:
param args:
return:返回查询结果 （（），（））
 
"""


class MySql(object):

    # 链接数据库
    def conn(self):

        # 创建新链接
        self.conn = pymysql.connect(host='www.baidu.com',
                                    user='user', 
                                    password='password',
                                    database='database', 
                                    charset='utf8')
        # 获取 cursor 对象并以自定义格式为字典
        self.cursor = self.conn.cursor(cursor=pymysql.cursors.DictCursor)
        # cursor = conn.cursor()  # 执行完毕返回的结果集默认以元组显示
        return self.conn, self.cursor


    # 关闭Cursor对象
    def close_conn(self, conn, cursor):
        cursor.close()
        conn.close()

    def select(self, sql):
        conn, cursor = MySql().conn()
        cursor.execute(sql)
        # 读取，执行 sql 后的所有数据
        cds = cursor.fetchall()
        # json 格式化读取到的数据
        rest = json.dumps(cds, default=str, ensure_ascii=False)
        self.close_conn(conn, cursor)
        return rest