接口自动化（二）--操作Excel获取需要数据

这一部分的内容记述一下对Excel表格的操作，本实战中的测试用例是由Excel来管理的，因此操作Excel是重要的一部分。

再次贴出这张图，所有的测试用例都在这个sheet内，请求数据真实存放在json文件内，表格中的请求数据列放置的是json文件内字典的key值。

操作Excel主要就是取出Excel中的case（即数据），然后执行，重点是怎样取出数据，因为有些数据是不需要的，所以不能用遍历Excel的方法来取数据。

思路一：把列数作为常量，行数作为变量来取想要的数据

思路二：把Excel中不需要的列数据剪切掉，留下需要的数据，然后遍历Excel

这里使用的思路一

下面是excelColNum.py的源码（请根据需要灵活使用，模块作用参考第一篇文章）：

 class ExcelColNum():
     case_id = 0
     model = 1
     case_name = 2
     url = 4
     header = 5
     run = 6
     requestMethord = 7
     caseDepend = 8
     depReKey = 9
     depKey = 10
     requestData = 11
     respectResult = 12
     returnData = 13
     realResult = 14

     #获取用例ID
     def get_caseID(self):
         return ExcelColNum.case_id

     #获取请求的url
     def get_url(self):
         return ExcelColNum.url

     #获取是否运行
     def get_run(self):
         return ExcelColNum.run

     #获取请求方式
     def get_methord(self):
         return ExcelColNum.requestMethord

     #获取请求数据
     def get_requestData(self):
         return ExcelColNum.requestData

     #或取是否携带header
     def get_haeder(self):
         return ExcelColNum.header

     #获取被依赖的caseID
     def getDepCaseID(self):
         return ExcelColNum.caseDepend

     #获取被依赖的用例的返回数据的key值
     def getDepReKey(self):
         return ExcelColNum.depReKey

     #获取依赖的需要传入值的key值
     def getDepKey(self):
         return ExcelColNum.depKey

     #获取预期结果
     def getRespectResult(self):
         return ExcelColNum.respectResult

     #获取返回数据
     def getReturnResult(self):
         return ExcelColNum.returnData

     #实际结果
     def getRealResult(self):
         return ExcelColNum.realResult

然后是operateExcel.py的源码：

 import xlrd
 from xlutils.copy import copy

 class operateExcel():
     def __init__(self,excelPath=None,sheetName=None):
         #容错处理
         if excelPath:
             self.excelPath = excelPath
             self.sheetName = sheetName
         else:
             self.excelPath = "******"                #Excel文件的路径
             self.sheetName = "******"                #sheet的名字
         self.data = self.openExcel()
     #打开Excel表格
     def openExcel(self):
         file = xlrd.open_workbook(self.excelPath)
         sheet = file.sheet_by_name(self.sheetName)
         return sheet

     #获取sheet的行数
     def getSheetRow(self):
         rows = self.data.nrows
         return rows

     #获取sheet的列数
     def getSheetCol(self):
         cols = self.data.ncols
         return cols

     #获取某一个单元格的内容
     def getCellContent(self,rowNum,colNum):
         content = self.data.cell_value(rowNum,colNum)
         return content

     #向excel里追加写入数据
     def writeValue(self,rowNum,colNum,value):
         file = xlrd.open_workbook(self.excelPath)
         workbook = copy(file)
         sheetData = workbook.get_sheet(0)
         sheetData.write(rowNum,colNum,value)
         workbook.save(self.excelPath)

     #获取Excel中某一列的内容
     def getColData(self,colNum=None):
         if colNum != None:
             colData = self.data.col_values(colNum)
         else:
             colData = self.data.col_values(0)
         return colData

     #根据依赖的caseID找到对应的行号
     def depRowNum(self,caseID):
         rowNum = 0
         colDatas = self.getColData()
         for colData in colDatas:
             if caseID in colData:
                 return rowNum
             rowNum += 1

     # 根据行号获取对应行的整行内容
     def getRowData(self, rowNum):
         rowData = self.data.row_values(rowNum)
         return rowData

     #根据依赖的caseID，获取依赖case的行号，根据行号取出对应行的内容
     def dependCaseData(self,caseID):
         rowNum = self.depRowNum(caseID)
         rowData = self.getRowData(rowNum)
         return rowData

下面是get_Case的源码：

 from data import operateExcel
 from data import excelColNum
 from data import operateJson
 import json

 class getCaseData():
     def __init__(self):
         self.operateExcel = operateExcel.operateExcel()
         self.excelColNum = excelColNum.ExcelColNum()
         self.operateJson = operateJson.operateJson()

     # 获取测试用例的行数,即case的个数，即表格的行数
     def getExcelRow(self):
         excelRows = self.operateExcel.getSheetRow()
         return excelRows

     #获取是否执行
     def getRun_or_not(self,rowNum):
         flag = None
         colNum = self.excelColNum.get_run()
         runOrNot = self.operateExcel.getCellContent(rowNum,colNum)
         if runOrNot == "yes":
             flag = True
         else:
             flag = False
         return flag

     #获取是否携带header,header可以放在配置文件内
     def getHeader(self,rowNum):
         colNum = self.excelColNum.get_haeder()
         headers = self.operateExcel.getCellContent(rowNum,colNum)
         headers = json.loads(headers)
         return headers

     #获取请求方式
     def getRequestMethord(self,rowNum):
         colNum = self.excelColNum.get_methord()
         requestmethord = self.operateExcel.getCellContent(rowNum,colNum)
         return requestmethord

     #获取请求的url
     def getUrl(self,rowNum):
         colNum = self.excelColNum.get_url()
         url = self.operateExcel.getCellContent(rowNum,colNum)
         return url

     #获取请求数据
     def getRequestData(self,rowNum):
         colNum = self.excelColNum.get_requestData()
         reqData = self.operateExcel.getCellContent(rowNum,colNum)
         if reqData == "":
             return None
         else:
             return reqData

     #获取json文件内的请求数据
     def getJsonData(self,rowNum):
         data = self.operateJson.getJsonData(self.getRequestData(rowNum))
         data = json.dumps(data)
         return data

     #获取预期结果
     def getRespectResult(self,rowNum):
         colNum = self.excelColNum.getRespectResult()
         resResult = self.operateExcel.getCellContent(rowNum,colNum)
         if resResult == "":
             return None
         return resResult

     #写入返回数据
     def wriReturn(self,rowNum,value):
         colNum = self.excelColNum.getReturnResult()
         self.operateExcel.writeValue(rowNum,colNum,value)

     #写入实际结果(pass、fail)
     def wriRealResult(self,rowNum,value):
         colNum = self.excelColNum.getRealResult()
         self.operateExcel.writeValue(rowNum,colNum,value)

     # 判断是否有case依赖
     def judgeCaseDep(self, rowNum):
         colNum = self.excelColNum.getDepCaseID()
         caseID = self.operateExcel.getCellContent(rowNum, colNum)
         if caseID != "":
             return caseID
         else:
             return None

     #获取Excel中被依赖返回数据中的值
     def getDepReKey(self,rowNum):
         colNum = self.excelColNum.getDepReKey()
         reKey = self.operateExcel.getCellContent(rowNum,colNum)
         if reKey == "":
             return None
         else:
             return reKey

     #获取Excel中需要传入的依赖的数据
     def getDependKey(self,rowNum):
         colNum = self.excelColNum.getDepKey()
         depkey = self.operateExcel.getCellContent(rowNum,colNum)
         if depkey != "":
             return depkey
         else:
             return None