python 讀取excel數據並將測試結果填入Excel

python 讀取excel數據並將測試結果填入Excel

讀取一個Excel中的一條數據用例，請求接口，然后返回結果並反填到excel中。過程中會生成請求回來的文本，當然還會生成一個xml文件。具體的excel文件如下：

代碼如下：

# -*- coding: UTF-8 -*- 
from xml.dom import minidom
import xlrd
import openpyxl
import requests
import json
import sys
import HTMLParser
import os
import re
import codecs
import time
import datetime

reload(sys)
sys.setdefaultencoding('utf-8')

class OptionExcelData(object):
    """對Excel進行操作，包括讀取請求參數，和填寫操作結果"""
    def __init__(self, excelFile,excelPath=''):
        self.excelFile = excelFile
        self.excelPath = excelPath
        self.caseList = []

    """
    傳入：傳入用例Excel名稱
    返回：[],其中元素為{},每個{}包含行號、城市、國家和期望結果的鍵值對
    """
    def getCaseList(self,excelFile,excelPath=''):
        readExcel = xlrd.open_workbook(fileName)   #讀取指定的Excel
        try:
            table = readExcel.sheet_by_index(0)    #獲取Excel的第一個sheet
            trows = table.nrows                    #獲取Excel的行數
            for n in range(1,trows):
                tmpdict = {}                       #把一行記錄寫進一個{}
                tmpdict['id'] = n                  #n是Excel中的第n行
                tmpdict['CityName'] = table.cell(n,2).value
                tmpdict['CountryName'] = table.cell(n,3).value
                tmpdict['Rspect'] = table.cell(n,4).value
                self.caseList.append(tmpdict)
        except Exception, e:
            raise
        finally:
            pass
        return self.caseList

    """
    傳入：請求指定字段結果，是否通過，響應時間
    返回：
    """
    def writeCaseResult(self,resultBody,isSuccess,respTime,\
        excelFile,theRow,theCol=5):
        writeExcel = openpyxl.load_workbook(excelFile)      #加載Excel，后續寫操作
        try:
            wtable = writeExcel.get_sheet_by_name('Sheet1') #獲取名為Sheet1的sheet
            wtable.cell(row=theRow+1,column=theCol+1).value = resultBody #填寫實際值
            wtable.cell(row=theRow+1,column=theCol+2).value = isSuccess  #填寫是否通過
            wtable.cell(row=theRow+1,column=theCol+3).value = respTime   #填寫響應時間
            writeExcel.save(excelFile)
        except Exception, e:
            raise
        finally:
            pass


class GetWeather(object):
    """獲取天氣的http請求"""
    def __init__(self, serviceUrl,requestBody,headers):
        self.serviceUrl = serviceUrl
        self.requestBody = requestBody
        self.headers = headers
        self.requestResult = {}

    """
    傳入：請求地址，請求體，請求頭
    返回：返回{}，包含響應時間和請求結果的鍵值對
    """
    def getWeath(self,serviceUrl,requestBody,headers):
        timebefore = time.time()                           #獲取請求開始的時間，不太嚴禁
        tmp = requests.post(serviceUrl,data=requestBody,\
            headers=headers)
        timeend = time.time()                              #獲取請求結束的時間
        tmptext = tmp.text
        self.requestResult['text'] = tmptext                #記錄響應回來的內容
        self.requestResult['time'] = round(timeend - timebefore,2) #計算響應時間
        return self.requestResult

class XmlReader:
    """操作XML文件"""
    def __init__(self,testFile,testFilePath=''):
        self.fromXml = testFile
        self.xmlFilePath = testFilePath
        self.resultList = []

    def writeXmlData(self,resultBody,testFile,testFilePath=''):
        tmpXmlFile = codecs.open(testFile,'w','utf-16')        #新建xml文件
        tmpLogFile = codecs.open(testFile+'.log','w','utf-16') #新建log文件

        tmp1 = re.compile(r'\<.*?\>')                          #生成正則表達式:<*?>
        tmp2 = tmp1.sub('',resultBody['text'])                 #替換相應結果中的<*?>
        html_parser = HTMLParser.HTMLParser()
        xmlText = html_parser.unescape(tmp2)                   #轉換html編碼

        try:
            tmpXmlFile.writelines(xmlText.strip())             #去除空行並寫入xml
            tmpLogFile.writelines('time: '+\
                str(resultBody['time'])+'\r\n')                #把響應時間寫入log
            tmpLogFile.writelines('text: '+resultBody['text'].strip())#把請求回來的文本寫入log
        except Exception, e:
            raise
        finally:
            tmpXmlFile.close()
            tmpLogFile.close()

    """返回一個list"""
    def readXmlData(self,testFile,testFilePath=''):
        tmpXmlFile = minidom.parse(testFile)
        root = tmpXmlFile.documentElement
        tmpValue = root.getElementsByTagName('Status')[0].\
        childNodes[0].data
        return tmpValue                   #獲取特定字段並返回結果，此處選取Status


if __name__ == '__main__':

    requesturl = 'http://www.webservicex.net/globalweather.asmx/GetWeather'
    requestHeadrs = {"Content-Type":"application/x-www-form-urlencoded"}
    fileName = u'用例內容.xlsx'

    ed = OptionExcelData(fileName)  
    testCaseList = ed.getCaseList(ed.excelFile)

    for caseDict in testCaseList:
        caseId = caseDict['id']
        cityName = caseDict['CityName']
        countryName = caseDict['CountryName']
        rspect = caseDict['Rspect']
        requestBody = 'CityName='+cityName+'&CountryName='+countryName

        getWeather = GetWeather(requesturl,requestBody,requestHeadrs)
        #獲取請求結果
        tmpString = getWeather.getWeath(getWeather.serviceUrl,\
            getWeather.requestBody,getWeather.headers)

        xd = XmlReader(str(caseId) + '.xml')
        #把請求內容寫入xml和log
        xd.writeXmlData(tmpString,xd.fromXml)
        response = xd.readXmlData(str(caseId) + '.xml')
        respTime = tmpString['time']
        if response == rspect:
            theResult = 'Pass'
        else:
            theResult = 'False'
      ed.writeCaseResult(response,theResult,respTime,fileName,caseId)

原文地址https://blog.csdn.net/cakushin7433/article/details/52929042