default timeout for requests (openWB#1778)

rleidner · Aug 12, 2024 · 88edf35 · 88edf35
1 parent 2d5a995
commit 88edf35
Showing 1 changed file with 12 additions and 2 deletions.
diff --git 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