python接口自動化測試 - openpyxl封裝類

前言

為了更好的讓openpyxl在工作中使用，將openpyxl的常用操作封裝起來，這樣不僅復用性高，而且閱讀性好

 

直接上代碼

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
__title__  = openpyxl操作Excel工具類
"""

import openpyxl


class ExcelUtil:
    workBook = None
    workSheet = None

    def load_excel(self, path):
        """
        加載Excel
        :param path: 需要打開的Excel的路徑
        """
        self.workBook = openpyxl.load_workbook(path)

    def get_sheet_by_name(self, name):
        """
        獲取sheet對象
        :param name: sheet名
        """
        self.workSheet = self.workBook.get_sheet_by_name(name)

    def get_sheet_by_index(self, index=0):
        """
        獲取sheet對象
        :param index: sheet的索引
        """
        # 獲取workBook里所有的sheet名 -> list
        sheet_names = self.workBook.get_sheet_names()
        # 根據索引獲取指定sheet
        self.workSheet = self.workBook[sheet_names[index]]

    def get_cell_value(self, col, row):
        """
        獲取cell的值
        :param col: 所在列
        :param row: 所在行
        """
        try:
            return self.workSheet.cell(column=col, row=row).value
        except BaseException as e:
            return None

    def get_cell_value_by_xy(self, str):
        """
        獲取cell的值
        :param str: 坐標
        """
        try:
            return self.workSheet[str].value
        except BaseException as e:
            return None

    def get_sheet_rows(self):
        """
        獲取最大行數
        """
        return self.workSheet.max_row

    def get_sheet_cols(self):
        """
        獲取最大列數
        """
        return self.workSheet.max_column

    def write_data(self, row, col, value, path):
        """
        寫入數據
        """
        try:
            self.workSheet = self.workBook.active
            self.workSheet.cell(column=col, row=row, value=value)
            self.workBook.save(path)
        except BaseException as e:
            print(e)
            return None

    def get_excel_data(self):
        """
        獲取表所有數據
        :return: list
        """
        # 方式一
        data_list = tuple(self.workSheet.values)
        # 方式二
        # data_list = []
        # for i in range(self.get_sheet_rows()):
        #     data_list.append(self.get_row_value(i + 2))
        return data_list

    def get_row_value(self, row):
        """
        獲取某一行的內容
        :param row: 第幾行 -> str  **從1開始**
        :return: list
        """
        # 方式一
        row_list = self.get_excel_data()[row]
        # 方式二
        # row_list = []
        # for i in self.workSheet[str(row + 1)]:
        #     row_list.append(i.value)
        return row_list

    def get_col_value(self, col='A'):
        """
        獲取某一列的內容
        :param col: 第幾列 -> str
        :return: list
        """
        col_list = []
        for i in self.workSheet[col]:
            col_list.append(i.value)
        return col_list

    def get_row_num(self, case_id):
        """
        獲取行號
        :param case_id: 用例編號
        :return:
        """
        num = 1
        col_data = self.get_col_value()
        for data in col_data:
            if case_id == data:
                return num
            num += 1
        return 0


excelUtil = ExcelUtil()

if __name__ == '__main__':
    path = 'F:/interface/case/test.xlsx'
    # 讀取excel文件
    excelUtil.load_excel(path)
    # 獲取某個sheet
    excelUtil.get_sheet_by_name("Sheet1")
    excelUtil.get_sheet_by_index()
    # 獲取某個cell的值
    data = excelUtil.get_cell_value(col=1, row=1)
    print(data)
    data = excelUtil.get_cell_value_by_xy("A3")
    print(data)
    # 獲取sheet行數
    data = excelUtil.get_sheet_rows()
    print(data)
    # 獲取sheet列數
    data = excelUtil.get_sheet_cols()
    print(data)
    # 獲取某一行數據
    data = excelUtil.get_row_value(0)
    print(data)
    # 獲取某一列數據
    data = excelUtil.get_col_value()
    print(data)
    # 寫入數據
    excelUtil.write_data(row=9, col=1, value="test", path=path)
    # 獲取全部數據
    data = excelUtil.get_excel_data()
    print(data)
    # 獲取行號
    data = excelUtil.get_row_num('imooc_001')
    print(data)