接口自动化测试框架搭建(7、连接数据库提交sql方法以及打印日志方法封装)--python+HTMLTestRunnerCN+request+unittest+mock+db

基于之前文章：

1、环境的搭建：https://blog.csdn.net/Mojitoice/article/details/108992622?spm=1001.2014.3001.5501

2、配置文件的创建，封装配置文件的读取方法：https://blog.csdn.net/Mojitoice/article/details/109031074

3、excel读取方法的封装：https://blog.csdn.net/Mojitoice/article/details/109203605?spm=1001.2014.3001.5501

4、url拼接方法的封装：https://blog.csdn.net/Mojitoice/article/details/114400695

5、封装get，post请求方法：https://blog.csdn.net/Mojitoice/article/details/114402767

6、进行mock的方法封装：https://blog.csdn.net/Mojitoice/article/details/114403473

 

背景：

我们搞接口自动化测试框架，对数据库的操作必不可少，比如对比接口返回的数据是否正确，再某些请求参数需要从数据库中取，所以我们需要封装一个对数据库操作的方法。我这里只做了简单的封装。

now，献上代码：/Users/dongyue/Documents/framework/common/configDB.py

import pymysql
from testFile.readConfig import ReadConfig
from common.log import logger



class GetMysql():


    def __init__(self,dbname):
        #读取配置文件中需要的数据库(不同的需求需要链接不同的数据库，所以需要有数据库名字的入参)
        self.username = ReadConfig().get_db(dbname,"username")
        self.password = ReadConfig().get_db(dbname,"password")
        self.host = ReadConfig().get_db(dbname,"host")
        self.port = ReadConfig().get_db(dbname,"port")

    def connectdb(self):
        #连接数据库，创建游标对象
        con = pymysql.connect(host=self.host,
                               port=int(self.port),
                               user=self.username,
                               password=self.password,
                               charset="utf8",
                               cursorclass=pymysql.cursors.DictCursor
                               )
        cur = con.cursor()
        return cur

    def getone(self,sql):
        #提交sql语句。获取第一条返回结果
        try:
            cur = self.connectdb()
            cur.execute(sql)
            return cur.fetchone()
        except pymysql.Error as e:
            logger.debug("sql提交失败",e)

    def getall(self,sql):
        # 提交sql语句。获取全部返回结果，以列表形式返回
        try:
            cur = self.connectdb()
            cur.execute(sql)
            return cur.fetchall()
        except pymysql.Error as e:
            logger.debug("sql提交失败",e)



if __name__ == '__main__':
    print("ok")

在代码中读取了配置文件中数据库的配置：

[ecqc_DB]
username = ecqc_w
password = 0Oe*I*****iFIUt02CB
host = 10.225.76.*
port = 3306

数据库的操作，不要忘记try一下哈。其中的log.py。/Users/dongyue/Documents/framework/common/log.py

#作用：生成log方法，写入/result/logfile下
import os
import logging
from logging.handlers import TimedRotatingFileHandler
from getPath import GetPath

path = GetPath().getPath()
log_path = os.path.join(path, 'result/logfile')  # 存放log文件的路径


class Logger(object):
    def __init__(self, logger_name='logs…'):
        self.logger = logging.getLogger(logger_name)
        logging.root.setLevel(logging.NOTSET)
        self.log_file_name = 'logs'  # 日志文件的名称
        self.backup_count = 5  # 最多存放日志的数量
        # 日志输出级别
        self.console_output_level = 'WARNING'
        self.file_output_level = 'DEBUG'
        # 日志输出格式
        self.formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

    def get_logger(self):
        """在logger中添加日志句柄并返回，如果logger已有句柄，则直接返回"""
        if not self.logger.handlers:  # 避免重复日志
            console_handler = logging.StreamHandler()
            console_handler.setFormatter(self.formatter)
            console_handler.setLevel(self.console_output_level)
            self.logger.addHandler(console_handler)

            # 每天重新创建一个日志文件，最多保留backup_count份
            file_handler = TimedRotatingFileHandler(filename=os.path.join(log_path, self.log_file_name), when='D',
                                                    interval=1, backupCount=self.backup_count, delay=True,
                                                    encoding='utf-8')
            file_handler.setFormatter(self.formatter)
            file_handler.setLevel(self.file_output_level)
            self.logger.addHandler(file_handler)
        return self.logger


logger = Logger().get_logger()