python+selenium九：ddt数据驱动

python+selenium九：ddt数据驱动

 

读取excel内容

import xlrd
class ExcelUtil():

    def __init__(self, excelPath, sheetName):
    #def __init__(self, excelPath, sheetName):
        self.data = xlrd.open_workbook(excelPath)
        self.table = self.data.sheet_by_name(sheetName)
        # 获取第一行作为key值
        self.keys = self.table.row_values(0)
        # 获取总行数
        self.rowNum = self.table.nrows
        # 获取总列数
        self.colNum = self.table.ncols

    def dict_data(self):
        if self.rowNum <= 1:
            print("总行数小于1")
        else:
            r = []
            j=1
            for i in range(self.rowNum-1):
                s = {}
                # 从第二行取对应values值
                values = self.table.row_values(j)
                for x in range(self.colNum):
                    s[self.keys[x]] = values[x]
                r.append(s)
                j+=1
            return r

if __name__ == "__main__":
    # filepath = "D:\test\web-project\5ke\testdata.xlsx"
    filepath = "E:\testdata.xlsx"
    sheetName = "登录"
    data = ExcelUtil(filepath, sheetName)
    print(data.dict_data())