1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
import sys
from requests import Session
from requests.adapters import HTTPAdapter
from requests.exceptions import ConnectionError, RequestException, BaseHTTPError, SSLError
class RetrySession(Session):
def __init__(self, max_retries):
super(RetrySession, self).__init__()
if max_retries is not None:
self.mount('https://', HTTPAdapter(max_retries=max_retries))
self.mount('http://', HTTPAdapter(max_retries=max_retries))
def new_requests_session(agent=None, max_retries=None):
if agent is None:
agent = "devpi"
else:
agent = "devpi-%s/%s" % agent
agent += " (py%s; %s)" % (sys.version.split()[0], sys.platform)
session = RetrySession(max_retries)
session.headers["user-agent"] = agent
session.ConnectionError = ConnectionError
session.RequestException = RequestException
session.Errors = (RequestException, BaseHTTPError)
session.SSLError = SSLError
return session
|
