操作excel文件

安装方式:

pip install openpyxl
rom openpyxl import load_workbook
"""
核心功能:jmeter中的读取excel功能
"""
class HandleExcel:
    def __init__(self, file_name, sheet_name):
        """
        :param file_name:测试用例文件名称(绝对路径)
        :param sheet_name: excel中的sheet名称
        """
        self.file_name = file_name
        self.sheet_name = sheet_name
        # 加载excel
        self.wb_obj = load_workbook(filename=self.file_name)
    def get_excel_test_case(self):
        case_list = []
        # 获取表单对象
        sheet_obj = self.wb_obj[self.sheet_name]
        # 读取表单数据
        datas = list(sheet_obj.iter_rows(values_only=True))
        case_title = datas[0]  # 获取表头
        case_datas = datas[1:]  # 获取表单
        for i in case_datas:
            dict_list = dict(zip(case_title, i))
            case_list.append(dict_list)
        # 关闭表单
        self.cose_file()
        return case_list
        # 写excel
        # sheet_obj["B3"] = "test"
        # sheet_obj.cell(row=2, column=3, value="test_demo")
        # 保存excel
        # wb_obj.save(self.filename)
    def cose_file(self):
        # 关闭表单
        self.wb_obj.close()
if __name__ == '__main__':
    list_data = HandleExcel(r"E:\ningMengClass\py48\api\test_data\case_data.xlsx", "login").get_excel_test_case()
    print(list_data)

 

posted on 2020-03-05 18:57  诚实的表达自己  阅读(413)  评论(0编辑  收藏  举报