forked from DxvLwRYF/apiAutoTest
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
#!/usr/bin/env/python3
|
||
# -*- coding:utf-8 -*-
|
||
"""
|
||
@project: apiAutoTest
|
||
@author: zy7y
|
||
@file: read_file.py
|
||
@ide: PyCharm
|
||
@time: 2020/7/31
|
||
@desc： 更新时间 2020/11/21 15：34 后续所有关于文件读取的方法全部收纳与此
|
||
"""
|
||
import yaml
|
||
import xlrd
|
||
from tools import extractor
|
||
|
||
|
||
class ReadFile:
|
||
config_dict = None
|
||
|
||
@classmethod
|
||
def get_config_dict(cls, config_path: str = 'config/config.yaml') -> dict:
|
||
"""读取配置文件，并且转换成字典
|
||
:param config_path: 配置文件地址， 默认使用当前项目目录下的config/config.yaml
|
||
return cls.config_dict
|
||
"""
|
||
if cls.config_dict is None:
|
||
# 指定编码格式解决，win下跑代码抛出错误
|
||
with open(config_path, 'r', encoding='utf-8') as file:
|
||
cls.config_dict = yaml.load(
|
||
file.read(), Loader=yaml.FullLoader)
|
||
return cls.config_dict
|
||
|
||
@classmethod
|
||
def read_config(cls, expr: str = '.') -> dict:
|
||
"""默认读取config目录下的config.yaml配置文件，根据传递的expr jsonpath表达式可任意返回任何配置项
|
||
:param expr: 提取表达式, 使用jsonpath语法,默认值提取整个读取的对象
|
||
return 根据表达式返回的值
|
||
"""
|
||
return extractor(cls.get_config_dict(), expr)
|
||
|
||
@classmethod
|
||
def read_testcase(cls):
|
||
"""
|
||
读取excel格式的测试用例
|
||
:return: data_list - pytest参数化可用的数据
|
||
"""
|
||
data_list = []
|
||
book = xlrd.open_workbook(cls.read_config('$.file_path.test_case'))
|
||
# 读取第一个sheet页
|
||
table = book.sheet_by_index(0)
|
||
for norw in range(1, table.nrows):
|
||
# 每行第4列 是否运行
|
||
if table.cell_value(norw, 4) != '否': # 每行第4列等于否将不读取内容
|
||
value = table.row_values(norw)
|
||
value.pop(4)
|
||
data_list.append(list(value))
|
||
return data_list
|