Merge pull request #28 from green-api/SW-2934

SW-2934
green-api · Feb 5, 2024 · d78a6d4 · d78a6d4
2 parents 49239b6 + 3b5ea88
commit d78a6d4
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/whatsapp_chatbot_python/__init__.py b/whatsapp_chatbot_python/__init__.py
@@ -1,4 +1,4 @@
-from .bot import Bot, GreenAPI, GreenAPIBot, GreenAPIError
+from .bot import Bot, GreenAPI, GreenAPIBot, GreenAPIError, GreenAPIBotError
 from .manager.handler import Notification
 from .manager.state import BaseStates
 
@@ -7,6 +7,7 @@
     "GreenAPI",
     "GreenAPIBot",
     "GreenAPIError",
+    "GreenAPIBotError",
     "Notification",
     "BaseStates"
 ]
diff --git a/whatsapp_chatbot_python/bot.py b/whatsapp_chatbot_python/bot.py
@@ -1,4 +1,5 @@
 import logging
+import time
 from typing import NoReturn, Optional
 
 from whatsapp_api_client_python.API import GreenAPI, GreenAPIError
@@ -78,6 +79,14 @@ def run_forever(self) -> Optional[NoReturn]:
                 self.api.receiving.deleteNotification(response["receiptId"])
             except KeyboardInterrupt:
                 break
+            except Exception as error:
+                if self.raise_errors:
+                    raise GreenAPIBotError(error)
+                self.logger.log(logging.ERROR, error)
+
+                time.sleep(5.0)
+
+                continue
 
         self.api.session.headers["Connection"] = "close"
 
@@ -158,4 +167,14 @@ class GreenAPIBot(Bot):
     pass
 
 
-__all__ = ["Bot", "GreenAPI", "GreenAPIBot", "GreenAPIError"]
+class GreenAPIBotError(Exception):
+    pass
+
+
+__all__ = [
+    "Bot",
+    "GreenAPI",
+    "GreenAPIBot",
+    "GreenAPIError",
+    "GreenAPIBotError"
+]