1、封装yaml文件
做用例参数化或一些通用的数据，会习惯性放在yaml文件中。
- 先自己创建并写好一个login_info.yaml文件，名称随意，然后封装读这个yaml文件的步骤，供程序中调用
image.png
封装yaml文件的源码~
思路：前提我知道yaml文件写在data下面，我需要打开login_info.yaml这个文件，怎么打开？先找到自己在哪儿，然后找到根目录，在根目录下去找data下的yaml文件。
import os
import yaml
class FileTools:
"""
封装yaml文件
"""
@classmethod
def get_interface_dir(cls):
"""
当前文件的获取
:return:
"""
# 获取当前文件的绝对路径，再获取这个文件所在的目录utils，再获取utils所在的目录po
return os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@classmethod
def read_yaml(cls, file_name):
"""
读取yaml
:return:
"""
# 获取当前文件的绝对路径，再获取这个文件所在的目录utils，再获取utils所在的目录po
_path = cls.get_interface_dir()
# 拼接yaml文件所在的绝对路径，sep相当于win的\,linux的/ sep.join需要的参数是一个列表
yaml_file = os.sep.join([_path, "data", file_name + ".yaml"])
# 打开yaml文件，并使用safe_load将内容return
with open(yaml_file, encoding="utf-8") as f:
return yaml.safe_load(f)
if __name__ == '__main__':
print(FileTools.read_yaml("login_info"))
print(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
- 然后
FileTools.read_yaml("login_info")调用login_info这个文件的数据
image.png
2、日志的封装
- 调试case时，需要打印关键日志供排查问题，我们可以封装日志，代码中直接调用
import logging
import os
from interface.po.utils.file_tools import FileTools
"""
封装日志
"""
# 初始化句柄
logger = logging.getLogger(__name__)
# 拼接日志文件所在路径
file_path = os.sep.join([FileTools.get_interface_dir(), "logs"])
# 判断文件路径是否存在，不存在则创建
if not os.path.exists(file_path):
os.mkdir(file_path)
# 拼接log文件和句柄
file_handle = logging.FileHandler(filename=file_path+"/po.log", encoding="utf-8")
# 日志的输出格式
formatter = logging.Formatter('[%(asctime).19s] %(process)d:%(levelname).1s %(filename)s:%(lineno)d:%(funcName)s: %(message)s]')
# 日志格式与句柄的绑定
file_handle.setFormatter(formatter)
# 初始化控制台句柄
stream_handler = logging.StreamHandler()
# 控制台日志格式与句柄绑定
stream_handler.setFormatter(formatter)
# 与logger绑定
logger.addHandler(file_handle)
logger.addHandler(stream_handler)
# 设置日志级别
logger.setLevel(logging.INFO)
-
使用，一般是在通用api中去调用封装的方法
image.png -
然后python在运行用例时，会看到我们封装的logs文件夹中打印出所需的日志
image.png
