diff --git a/packages/modules/common/req.py b/packages/modules/common/req.py index 00b7c31993..aaaa7f5f6b 100644 --- a/packages/modules/common/req.py +++ b/packages/modules/common/req.py @@ -5,8 +5,18 @@ log = logging.getLogger(__name__) -def get_http_session() -> Session: - session = Session() +class CustomSession(Session): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + self.default_timeout = 5 + + def request(self, method, url, *args, **kwargs): + kwargs.setdefault('timeout', self.default_timeout) + return super().request(method, url, *args, **kwargs) + + +def get_http_session() -> CustomSession: + session = CustomSession() session.hooks['response'].append(lambda r, *args, **kwargs: r.raise_for_status()) session.hooks['response'].append(lambda r, *args, **kwargs: log.debug("Get-Response: " + r.text)) return session