python对于requests的封装方法详解-创新互联
由于requests是http类接口的核心,因此封装前考虑问题比较多:
创新互联成立与2013年,是专业互联网技术服务公司,拥有项目成都做网站、成都网站制作、成都外贸网站建设网站策划,项目实施与项目整合能力。我们以让每一个梦想脱颖而出为使命,1280元淳安做网站,已为上家服务,为淳安各地企业和个人服务,联系电话:189820811081. 对多种接口类型的支持;
2. 连接异常时能够重连;
3. 并发处理的选择;
4. 使用方便,容易维护;
当前并未全部实现,后期会不断完善。重点提一下并发处理的选择:python的并发处理机制由于存在GIL的原因,实现起来并不是很理想,综合考虑多进程、多线程、协程,在不考虑大并发性能测试的前提下使用了多线程-线程池的形式实现。使用的是
concurrent.futures模块。当前仅方便支持webservice接口。
# -*- coding:utf-8 -*- import requests from concurrent.futures import ThreadPoolExecutor from Tools.Config import Config # 配置文件读取 from Tools.Log import Log # 日志管理 from Tools.tools import decoLOG # 日志装饰 ''' 功能: Requests类 使用方法: 作者: 郭可昌 作成时间: 20180224 更新内容: 更新时间: ''' class Requests(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's data @decoLOG def post(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() except Exception as e: self.errorMsg = str(e) Log().logger.error("HTTP请求异常,异常信息:%s" % self.errorMsg) return response # 复数请求并发处理,采用线程池的形式,用例数>线程池的容量:线程池的容量为并发数,否则,用例数为并发数 # dicDatas: {用例编号:用例数据} @decoLOG def req_all(self, dicDatas, iThreadNum=5): if len(dict(dicDatas)) < 1: Log().logger.error("没有测试对象,请确认后再尝试。。。") return self.responses.clear() # 请求用例集合转换(用例编号,用例数据) seed = [i for i in dicDatas.items()] self.responses.clear() # 线程池并发执行,iThreadNum为并发数 with ThreadPoolExecutor(iThreadNum) as executor: executor.map(self.req_single,seed) # 返回所有请求的响应信息({用例编号:响应数据}),http连接异常:对应None return self.responses # 用于单用例提交,http连接失败可以重新连接,大重新连接数可以动态设定 def req_single(self, listData, reqType="post", iLoop=1): response = None # 如果达到大重连次数,连接后提交结束 if iLoop == self.iRetryNum: if reqType == "post": try: response = requests.post(self.URL, data=listData[1].encode('utf-8'), headers=self.header, timeout=self.timeout) response.raise_for_status() except Exception as e: # 异常信息保存只在大连接次数时进行,未达到大连接次数,异常信息为空 self.resErr[listData[0]] = str(e) Log().logger.error("HTTP请求异常,异常信息:%s【%d】" % (str(e), iLoop)) self.responses[listData[0]] = response else: # for future: other request method expand pass # 未达到大连接数,如果出现异常,则重新连接尝试 else: if reqType == "post": try: response = requests.post(self.URL, data=listData[1].encode('utf-8'), headers=self.header, timeout=self.timeout) response.raise_for_status() except Exception as e: Log().logger.error("HTTP请求异常,异常信息:%s【%d】" % (str(e), iLoop)) # 重连次数递增 iLoop += 1 # 进行重新连接 self.req_single(listData, reqType, iLoop) # 当前连接终止 return None self.responses[listData[0]] = response else: # for future: other request method expand pass # 设定SoapAction, 快捷完成webservice接口header设定 def setSoapAction(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的封装方法详解-创新互联
标题来源:http://scpingwu.com/article/hssep.html