python---讀取/寫入excel用例數據

使用ｅｘｃｅｌ管理用例

①、讀取excel的用例數據，每一類參數保存在list中返回；
②、那么接下來使用unitest編寫用例時，可以去調用這個函數。使用里面的數據；

個人認為好處是，大部分人還是習慣excel的編寫用例習慣，將實際用例與代碼分離，有助於分工和維護

一、在excel中寫好用例后，在python(unitest)中調用

#寫文件
import xlwt;
#讀文件
import xlrd;
#若要對excel文件進行修改必須！要引入這個工具模塊，其中含有復制和修改功能---------注意引入方式
import xlutils;
from xltuils.copy import copy;


import unittest;
import requests;
#使用寫文件模式打開文件
open_workbook=xlrd.open_workbook(r'D:\Users\4399-3046\Desktop\test.xls');
#最終返回的是文件類型
print(open_workbook);
#打印該文件下，對應數據表的行數
#print(open_workbook.sheet_by_name('mytest').nrows);

#打印該文件下，對應數據表的列數
#print(open_workbook.sheet_by_name('mytest').ncols);
#for item_rows in range(1,open_workbook.sheet_by_name('mytest').nrows):
    #for item_cols in range(1,open_workbook.sheet_by_name('mytest').ncols):
        #print(open_workbook.sheet_by_name('mytest').cell(item_rows,item_cols));


#循環打印出所有數據,並分別保存在list中
#編號
listkey=[];
#測試地址
listurl=[];
#提交方式
listtype=[];
#參數1
listcanshu1=[];
#參數2
listcanshu2=[];
#預期結果
listyuqi=[];
#實際結果
listresult=[];
#在unitest框架中，通過調用參數，執行用例，將實際結果比對預期結果是否一致，若一致，則通過，否則用例失敗，並將實際結果寫入excel中（寫入代碼后文將寫到）
for item_row in range(1,open_workbook.sheet_by_name('mytest').nrows):
    #獲取用例編號
    listkey.append(open_workbook.sheet_by_name('mytest').cell(item_row,0).value);
    # 獲取參數1
    listcanshu1.append(open_workbook.sheet_by_name('mytest').cell(item_row, 1).value);
    # 獲取參數2
    listcanshu2.append(open_workbook.sheet_by_name('mytest').cell(item_row, 2).calue);
    # 獲取預期結果
    listyuqi.append(open_workbook.sheet_by_name('mytest').cell(item_row, 3).value);
    # 獲取實際結果
    listresult.append(open_workbook.sheet_by_name('mytest').cell(item_row, 4).value);------------->注意：必須要有【.value】，因為返回的是dict字典，而實際我們只需要其中的具體值


print(listkey);
print(listcanshu1);
print(listcanshu2);
print(listyuqi);
print(listresult);

 

運行結果：

 2、新建文件，並在新文件中寫入數據

 open_workbook=xlwt.Workbook();---定義一個工作簿


new_sheet=open_workbook.add_sheet('lianxi',cell_overwrite_ok = True);----在工作簿中新增一個工作表，自定義一個工作表名稱

注意1：若當前單元格已經有值，無法重復寫入覆蓋原有，會提示異常，需要在新建表單sheet時，就要指定，可以重復寫入【cell_overwrite_ok = True】，默認是false

new_sheet.write(0,5,'lianxi_test');-----在工作表中寫入數據（注意：寫入數據只在工作表sheet中生效）---

注意2：如果當前目錄下已存在同名的工作簿名稱，則會報錯

open_workbook.save(r'D:\Users\Desktop\test01.xls');----保存工作簿（需要指定地址和名稱）

 

 

3、在已有文件下excel下，寫入數據

1#思路：打開文件----復制文件----在文件中寫入數據----刪除原文件----保存新文件 

