Python +Selenium 數據驅動 excel文件讀寫

在測試過程中我們經常會遇到需要大量測試數據進行支持的時候，如添加批量數據，多用戶登錄等。

爲此我們需要用excel文件來準備大量數據，這裏就涉及到了 excel文件讀取的讀取：我們直接貼代碼吧！代碼中有詳細的說明

class ExcelReader:
    """
    讀取excel文件中的內容。返回list。

    如：
    excel中內容爲：
    | A  | B  | C  |
    | A1 | B1 | C1 |
    | A2 | B2 | C2 |

    如果 print(ExcelReader(excel, title_line=True).data)，輸出結果：
    [{A: A1, B: B1, C:C1}, {A:A2, B:B2, C:C2}]

    如果 print(ExcelReader(excel, title_line=False).data)，輸出結果：
    [[A,B,C], [A1,B1,C1], [A2,B2,C2]]

    可以指定sheet，通過index或者name：
    ExcelReader(excel, sheet=2)
    ExcelReader(excel, sheet='BaiDuTest')
    """
    def __init__(self, excel, sheet=0, title_line=True):
        if os.path.exists(excel):
            self.excel = excel
        else:
            raise FileNotFoundError('文件不存在！')
        self.sheet = sheet
        self.title_line = title_line
        self._data = list()

    @property
    def data(self):
        if not self._data:
            workbook = open_workbook(self.excel)
            if type(self.sheet) not in [int, str]:
                raise SheetTypeError('Please pass in <type int> or <type str>, not {0}'.format(type(self.sheet)))
            elif type(self.sheet) == int:
                s = workbook.sheet_by_index(self.sheet)
            else:
                s = workbook.sheet_by_name(self.sheet)

            if self.title_line:
                title = s.row_values(0)  # 首行爲title
                for col in range(1, s.nrows):
                    # 依次遍歷其餘行，與首行組成dict，拼到self._data中
                    self._data.append(dict(zip(title, s.row_values(col))))
            else:
                for col in range(0, s.nrows):
                    # 遍歷所有行，拼到self._data中
                    self._data.append(s.row_values(col))
        return self._data


class ExcelAdd:
    """
    向已存在excel文件中寫入內容
    """
    def __init__(self, excel, sheet=0):
        """
        :param excel: 文件地址
        :param sheet: 選定sheet  默認爲：0
        """
        if os.path.exists(excel):
            self.excel = excel
        else:
            raise FileNotFoundError('文件不存在！')
        self.sheet = sheet

    def add_data(self, row=0, col=0, data=""):
        """
        寫入數據（覆蓋原數據）
        :param row: 行
        :param col: 列
        :param data: 數據
        :return:
        """
        rb = xlrd.open_workbook(self.excel)
        wb = copy(rb)
        ws = wb.get_sheet(self.sheet)
        ws.write(row, col, data)
        wb.save(self.excel)

使用上分別使用3個參數就可以了：

excel：文件地址

sheet：初始讀取的行（默認第一行開始，有的時候不用要讀取第一行）

title_line：每次是否需標題項“第一行”（默認需要）

 

row：數據插入的行

col：數據插入的列

row：插入的數據

from util_s.file_reader import ExcelReader
e = r'E:\Study\Selenium\SeleniumPython\data\baidu.xlsx'
reader = ExcelReader(e, sheet=0，title_line=True)
print(reader.data)

from util_s.file_reader import ExcelAdd
e = r'E:\Study\Selenium\SeleniumPython\data\baidu.xlsx'
row = 1
col = 1
data = “數據”
ExcelAdd(e).add_data(row , col , data )