appium+Python+Selenium+Unittest+HTMLTestRunner(五)读取Excel文件测试数据

readExcel.py

# coding:utf-8
'''
Created on 2024-8-30
description:读取Excel
@author: xxx
'''
import os, sys
from openpyxl import load_workbook

class TestExcel(object):
    #Excel存放地址
    path = (os.path.abspath('.') + '/data/data.xlsx').replace('\\', '/')

    def __init__(self):
        # 打开用例Excel
        self.file = load_workbook(self.path)

    def get_sheetnames(self):
        """
        返回所有sheet名称
        """
        return self.file.sheetnames

    def get_xlsx(self, sheet):
        """
        获取Excel中测试用例相关信息
        """
        list = []  # 定义一个空列表
        workbook = load_workbook(self.path)  # 打开表
        sheet = workbook[sheet]  # 定位表单

        row_value1 = [cell.value for cell in sheet[1]]  # 获取第1行的标题
        rows = sheet.max_row # 获取当前sheet行数
        rows=rows+1
        cols = sheet.max_column  # 获取当前sheet列数
        for i in range(1, rows):  # 从第2行遍历当前sheet
            row = [cell.value for cell in sheet[i]]  # 获取行数据
            dict = {}  # 创建空字典
            for j in range(0, cols):  # 遍历sheet列，组成字典
                if row_value1[j] == 'NO.' or row_value1[j] == 'code':
                    dict[row_value1[j]] = int(row[j])  # NO和code值取int
                else:
                    dict[row_value1[j]] = row[j]  # 从第一列开始，将每一列的数据与第1行的数据组成一个键值对，形成字典
            list.append(dict)  # 将字典添加list中
        return list  # 返回列表