Python 接口測試之Excel表格數據操做方法封裝

　　引言

　　咱們在作接口測試，常常會用到excel去管理測試數據，對Excel的操做比較頻繁，那麼使用python如何操做Excel文件的讀與寫呢？因爲以前講的都是大的框框，沒有講這麼小的模塊使用，如今就化整爲0的講解。

　　讀寫模塊介紹

　　python有三個模塊是對Excel文件的操做，分別是：openpyxl,xlrd和xlwt。下面簡單的介紹一下各個模塊的特色及使用案例。

 

　　xlrd

　　xlrd是用來從Excel中讀寫數據的，但我日常只用它進行讀操做，寫操做會遇到些問題。用xlrd進行讀取比較方便，流程和日常手動操做Excel同樣，打開工做簿(Workbook)，選擇工做表(sheets)，而後操做單元格(cell)。下面舉個例子，例如要打開當前目錄下名爲」data.xlsx」的Excel文件，選擇第一張工做表，而後讀取第一行的所有內容並打印出來。Python代碼以下：

import os
import xlrd
from xlutils.copy import copy

# 獲取當前文件的絕對路徑
curPath = os.path.abspath(os.path.dirname(__file__))
print(curPath)
# 獲取項目根目錄
rootPath = os.path.abspath(os.path.dirname(curPath))
print(rootPath)

# 獲取文件路徑
file_path = r'APItest_ddt\data\data.xls'
file_path = os.path.join(rootPath,file_path)
print(file_path)
#打開excel文件
data=xlrd.open_workbook(file_path)
#獲取第一張工做表（經過索引的方式）
table=data.sheets()[0]
#data_list用來存放數據
data_list=[]
#將table中第一行的數據讀取並添加到data_list中
data_list.extend(table.row_values(0))
#打印出第一行的所有數據
for item in data_list:
    print(item)


# 寫入數據
copy_data = copy(data)
# 讀取複製的excel的sheet頁
copy_data_to_sheet = copy_data.get_sheet(0)
# 經過get_sheet()獲取的sheet有write()方法，寫入數據
copy_data_to_sheet.write(5,5,"這是我寫入的測試數據：哈哈")
# 保存數據
copy_data.save(file_path)

 

運行結果：

C:\Users\Administrator\Desktop\APItest_ddt
C:\Users\Administrator\Desktop
C:\Users\Administrator\Desktop\APItest_ddt/data/data.xls
地址

 

打開data.xls文件：

 

 

在處理excel數據時發現了xlwt的侷限性–不能寫入超過65535行、256列的數據（由於它只支持Excel 2003及以前的版本，在這些版本的Excel中行數和列數有此限制），這對於實際應用仍是不夠的。爲此通過一番尋找發現了一個支持07/10/13版本Excel的openpyxl，雖然功能很強大，可是操做起來感受沒有xlwt方便。以上是xlrd的幾個簡單操做，而且能夠發現，xlrd能夠讀，也能夠寫的。只是寫的話，沒那麼方便。

 

　　xlwt

　　若是說xlrd不是一個單純的Reader（若是把xlrd中的後兩個字符當作Reader，那麼xlwt後兩個字符相似當作Writer），那麼xlwt就是一個純粹的Writer了，由於它只能對Excel進行寫操做。xlwt和xlrd不光名字像，連不少函數和操做格式也是徹底相同。下面簡要概括一下經常使用操做。

import xlwt
# 新建一個Excel文件（只能經過新建寫入）
data=xlwt.Workbook()
# 新建一個工做表
table=data.add_sheet('name')
# 寫入數據到A1單元格
table.write(0,0,u'呵呵')
# 注意：若是對同一個單元格重複操做，會引起overwrite Exception，想要取消該功能，須要在添加工做表時指定爲可覆蓋，像下面這樣
table=data.add_sheet('sheet1',cell_overwrite_ok=True)
# 保存
data.save('test.xls')

查看結果：

 

 

　　openpyxl

　　該模塊支持最新版的Excel文件格式，對Excel文件具備響應的讀寫操做，對此有專門的Reader和Writer兩個類，便於對Excel文件的操做。雖然如此，但我通常仍是用默認的workbook來進行操做。經常使用操做概括以下：

from openpyxl.reader.excel import load_workbook

