From 88edf3507272d819ec371678afa93d70fba7a14c Mon Sep 17 00:00:00 2001 From: LKuemmel <76958050+LKuemmel@users.noreply.github.com> Date: Mon, 12 Aug 2024 15:02:41 +0200 Subject: [PATCH] default timeout for requests (#1778) --- packages/modules/common/req.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) 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