python中数据和日志的封装

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