# coding: utf-8

import requests
from requests.adapters import HTTPAdapter
from urllib3.util import Retry


class RetryRequests(object):

    def __init__(self, connect_timeout, read_timeout, max_retry, status_forcelist=None, session=None):
        self.session = session or requests.Session()
        retry = Retry(
            total=max_retry,  #最大重試多少次
            backoff_factor=0.1,  # 每次重試時(shí)間間隔（0.1， 0.2， 0.4）
            status_forcelist=status_forcelist,    # 返回哪些狀態(tài)碼時(shí)進(jìn)行重試
        )
        adapter = HTTPAdapter(max_retries=retry)
        self.session.mount('http://', adapter)
        self.session.mount('https://', adapter)
        self.timeout = (connect_timeout, read_timeout)

    def get(self, url, timeout=None, **kwargs):
        if timeout is None:
            timeout = self.timeout
        return self.session.get(url, timeout=timeout, **kwargs)

    def post(self, url, timeout=None, data=None, json_data=None, **kwargs):
        if timeout is None:
            timeout = self.timeout
        return self.session.post(url, timeout=timeout, data=data, json=json_data, **kwargs)


common_request_service = RetryRequests(10, 30, 3, [500, 502, 503, 504])


if __name__ == "__main__":
    import time
    rr = common_request_service
    now = time.time()
    try:
        print rr.get("http://localhost:8888").text
    except Exception as e:
        print e
    print time.time() - now