ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

python打造更适合测试的HTTP协议类

2021-07-01 16:05:38  阅读:160  来源: 互联网

标签:__ HTTP python self init timeout kwargs 打造 def


#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2021/6/29 1:39 下午
# @Author  : CrissChan
# @Site    : https://blog.csdn.net/crisschan
# @File    : __init__.py.py
# @Software: 创建一个更适合测试使用的http协议类


from requests_toolbelt import sessions
from requests.adapters import HTTPAdapter
MAX_RETRIES=5
BASE_URL="https://blog.csdn.net/"
DEFAULT_TIMEOUT=10

class ReTriesHTTPAdapter(HTTPAdapter):
    """设置HTTP失败重跑次数类"""
    def __init__(self,*arg,**kwargs):
        """
        设置HTTP失败重跑次数类,重跑次数MAX_RETRIES在CONFIG里面设置
        """
        if "max_retries" not in kwargs:
            kwargs["max_retries"]= MAX_RETRIES
        super().__init__(*arg,**kwargs)



class TimeoutHTTPAdapter(HTTPAdapter):
    """HTTP访问请求等待时间,等同于思考时间"""
    def __init__(self, *args, **kwargs):
        self.timeout = DEFAULT_TIMEOUT
        if "timeout" in kwargs:
            self.timeout = kwargs["timeout"]
            del kwargs["timeout"]
        super().__init__(*args, **kwargs)


class Httper(sessions.BaseUrlSession):
    def __init__(self,base_url=BASE_URL):

        self.base_url=base_url
        super(Httper, self).__init__(base_url=base_url)
        self.__retries()
        self.__timeout()
    def __retries(self):
        """
        设置失败重跑次数,如果本次http请求失败了,重试次
        """
        readapter = ReTriesHTTPAdapter()
        return super().mount("http://",readapter)
    def __timeout(self):
        """
        设置请求超市时间,和测试中的thinktime一样
        """
        tadapter = TimeoutHTTPAdapter()
        return super(Httper, self).mount("http://",tadapter)
    def get_cookies(self):
        return self.cookies

标签:__,HTTP,python,self,init,timeout,kwargs,打造,def
来源: https://blog.csdn.net/chenlei_525/article/details/118388199

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有