python对于requests的封装方法详解
由于requests是http类接口的核心，因此封装前考虑问题比较多：
1.对多种接口类型的支持；
2.连接异常时能够重连；
3.并发处理的选择；
4.使用方便，容易维护；
当前并未全部实现，后期会不断完善。重点提一下并发处理的选择：python的并发处理机制由于存在GIL的原因，实现起来并不是很理想，综合考虑多进程、多线程、协程，在不考虑大并发性能测试的前提下使用了多线程-线程池的形式实现。使用的是
concurrent.futures模块。当前仅方便支持webservice接口。
#-*-coding:utf-8-*-
importrequests
fromconcurrent.futuresimportThreadPoolExecutor
fromTools.ConfigimportConfig#配置文件读取
fromTools.LogimportLog#日志管理
fromTools.toolsimportdecoLOG#日志装饰
'''
功能:Requests类
使用方法：
作者：郭可昌
作成时间：20180224
更新内容：
更新时间：
'''
classRequests(object):
def__init__(self):
self.session=requests.session()
self.header={}
#URL默认来源于配置文件，方便不同测试环境的切换，也可以动态设定
self.URL=Config().getURL()
#默认60s，可以动态设定
self.timeout=60
#http连接异常的场合，重新连接的次数，默认为3，可以动态设定
self.iRetryNum=3
self.errorMsg=""
#内容={用例编号：响应数据}
self.responses={}
#内容={用例编号：异常信息}
self.resErr={}
#原始post使用保留
#bodyData:request'sdata
@decoLOG
defpost(self,bodyData):
response=None
self.errorMsg=""
try:
response=self.session.post(self.URL,data=bodyData.encode('utf-8'),headers=self.header,timeout=self.timeout)
response.raise_for_status()
exceptExceptionase:
self.errorMsg=str(e)
Log().logger.error("HTTP请求异常，异常信息：%s"%self.errorMsg)
returnresponse
#复数请求并发处理，采用线程池的形式，用例数>线程池的容量：线程池的容量为并发数，否则，用例数为并发数
#dicDatas:{用例编号：用例数据}
@decoLOG
defreq_all(self,dicDatas,iThreadNum=5):
iflen(dict(dicDatas))<1:
Log().logger.error("没有测试对象，请确认后再尝试。。。")
returnself.responses.clear()
#请求用例集合转换（用例编号，用例数据）
seed=[iforiindicDatas.items()]
self.responses.clear()
#线程池并发执行，iThreadNum为并发数
withThreadPoolExecutor(iThreadNum)asexecutor:
executor.map(self.req_single,seed)
#返回所有请求的响应信息（{用例编号：响应数据})，http连接异常：对应None
returnself.responses
#用于单用例提交，http连接失败可以重新连接，最大重新连接数可以动态设定
defreq_single(self,listData,reqType="post",iLoop=1):
response=None
#如果达到最大重连次数，连接后提交结束
ifiLoop==self.iRetryNum:
ifreqType=="post":
try:
response=requests.post(self.URL,data=listData[1].encode('utf-8'),headers=self.header,
timeout=self.timeout)
response.raise_for_status()
exceptExceptionase:
#异常信息保存只在最大连接次数时进行，未达到最大连接次数，异常信息为空
self.resErr[listData[0]]=str(e)
Log().logger.error("HTTP请求异常，异常信息：%s【%d】"%(str(e),iLoop))
self.responses[listData[0]]=response
else:
#forfuture：otherrequestmethodexpand
pass
#未达到最大连接数，如果出现异常，则重新连接尝试
else:
ifreqType=="post":
try:
response=requests.post(self.URL,data=listData[1].encode('utf-8'),headers=self.header,
timeout=self.timeout)
response.raise_for_status()
exceptExceptionase:
Log().logger.error("HTTP请求异常，异常信息：%s【%d】"%(str(e),iLoop))
#重连次数递增
iLoop+=1
#进行重新连接
self.req_single(listData,reqType,iLoop)
#当前连接终止
returnNone
self.responses[listData[0]]=response
else:
#forfuture：otherrequestmethodexpand
pass
#设定SoapAction,快捷完成webservice接口header设定
defsetSoapAction(self,soapAction):
self.header["SOAPAction"]=soapAction
self.header["Content-Type"]="text/xml;charset=UTF-8"
self.header["Connection"]="Keep-Alive"
self.header["User-Agent"]="InterfaceAutoTest-run"
以上这篇python对于requests的封装方法详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。