Function create_session

malwoverview/utils/session.py:26–48

(headers=None)

def create_session(headers=None):
    session = requests.Session()

    retry_strategy = Retry(
        total=3,
        backoff_factor=1,
        status_forcelist=[500, 502, 503, 504],
        allowed_methods=["HEAD", "GET", "POST", "PUT", "DELETE", "OPTIONS"],
    )

    adapter = HTTPAdapter(max_retries=retry_strategy)
    session.mount("https://", adapter)
    session.mount("http://", adapter)

    if cv.proxy:
        session.proxies = {"http": cv.proxy, "https": cv.proxy}

    session.hooks['response'].append(_rate_limit_hook)

    if headers:
        session.headers.update(headers)

    return session