mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-04 19:40:02 +03:00
b41dfaa594
Whoops. Only fetch a new session if it's None. Co-Authored-By: Phoebe <40956085+C1701D@users.noreply.github.com>
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
"""
|
|
timeout_session.py - requests session with timeout adapter.
|
|
|
|
Copyright (c) EDCD, All Rights Reserved
|
|
Licensed under the GNU General Public License.
|
|
See LICENSE file.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from requests import Session, Response
|
|
from requests.adapters import HTTPAdapter
|
|
from config import user_agent
|
|
|
|
REQUEST_TIMEOUT = 10 # reasonable timeout that all HTTP requests should use
|
|
|
|
|
|
class TimeoutAdapter(HTTPAdapter):
|
|
"""An HTTP Adapter that enforces an overridable default timeout on HTTP requests."""
|
|
|
|
def __init__(self, timeout: int, *args, **kwargs):
|
|
self.default_timeout = timeout
|
|
if kwargs.get("timeout") is not None:
|
|
del kwargs["timeout"]
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def send(self, *args, **kwargs) -> Response:
|
|
"""Send, but with a timeout always set."""
|
|
if kwargs["timeout"] is None:
|
|
kwargs["timeout"] = self.default_timeout
|
|
|
|
return super().send(*args, **kwargs)
|
|
|
|
|
|
def new_session(timeout: int = REQUEST_TIMEOUT, session: Session | None = None) -> Session:
|
|
"""
|
|
Create a new requests.Session and override the default HTTPAdapter with a TimeoutAdapter.
|
|
|
|
:param timeout: the timeout to set the TimeoutAdapter to, defaults to REQUEST_TIMEOUT
|
|
:param session: the Session object to attach the Adapter to, defaults to a new session
|
|
:return: The created Session
|
|
"""
|
|
session = session or Session()
|
|
session.headers.setdefault("User-Agent", user_agent)
|
|
|
|
adapter = TimeoutAdapter(timeout)
|
|
for prefix in ("http://", "https://"):
|
|
session.mount(prefix, adapter)
|
|
|
|
return session
|