基于之前文章：
1、环境的搭建：/Mojitoice/article/details/108992622?spm=1001..3001.5501
2、配置文件的创建，封装配置文件的读取方法：/Mojitoice/article/details/109031074
3、excel读取方法的封装：/Mojitoice/article/details/109203605?spm=1001..3001.5501
4、url拼接方法的封装：/Mojitoice/article/details/114400695
5、封装get，post请求方法：/Mojitoice/article/details/114402767
6、进行mock的方法封装：/Mojitoice/article/details/114403473
背景：
我们搞接口自动化测试框架，对数据库的操作必不可少，比如对比接口返回的数据是否正确，再某些请求参数需要从数据库中取，所以我们需要封装一个对数据库操作的方法。我这里只做了简单的封装。
now，献上代码：/Users/dongyue/Documents/framework/common/configDB.py
import pymysqlfrom testFile.readConfig import ReadConfigfrom common.log import loggerclass 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 curdef 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_wpassword = 0Oe*I*****iFIUt02CBhost = 10.225.76.*port = 3306
数据库的操作，不要忘记try一下哈。其中的log.py。/Users/dongyue/Documents/framework/common/log.py
#作用：生成log方法，写入/result/logfile下import osimport loggingfrom logging.handlers import TimedRotatingFileHandlerfrom getPath import GetPathpath = 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.loggerlogger = Logger().get_logger()
接口自动化测试框架搭建(7 连接数据库提交sql方法以及打印日志方法封装)--python+HTMLTestRunnerCN+request+unittest+mock+db