#打開當前已經存在的工作簿
open_workbook2=xlrd.open_workbook(r'D:\Users\Desktop\test.xls');
#復制當前工作簿
open_sheet2=copy(open_workbook2);
#打開置頂索引的工作表 5 open_ws=open_sheet2.get_sheet(3);
#在指定的單元格寫入數據
open_ws.write(1,4,'TEST');
#移除原有的文件 8 os.remove(r'D:\Users\4399-3046\Desktop\test.xls')
#保存新文件 9 open_sheet2.save(r'D:\Users\4399-3046\Desktop\test.xls');

運行結果：

 

 

4、擴展--結合python-unitest,通常我們需要將執行結果自動寫入表格中
5、拓展--將寫文件和讀文件，分別封裝為公共方法調用（已完成）

#by zhonghuihong
import xlwt;
import xlrd;
from xlutils.copy import copy;
import os;
#此接口主要用於excel數據的讀取和寫入

def get_excel_data(paths,sheetName):
    open_workbook=xlrd.open_workbook(paths);
    open_sheet=open_workbook.sheet_by_name(sheetName);
    rows=open_sheet.nrows;
    listkey = [];
    listcanshu1 = [];
    listcanshu2 = [];
    listyuqi = [];
    listresult = [];
    for item_row in range(1, rows):
        listkey.append(open_sheet.cell(item_row, 0).value);
        listcanshu1.append(open_sheet.cell(item_row, 1).value);
        listcanshu2.append(open_sheet.cell(item_row, 2).value);
        listyuqi.append(open_sheet.cell(item_row, 3).value);
        listresult.append(open_sheet.cell(item_row, 4).value);
    return listkey,listcanshu1,listcanshu2,listyuqi,listresult;


def write_excel_data(paths,sheet_id,rows_id,cols_id,values):
    open_workbook2 = xlrd.open_workbook(paths);
    open_sheet2 = copy(open_workbook2);
    open_ws = open_sheet2.get_sheet(sheet_id);
    open_ws.write(rows_id,cols_id,values);
    os.remove(paths);
    open_sheet2.save(paths);
    return 'success';

 

遇到的問題
1、讀取的整數變成浮點類型，解決辦法

 2、腳本中引入copy方法，但提示TypeError: 'module' object is not callable的解決辦法：將引入語句寫成from xlutils.copy import copy;

3、指定sheet時（ow.get_sheet[0];），提示TypeError: 'method' object is not subscriptable

產生原因：subscriptable的意思是可有下標，錯誤的原因就是：對象不具有下標，即把不具有下標操作的對象用成了對象[i]，比如int對象變量[i]就會報錯。仔細檢查錯誤行。

代碼最終優化：

*增加try--exception異常捕獲

*增加寫入輸入不能為空

#寫數據（文件薄,表格id，行id，列id，要寫的內容）
def write_excel_data(paths,sheet_id,rows_id,cols_id,values):
    if(len(values)==0 ):
        return '寫入數據不能為空';
    else:

      try:
          #如果寫入位置已經有值， 會直接使用新內容覆蓋
         open_workbook2 = xlrd.open_workbook(paths);
         open_sheet2 = copy(open_workbook2);
         open_ws = open_sheet2.get_sheet(sheet_id);
         open_ws.write(rows_id, cols_id, values);
         os.remove(paths);
         open_sheet2.save(paths);

         return 'success';
      except Exception as e:
         return e;


#讀取excel_data（文件薄地址，表格名）
   #通過循環讀出指定表格中的所有數據，以列表格式輸出

def get_excel_data(paths,sheetName):
        open_workbook = xlrd.open_workbook(paths);
        open_sheet = open_workbook.sheet_by_name(sheetName);

        if open_sheet.nrows <= 1:
            return '文件總行數小於1，請確認';

        else:
            res = []
            j = 1
            for i in list(range(open_sheet.nrows-1)):
                s = {};
                #行編號
                s['rowNum'] = i+1;
                values = open_sheet.row_values(j)
                for x in list(range(open_sheet.ncols)):
                    keys=open_sheet.row_values(0);
                    s[keys[x]] = values[x];
                res.append(s);
                j += 1;
            return res;