获取Excel用例数据的封装

超、凢脫俗 2022-09-03 08:26 109阅读 0赞

import xlrd
    # xlrd——.xls格式的excel文件；openpyxl——.xlxs格式的excel文件；pandas——大数据csv格式
    def get_excel_data(excelDir,sheetName,caseName,*args):
        resList=[]   # 存放Excel读取结果
        #1-把Excel加载到内存中---open-----formatting_info=True 保持原样式
        workBook=xlrd.open_workbook(excelDir,formatting_info=True)
        
        #2-获取对应的sheet
        workSheet=workBook.sheet_by_name(sheetName)
        
        #3-根据输入的列名，获取对应的列下标
    	colidx=[]#存放下标的列表
        for i in args:    # 遍历元祖元素
            # 列名在第一行数据里
            colidx.append(workSheet.row_values(0).index(i))
            
        #4-遍历第0列，根据caseName模糊匹配
        idx=0 #遍历变量
        for one in workSheet.col_values(0): #列表
            if caseName in one: #判断
                getcoldata=[]   #存放对应列的数据
                for num in colidx:      # 遍历列下标
                    tmp = workSheet.cell(idx,num).value
                    if is_json(tmp):  # tmp就是json，需要转换成字典
                        tmp = json.loads(tmp)
                    getcoldata.append(tmp)      # 把这一行列数据都存放起来
                resList.append(getcoldata)      # 每一行的列数据都存放起来
            idx += 1
        return resList
    
    def is_json(inStr):
        try:
            json.loads(inStr)
        except:
            return False  # 不是json
        return True   # 是json
    
    
    if __name__ == '__main__':
        res=get_excel_data('../data/testcase.xls','登录模块','Login','标题','URL','请求参数','响应预期结果')
        for one in res:
            print(one)