wb=load_workbook(filename=r"openpyxl_test.xlsx")

import openpyxl

if __name__ == '__main__':

    wb = load_workbook(filename=r"openpyxl_test.xlsx")
    # 顯示全部工做表的名字
    sheet_name = wb.get_sheet_names()
    print(sheet_name)
    # 取第一張表
    sheetnames = wb.sheetnames
    ws = wb[sheetnames[0]]
    # 顯示錶名，錶行數，表列數
    print("Work Sheet Title:", ws.title)
    print("Work Sheet Rows:", ws.max_row)
    print("Work Sheet Cols:", ws.max_column)
    # 獲取指定行的值，如第三行
    row3_values = []
    row3_cell_list = list(ws.rows)[2]
    for cell in row3_cell_list:
        row3_values.append(cell.value)
    print(row3_values)  # output:['A3', 'B3', 'C3', 'D3']

    # 獲取全部行的數據
    # 創建存儲數據的字典
    data = {}
    # 獲取表格全部值
    # 法1:
    for i in range(0, ws.max_row):
        every_row_values = []
        every_row_cell_list = list(ws.rows)[i]
        for cell in every_row_cell_list:
            every_row_values.append(cell.value)
        data[i + 1] = every_row_values
    print(data)
    # {1: ['A1', 'B1', 'C1', 'D1'], 2: ['A2', 'B2', 'C2', 'D2'], 3: ['A3', 'B3', 'C3', 'D3'], 4: ['A4', 'B4', 'C4', 'D4'], 5: ['A5', 'B5', 'C5', 'D5'],
    #  6: ['A6', 'B6', 'C6', 'D6'], 7: ['A7', 'B7', 'C7', 'D7'], 8: ['A8', 'B8', 'C8', 'D8'], 9: ['A9', 'B9', 'C9', 'D9'], 10: ['A10', 'B10', 'C10', 'D10'],
    # 11: ['A11', 'B11', 'C11', 'D11']}
    # 法2
    for row in ws.rows:
        line = [cell.value for cell in row]
    print(
        line)  # output:['A1', 'B1', 'C1', 'D1'],['A2', 'B2', 'C2', 'D2'], ['A3', 'B3', 'C3', 'D3'],['A4', 'B4', 'C4', 'D4'], ['A5', 'B5', 'C5', 'D5'],['A6', 'B6', 'C6', 'D6'],['A7', 'B7', 'C7', 'D7'],['A8', 'B8', 'C8', 'D8'],['A9', 'B9', 'C9', 'D9'],['A10', 'B10', 'C10', 'D10'],['A11', 'B11', 'C11', 'D11']
    # 獲取某個區間的值，例：得到了以A3爲左上角，C6爲右下角矩形區域的全部單元格
    # 法1：使用range
    data1 = {}
    for i in range(2, 6):
        every_row_values = []
        every_row_cell_list = list(ws.rows)[i]
        for cell in every_row_cell_list:
            every_row_values.append(cell.value)
        data1[i + 1] = every_row_values
    print(data1)
    # output{3: ['A3', 'B3', 'C3', 'D3'], 4: ['A4', 'B4', 'C4', 'D4'], 5: ['A5', 'B5', 'C5', 'D5'], 6: ['A6', 'B6', 'C6', 'D6']}

    # 法2:使用切片
    data_list = []
    for row_cell in ws['A3':'C6']:
        every_row_value = []
        for cell in row_cell:
            every_row_value.append(cell.value)
        data_list.append(every_row_value)
    print(data_list)  # output:[['A3', 'B3', 'C3'], ['A4', 'B4', 'C4'], ['A5', 'B5', 'C5'], ['A6', 'B6', 'C6']]





    # 測試寫入數據：

    wb = load_workbook(filename=r"openpyxl_test.xlsx")
    ws = wb.active

    # 第一個sheet是ws
    ws = wb.worksheets[0]

    # 設置ws的名稱
    ws.title = "range names"

    # 向某個單元格中寫入數據
    ws.cell(11,11).value = u'Test write data!!!'

    # 最後保存文件
    wb.save(filename=r"openpyxl_test.xlsx")

　　

　　結果以下：

 

 

 

 

　　封裝讀寫

