接口自动化（二）--操作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