Merge pull request #61 from madeddie/Add-global-timeout

Add global http timeout to the session
madeddie · May 28, 2024 · f319495 · f319495
2 parents 87451bf + f30cc3a
commit f319495
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tyora/session.py b/tyora/session.py
@@ -10,6 +10,7 @@
 
 from .utils import find_link, parse_form
 
+HTTP_TIMEOUT = int(os.getenv("HTTP_TIMEOUT", 10))
 logger = logging.getLogger(__name__)
 
 try:
@@ -33,6 +34,10 @@ def __init__(
             {"User-Agent": user_agent(os.path.basename(sys.argv[0]), __version__)}
         )
 
+    def request(self, *args, **kwargs):
+        kwargs.setdefault("timeout", HTTP_TIMEOUT)
+        return super(MoocfiCsesSession, self).request(*args, **kwargs)
+
     @property
     def is_logged_in(self) -> bool:
         res = self.get(urljoin(self.base_url, "list"))