Python接口自动化核心模块 - 数据库操作和日志
每天进步一点点，关注我们哦，每天分享测试技术文章
本文章出自【码同学软件测试】
码同学公众号：自动化软件测试，领取资料可加：magetest
码同学抖音号：小码哥聊软件测试
进行接口测试时，我们需要连接到数据库中，对数据源进行备份、还原、验证等操作。
Python连接数据库常见模块
- MysqlDB
- python2时代最火的驱动库。基于C开发，对windows平台不友好。现在已经进入python3时代，基本不再使用
- MysqlClient
- mysqldb的衍生版本，完全兼容python3.它是重量级Web开发框架Django中ORM功能依赖工具
- Pymysql
- 纯Python实现的驱动，性能比MysqlDb差，但是安装简单，容易使用
- SQLAlchemy
- 即支持原生SQL也支持ORM的库
我们以pymysql为例
01
Pymysql使用方法
安装方法：pip install pymysql
Pymsyql使用流程
- 获取连接
- 获取游标 --游标的作用就是用于对查询数据库所返回的记录进行遍历，以便进行相应的操作
- 执行SQL语句
- 关闭游标
- 关闭连接
代码实现：
# 创建数据库连接对象connect = pymysql.Connect( host='xxxxx', port=3306, user='root', password='XXXX', charset='utf8mb4', cursorclass=pymysql.cursors.DictCursor)
connect：接收Pymysql.connect方法返回的对象,建立连接对象pymysql.Connect 建立连接方法host=XXXX 连接数据库服务器port=3306 连接数据库端口号user="root"：用户名password="xxxxx"：密码charset="utf8mb4"：建立连接的编码
cursorclass=pymysql.cursors.DictCursor:设置返回数据类型 返回字典
# 建立游标cursor = conn.cursor()# 执行核心SQL语句cursor.execute("select version();")# 关闭游标cursor.close()# 关闭连接conn.close()
查询订单库得到一个游标对象cursor = connect.cursor()查询库名加表名 mtxshop_trade.es_ordercursor.execute("SELECT order_id,trade_sn FROM mtxshop_trade.es_order WHERE order_id=47050 OR order_id=47049")data = cursor.fetchall() #得到查询的所有结果cursor.close()#关闭游标对象
免费领取码同学软件测试课程笔记+超多学习资料+完整视频+面试题，可加微信：magetest
02
日志的基本概念
提到日志，无论是写框架代码还是业务代码，都离不开日志的记录，他能给我们定位问题带来极大的帮助，最佳的做法是使用内置的logging模块， 因为 logging 模块给开发者提供了非常丰富的功能。
日志级别
日志级别是为了控制打印日志的信息程度的
首先，配置日志模块时，需要先设置日志模块的日志等级
例如，如果设置为INFO级别，那么打印日志用DEBUG就不会输出日志。
- DEBUG
- INFO
- WARNING
- ERROR
- CRITICAL
日志输出模式
- 输出到控制台
- 输出到文件
日志格式
指定输出的日志的格式和内容常见的格式有：
%(levelno)s: 打印日志级别的数值
%(levelname)s: 打印日志级别名称%(pathname)s: 打印当前执行程序的路径，其实就是sys.argv[0]%(filename)s: 打印当前执行程序名%(funcName)s: 打印日志的当前函数%(lineno)d: 打印日志的当前行号%(asctime)s: 打印日志的时间%(thread)d: 打印线程ID%(threadName)s: 打印线程名称%(process)d: 打印进程ID%(message)s: 打印日志信息logging使用流程
- 第一步：实例化logging模块
- 第二步：设置日志等级
- 第三步：配置日志处理器、日志格式；日志处理器：控制日志的打印模式
- 第四步：打印日志
01
logging模块处理流程
接口测试框架实现日志收集功能
编写日志配置函数代码
这个函数配置了输出日志到控制台和文件，并且设置了日志打印格式
def logging_init():# 初始化日志器
logger = logging.getLogger()# 设置日志等级logger.setLevel(logging.INFO)# 添加控制器stream_handler = logging.StreamHandler()file_handler = logging.handlers.TimedRotatingFileHandler(config.BASE_DIR + "/logs/lagou_log.log", when='h',interval=1,backupCount=3, encoding="utf-8")# 设置日志格式fmt = "%(asctime)s %(levelname)s [%(name)s] [ %(filename)s %(funcName)s % (lineno)d ] %(message)s "formatter = logging.Formatter(fmt)# 将日志格式添加到控制器stream_handler.setFormatter(formatter)file_handler.setFormatter(formatter)# 将控制器添加到日志器logger.addHandler(stream_handler)logger.addHandler(file_handler)
return logger然后在api. init .py 中调用这个函数，完成日志的初始化
from utils import logging_init
# 初始化日志配置函数logging_init()
# 测试打印日志import logginglogging.info(" 测 试 info 级 别 的 日 志 打 印 ")logging.debug("测试debug级别的日志打印") #不会打印在api. init .py 初始化日志配置的原因：
execute执行script中的用例，script中的用例是调用api 的接口实现接口测试，按照模块语法，调用模块时，会自动执行模块下的 init .py 代码
后续，只需要在需要打印日志的模块，导入logging安装包，就可以输出我们配置好日志格式和日志等级的日志信息了
免费领取码同学软件测试课程笔记+超多学习资料+学习完整视频 ☞ 可加：magetest／关注码同学公众号：自动化软件测试
本文著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。
