All network requests use timeout

This commit is contained in:
norohind 2022-10-20 23:43:46 +03:00
parent 5c51fd2138
commit ee77549ce3
Signed by: norohind
GPG Key ID: 01C3BECC26FB59E1
2 changed files with 6 additions and 1 deletions

View File

@ -34,7 +34,7 @@ class BearerManager:
def _request(self, _endpoint: Endpoints) -> requests.Response:
endpoint = self.base_address + _endpoint.value
return self.session.get(url=endpoint, headers={'auth': self.demb_capi_auth})
return self.session.get(url=endpoint, headers={'auth': self.demb_capi_auth}, timeout=10)
class BearerManagerException(Exception):

View File

@ -85,12 +85,17 @@ def request(url: str, method: str = 'get', **kwargs) -> requests.Response:
method=method,
url=url,
headers={'Authorization': f'Bearer {bearer_manager.get_random_bearer()}'},
timeout=10,
**kwargs
)
logger.debug(f'Request complete, code {proxiedFapiRequest.status_code!r}, len '
f'{len(proxiedFapiRequest.content)}')
except requests.exceptions.Timeout as e:
logger.opt(exception=e).error(f'Exception during request {method} {url!r} {kwargs!r}')
continue
except requests.exceptions.ConnectionError as e:
logger.error(f'Proxy {proxy.url} is invalid: {str(e.__class__.__name__)}')
continue