Merge pull request #65 from green-api/sw-4280

SW-4280 Set up a timeout for requests
green-api · Sep 30, 2024 · 82b8d50 · 82b8d50
2 parents 1365e5b + 2c2b8ad
commit 82b8d50
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="whatsapp-api-client-python",
-    version="0.0.46",
+    version="0.0.47",
     description=(
         "This library helps you easily create"
         " a Python application with WhatsApp API."

diff --git a/whatsapp_api_client_python/API.py b/whatsapp_api_client_python/API.py
@@ -33,13 +33,19 @@ def __init__(
             debug_mode: bool = False,
             raise_errors: bool = False,
             host: str = "https://api.green-api.com",
-            media: str = "https://media.green-api.com"
+            media: str = "https://media.green-api.com",
+            host_timeout: float = 180, # sec per retry
+            media_timeout: float = 10800, # sec per retry
     ):
         self.host = host
         self.media = media
         self.debug_mode = debug_mode
         self.raise_errors = raise_errors
 
+        # Change default values in init() if required
+        self.host_timeout = host_timeout
+        self.media_timeout = media_timeout
+
         self.idInstance = idInstance
         self.apiTokenInstance = apiTokenInstance
 
@@ -75,11 +81,11 @@ def request(
         try:
             if not files:
                 response = self.session.request(
-                    method=method, url=url, json=payload
+                    method=method, url=url, json=payload, timeout=self.host_timeout
                 )
             else:
                 response = self.session.request(
-                    method=method, url=url, data=payload, files=files
+                    method=method, url=url, data=payload, files=files, timeout=self.media_timeout
                 )
         except Exception as error:
             error_message = f"Request was failed with error: {error}."