　　上面已經介紹了三個模塊的使用，那麼在搭建接口測試框架的時候，不可能這樣寫，一方面不利於代碼可讀性與維護，另外一方面不雅觀。因此，要對零散的代碼進行二次封裝，一下就是針對經常使用方法進行二次封裝：

import xlrd
from xlutils.copy import copy
import os
# 獲取當前文件的絕對路徑
curPath = os.path.abspath(os.path.dirname(__file__))
print(curPath)
# 獲取項目根目錄
rootPath = os.path.abspath(os.path.dirname(curPath))
print(rootPath)

# 類中使用裝飾器 @classmethod定義方法，是類方法
# 類中使用裝飾器 @staticmethod定義方法，是靜態方法


class Operate_Excel(object):
    """
    操做excel類

    """
    # 定義構造函數，建立對象自動執行
    def __init__(self,file_path=None,sheet_id=None):
        """

        :param file_path:若是沒傳值，默認爲excel路徑
        :param sheet_id:若是沒傳值，默認爲第一個sheet頁
        """
        if file_path:
            # 成員變量
            self.file_path = file_path
            self.sheet_id = sheet_id
        else:
            self.file_path = r'data/util_data/operate_excel.xls'
            # 將文件目錄拼接成絕對路徑
            self.file_path = os.path.join(rootPath,self.file_path)
            print(self.file_path)

        if sheet_id:
            self.sheet_id = sheet_id
        else:
            self.sheet_id = 0
        # 調用成員方法
        self.sheet_table = self.get_sheet()

    """成員方法"""
    # 獲取sheet頁操做對象
    def get_sheet(self):
        data = xlrd.open_workbook(self.file_path)
        sheet_table = data.sheets()[self.sheet_id]
        return sheet_table

    # 獲取sheet頁的行數和列數，返回的是一個元組
    def get_sheet_nrows_ncols(self):
        return self.sheet_table.nrows,self.sheet_table.ncols

    # 獲取sheet頁的行數
    def get_sheet_nrows(self):
        return self.sheet_table.nrows

    # 獲取sheet頁的列數
    def get_sheet_ncols(self):
        return self.sheet_table.ncols

    # 獲取具體單元格的數據
    def get_sheet_cell(self,row,col):
        """
        :params row: 單元格行值
        :params col: 單元格列值
        :return: cell_data
        """
        cell_data = self.sheet_table.cell_value(row,col)
        return cell_data

    # 寫入數據到excel中
    def write_to_excel(self,row,col,values):
        # 打開excel文件讀取數據句柄
        data = xlrd.open_workbook(self.file_path)
        # 複製excel
        copy_data = copy(data)
        # 讀取複製的excel的sheet頁
        copy_data_to_sheet = copy_data.get_sheet(0)
        # 經過get_sheet()獲取的sheet有write()方法，寫入數據
        copy_data_to_sheet.write(row,col,values)
        # 保存數據
        copy_data.save(self.file_path)

if __name__ == '__main__':
    read_xls = Operate_Excel()
    print("獲取excel表的行數與列數,返回元組格式: ",read_xls.get_sheet_nrows_ncols())
    print("獲取excel表的行數: ",read_xls.get_sheet_nrows())
    print("獲取excel表的列數: ",read_xls.get_sheet_ncols())
    print("獲取excel表的單元格(1,1)的值: ",read_xls.get_sheet_cell(1,1))
    print("獲取excel表的單元格(1,2)的值: ",read_xls.get_sheet_cell(1,2))
    print("獲取excel表的單元格(2,2)的值: ",read_xls.get_sheet_cell(2,2))
    print("寫入excel表的單元格(2,2)的值: ",read_xls.write_to_excel(8,8,'test'))

 

運行結果：

獲取excel表的行數與列數,返回元組格式:  (18, 9)
獲取excel表的行數:  18
獲取excel表的列數:  9
獲取excel表的單元格(1,1)的值:  case_name01
獲取excel表的單元格(1,2)的值:  case_method01
獲取excel表的單元格(2,2)的值:  case_method02
寫入excel表的單元格(2,2)的值:  None

　　excel文件：

 

 

　　總結

　　到此，Excel讀寫功能已經介紹完，以上功能大部分知足平常使用，固然你也能夠深刻研究，繼續加入其它方法進來。以上源碼，已上傳到羣文件中，能夠加入QQ測試開發交流羣：696400122索取。