自動化不知如何引數化?xlrd來幫你解決
阿新 • • 發佈:2020-07-25
平時在做自動化測試的時候,一直都是要求資料與業務邏輯分離。把測試資料都寫在業務裡面的話,比較混雜。為了方便管理測試資料,所以引入了python的一個擴充套件庫--xlrd。該庫使用簡單,能滿足自動化測試的資料分離的需求。
閒語少聊,我們直接來看,xlrd是如何完成資料讀取的。
安裝
安裝很簡單,方式①:直接命令pip install xlrd;方式②:在Pycharm中直接安裝即可。
基礎命令
①開啟excel檔案並建立物件
excel_object = xlrd.open_workbook(excel_path)
②根據工作表名稱獲取資料
sheet_object = excel_object.sheet_by_name('Sheet1')
③獲取excel檔案中所有工作表名稱
print(excel_object.sheet_names())
④獲取有效行數
row_nrows = sheet_object.nrows
⑤獲取有效列數
col_ncols = sheet_object.ncols
⑥獲取當前行的單元格長度
row_length = sheet_object.row_len(1)
⑦獲取第一行資料
rows_cell_value = sheet_object.row_values(0)
⑧獲取座標為第一行、第一列單元格資料
row_cell_value1 = sheet_object.cell_value(0, 0)# 座標起始位從0開始 或者 row_cell_value1 = sheet_object.cell(0, 0).value 或者 row_cell_value1 = sheet_object.row(0)[0].value
⑨獲取座標為第二行、第四列單元格資料
row_cell_value2 = sheet_object.cell_value(1, 3)
⑩獲取第四列資料
cols_cell_value = sheet_object.col_values(3)
單元格型別
單元格型別區分為六類,讀取單元格型別如下所示:
0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error cell_value_type1 = sheet_object.cell_type(2, 0) 或者 cell_value_type1 = sheet_object.cell(2, 0).value
命令實踐
我們來寫個例子驗證下這些命令,示例程式碼如下:
''' @author: wenyihuqingjiu @project: Pytest_Auto_Test @file: demo @time: 2020-07-23 21:40 @desc: ''' import xlrd import os current_path = os.path.dirname(__file__) excel_path = os.path.join(current_path, '..', 'file/excel/testcase.xlsx') excel_object = xlrd.open_workbook(excel_path) sheet_object = excel_object.sheet_by_name('Sheet1') # 獲取第一行資料 rows_cell_value = sheet_object.row_values(0) print('獲取到的第一行資料為:' + str(rows_cell_value)) # 獲取第一行、第一列單元格資料 row_cell_value1 = sheet_object.cell_value(0, 0) print('獲取座標為第一行、第一列單元格的資料為:' + row_cell_value1) # 獲取第二行、第四列單元格資料 row_cell_value2 = sheet_object.cell_value(1, 3) print('獲取座標為第二行、第四列單元格的資料為:' + row_cell_value2) # 獲取第四列資料 cols_cell_value = sheet_object.col_values(3) print('獲取第四列的資料為:' + str(cols_cell_value)) # 獲取單元格型別 # 0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error cell_value_type1 = sheet_object.cell_type(2, 0) print('單元格型別為:' + str(cell_value_type1)) cell_value_type2 = sheet_object.cell_type(1, 0) print('單元格型別為:' + str(cell_value_type2)) cell_value_type3 = sheet_object.cell_type(1, 7) print('單元格型別為:' + str(cell_value_type3)) cell_value_type4 = sheet_object.cell_type(1, 8) print('單元格型別為:' + str(cell_value_type4)) cell_value_type5 = sheet_object.cell_type(2, 8) print('單元格型別為:' + str(cell_value_type5)) # 獲取有效行數 row_nrows = sheet_object.nrows print('當前sheet頁中有效行數為:' + str(row_nrows)) # 獲取有效列數 col_ncols = sheet_object.ncols print('當前sheet頁中有效列數為:' + str(col_ncols)) # 獲取當前行的單元格長度 row_length = sheet_object.row_len(0) print('第一行的有效單元格長度為:' + str(row_length)) # 獲取所有工作表名稱 print('當前excel中的工作表名稱為:'+ str(excel_object.sheet_names()))
excel文件內容如下所示:
執行程式碼,我們結合excel文件內容來對比看下輸出結果,如下所示:
我們從上述簡單的命令瞭解到,讀取資料的方式其實很簡單,程式碼中加上邏輯便可獲取整個sheet頁的資料,如下所示,獲取當前sheet頁的資料程式碼:
# 獲取有效行數 row_index = sheet_object.nrows # 獲取有效列數 col_index = sheet_object.ncols # 定義空陣列 all_data_list = [] # 獲取首行資料 first_row = sheet_object.row_values(0) for row in range(1, row_index): # 定義空列表 row_dict = {} for col in range(col_index): c_cell = sheet_object.cell_value(row, col) # 迴圈每一個有效的單元格,將欄位與值對應儲存到字典中 row_dict[first_row[col]] = c_cell # 再將字典追加到列表中 all_data_list.append(row_dict) print(str(row_dict))
檢視控制檯,打印出來的資料如下所示:
excel檔案內容還是如上截圖,可以對比檢視下。
{'模組': '搜尋測試', '用例編號': '搜尋測試01', '輸入說明': '1.輸入:', '操作輸入值': '溫一壺清酒 部落格園', '按鈕操作': '2.點選搜尋按鈕', '斷言': '溫一壺清酒 部落格園_百度搜索', '預期值': '正確斷言', '測試序號': 1.0, '': 44035.0} {'模組': '', '用例編號': '搜尋測試02', '輸入說明': '', '操作輸入值': '溫一壺清酒 抓包', '按鈕操作': '', '斷言': '', '預期值': '錯誤斷言', '測試序號': 2.0, '': 1} {'模組': '', '用例編號': '搜尋測試03', '輸入說明': '', '操作輸入值': '溫一壺清酒 appium', '按鈕操作': '', '斷言': '溫一壺清酒 部落格園', '預期值': '', '測試序號': 3.0, '': ''}
我們從上述的列印資料來看,細心的博友們應該發現問題了。
我們可以看出,合併單元格的資料,只有首個單元格有資料顯示;單元格資料型別不正確。
資料轉換
今天,我們先來解決單元格資料型別顯示不正確的問題。
解決該問題的方式很簡單,就是將單元格型別獲取後,判斷單元格型別並轉換格式,示例程式碼如下所示:
''' @author: wenyihuqingjiu @project: Pytest_Auto_Test @file: demo @time: 2020-07-23 21:40 @desc: ''' import xlrd import os import datetime from xlrd import xldate_as_tuple current_path = os.path.dirname(__file__) excel_path = os.path.join(current_path, '..', 'file/excel/testcase.xlsx') excel_object = xlrd.open_workbook(excel_path) sheet_object = excel_object.sheet_by_name('Sheet1') # 獲取有效行數 row_index = sheet_object.nrows # 獲取有效列數 col_index = sheet_object.ncols # 定義空陣列 all_data_list = [] # 獲取首行資料 first_row = sheet_object.row_values(0) for row in range(1, row_index): # 定義空列表 row_dict = {} for col in range(col_index): c_cell = sheet_object.cell_value(row, col) # 獲取單元格資料型別 c_type = sheet_object.cell(row, col).ctype if c_type == 2 and c_cell % 1 == 0: # 如果是整形 c_cell = int(c_cell) elif c_type == 3: # 轉成datetime物件 date = datetime.datetime(*xldate_as_tuple(c_cell, 0)) c_cell = date.strftime('%Y-%m-%d') elif c_type == 4: c_cell = True if c_cell == 1 else False # 迴圈每一個有效的單元格,將欄位與值對應儲存到字典中 row_dict[first_row[col]] = c_cell # 再將字典追加到列表中 all_data_list.append(row_dict) print(str(row_dict))
執行程式碼,我們再來看下列印結果,如下所示:
{'模組': '搜尋測試', '用例編號': '搜尋測試01', '輸入說明': '1.輸入:', '操作輸入值': '溫一壺清酒 部落格園', '按鈕操作': '2.點選搜尋按鈕', '斷言': '溫一壺清酒 部落格園_百度搜索', '預期值': '正確斷言', '測試序號': 1, '': '2020-07-23'} {'模組': '', '用例編號': '搜尋測試02', '輸入說明': '', '操作輸入值': '溫一壺清酒 抓包', '按鈕操作': '', '斷言': '', '預期值': '錯誤斷言', '測試序號': 2, '': True} {'模組': '', '用例編號': '搜尋測試03', '輸入說明': '', '操作輸入值': '溫一壺清酒 appium', '按鈕操作': '', '斷言': '溫一壺清酒 部落格園', '預期值': '', '測試序號': 3, '': ''}
格式轉換後,資料就列印正常了,資料型別錯誤的問題就解決了。
好了,今天的分享就到此結束啦,希望對博友們有幫助,下期分享獲取合併單元格資料問題,敬請期