当前位置: 移动技术网 > IT编程>脚本编程>Python > python对于requests的封装方法详解

python对于requests的封装方法详解

2019年01月07日  | 移动技术网IT编程  | 我要评论

保罗太子龙,牛津迷案,kitty猫手机壁纸

由于requests是http类接口的核心,因此封装前考虑问题比较多:

1. 对多种接口类型的支持;

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的封装方法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持移动技术网。

如对本文有疑问,请在下面进行留言讨论,广大热心网友会与你互动!! 点击进行留言回复

相关文章:

验证码:
移动技术网