python對於requests的封裝方法詳解

今天小編就爲大家分享一篇python對於requests的封裝方法詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

由於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的封裝方法詳解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持神馬文庫。