python excel和yaml文件的读取封装

import osimport xlrdPATH = lambda p: os.path.abspath( os.path.join(os.path.dirname(__file__), p))class ExcelData: def __init__(self, file, sheet='sheet1', title=True): # 判断文件存在不存在 if os.path.isfile(PATH(file)): self.file = PATH(file) self.sheet = sheet self.title = title self.data = list() self.workbook = xlrd.open_workbook(self.file) else: raise FileNotFoundError('文件不存在') @property def get_data(self): '''获取表格数据''' if not self.data: # 判断表单名称 if type(self.sheet) not in [int, str]:raise Exception('表单名称类型错误') else:if type(self.sheet) == int: book = self.workbook.sheet_by_index(self.sheet)else: book = self.workbook.sheet_by_name(self.sheet) # 判断表格是否有表头，有则输出列表嵌套字典形式数据，否则输入列表嵌套列表形式数据 if self.title:title = book.row_values(0)for i in range(1, book.nrows): self.data.append(dict(zip(title, book.row_values(i)))) # 可参考字典章节 else:for i in range(book.nrows): self.data.append(book.row_values(i)) return self.data @property def get_sheets(self): '''获取所有表单，这个在后续会用到''' book = self.workbook.sheets() return book

调用操作

infos = ExcelData('htmls/测试用例.xlsx', '登入页面', True).get_dataprint(infos)sheets = ExcelData('htmls/测试用例.xlsx').get_sheetsprint(sheets)

import osimport yamlfrom yamlinclude import YamlIncludeConstructorYamlIncludeConstructor.add_to_loader_class(loader_class=yaml.FullLoader) # 用于yaml文件嵌套PATH = lambda p: os.path.abspath(os.path.join( os.path.dirname(__file__), p))class YamlData: def __init__(self, file): if os.path.isfile(PATH(file)): self.file = PATH(file) else: raise FileNotFoundError('文件不存在') @property # 设置属性，调用data方法时可通过调用属性，不需要带括号 def data(self): with open(file=self.file, mode='rb') as f: infos = yaml.load(f, Loader=yaml.FullLoader) # infos = yaml.load(f) return infos

调用操作

infos = YamlData('htmls/loginsucess.yaml').dataprint(infos)'D:Program FilesPythonPython37-32python.exe' D:/demo/yamldata.py{’id’: ’login_001’, ’module’: ’登入页面’, ’title’: ’登入时账号为空’, ’message’: ’已打开链接’, ’testcase’: [{’element_info’: ’css->[placeholder='请输入账号']’, ’operate_type’: ’send_keys’, ’keys’: ’SSSS’, ’info’: ’点击账号输入框，输入账号’}, {’element_info’: ’css->[placeholder='请输入密码']’, ’operate_type’: ’send_keys’, ’keys’: ’XXX’, ’info’: ’点击密码输入框，输入密码’}, {’element_info’: ’div->'登 录'’, ’operate_type’: ’click’, ’info’: ’点击登入菜单’}, {’operate_type’: ’is_sleep’, ’keys’: 3, ’info’: ’等待进入’}], ’check’: None}Process finished with exit code 0

以上就是python excel和yaml文件的读取与封装的详细内容，更多关于python 文件读取与封装的资料请关注好吧啦网其它相关文章！