ED/EDMarketConnector
1
0
Fork 0
mirror of https://github.com/EDCD/EDMarketConnector.git synced 2025-04-04 19:40:02 +03:00
Code Issues Projects Releases Wiki Activity

[2051] Don't Overwrite Every Session

Browse Source 
Whoops. Only fetch a new session if it's None.

Co-Authored-By: Phoebe <40956085+C1701D@users.noreply.github.com>
This commit is contained in:
David Sangrey 2023-11-17 11:51:41 -05:00
parent ebd1813612
commit b41dfaa594
No known key found for this signature in database
GPG Key ID: 3AEADBB0186884BC
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
timeout_session.py
View File

@ -7,8 +7,7 @@ See LICENSE file.
"""
from __future__ import annotations
import requests
from requests import Session
from requests import Session, Response
from requests.adapters import HTTPAdapter
from config import user_agent
@ -25,7 +24,7 @@ class TimeoutAdapter(HTTPAdapter):
super().__init__(*args, **kwargs)
def send(self, *args, **kwargs) -> requests.Response:
def send(self, *args, **kwargs) -> Response:
"""Send, but with a timeout always set."""
if kwargs["timeout"] is None:
kwargs["timeout"] = self.default_timeout
@ -33,9 +32,7 @@ class TimeoutAdapter(HTTPAdapter):
return super().send(*args, **kwargs)
def new_session(
timeout: int = REQUEST_TIMEOUT, session: requests.Session | None = None
) -> requests.Session:
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.
@ -43,9 +40,11 @@ def new_session(
:param session: the Session object to attach the Adapter to, defaults to a new session
:return: The created Session
"""
with Session() as session:
session.headers['User-Agent'] = user_agent
adapter = TimeoutAdapter(timeout)
session.mount("http://", adapter)
session.mount("https://", adapter)
return 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