用pymysql封装连接mysql数据库的工具类【接口自动化框架设计系列】【多测师】

#coding=utf-8
"""
===========================
Author:多测师_王sir
Time:2020/5/20 17:24
Company:上海多测师信息有限公司
===========================
"""

""" 
查询
"""

import pymysql
from common.handleconfig import conf


class DB:
    def __init__(self):
        # 创建一个连接对象
        self.conn = pymysql.connect(host=conf.get("db", "host"),
                                    port=conf.getint("db", "port"),
                                    user=conf.get("db", "user"),
                                    password=conf.get("db", "pwd"),
                                    charset=conf.get("db", "charset"),
                                    cursorclass=pymysql.cursors.DictCursor
                                    )
        # 创建一个游标
        self.cur = self.conn.cursor()

    def find_one(self, sql):
        """获取查询出来的第一条数据"""
        # 执行查询语句
        self.conn.commit()
        self.cur.execute(sql)
        data = self.cur.fetchone()
        return data

    def find_all(self, sql):
        """获取查询出来的所有数据"""
        self.conn.commit()
        self.cur.execute(sql)
        data = self.cur.fetchall()
        return data

    def close(self):
        """关闭游标，断开连接"""
        self.cur.close()
        self.conn.close()