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 文件讀取與封裝的資料請關注好吧啦網其它相關文章!
相關文章: