Fixed error handling

Author: Amele9

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"
 ]

whatsapp_chatbot_python/bot.py

@@ -79,9 +79,13 @@ def run_forever(self) -> Optional[NoReturn]:
                 self.api.receiving.deleteNotification(response["receiptId"])
             except KeyboardInterrupt:
                 break
-            except Exception as e:
-                self.logger.log(logging.ERROR, f"An unexpected error occurred: {e}")
-                time.sleep(5)
+            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"
@@ -163,4 +167,14 @@ class GreenAPIBot(Bot):
     pass
 
 
-__all__ = ["Bot", "GreenAPI", "GreenAPIBot", "GreenAPIError"]
+class GreenAPIBotError(Exception):
+    pass
+
+
+__all__ = [
+    "Bot",
+    "GreenAPI",
+    "GreenAPIBot",
+    "GreenAPIError",
+    "GreenAPIBotError"
+]