selenium自动化-数据驱动2

# -*- coding: utf-8 -*-
import xlrd

class ExcelUtil():

    #初始化
    def __init__(self,excelpath,sheetname):
        #先初始化打开数据文件
        self.data = xlrd.open_workbook(excelpath)
        #再通过sheetname读取数据文件内容
        self.table = self.data.sheet_by_name(sheetname)

        # 从第一行去对应的key值
        self.keys = self.table.row_values(0)

        #获取所有的行数
        self.allrow = self.table.nrows
        #获取所有列数
        self.allcol = self.table.ncols


    def dict_data(self):
        if self.allrow <= 1:
            print("总行数小于1")

        else:
            r = []
            j = 1
            #加一个while循环判断，如果数据文件在遍历时超出range定义的边界，会报错，在测试登陆时或者其他需求登陆用户的地方，一般在数据文件中设置几个用户就可以了
            while j < 3:
                for i in range(self.allrow-1):
                    s = {}
                    #从第二行去对应的values值
                    values = self.table.row_values(j)
                    for x in range(self.allcol):
                        s[self.keys[x]] = values[x]#遍历出一个{k:v}的键值对
                    j+=1

                    r.append(s)
                return r

if __name__ == "__main__":
    filepath = "user.xlsx"#你的数据文件所在的路径，由于我的数据文件和我的测试脚本在同一个目录下，所以不需要加前面的路径
    sheetname = "userinfo"
    data = ExcelUtil(filepath,sheetname)

    print(data.dict_data())

输出结果：

C:UsersJaysonPycharmProjects estvenvScriptspython.exe D:/tes1t/test/data/testdata.py
[{'username': 'zzz', 'password': 123456.0}, {'username': 'aaa', 'password': 234567.0}, {'username': 'qqq', 'password': 345678.0}]

Process finished with exit code 0

这个的userinfo是我提前在数据文件中报错的数据