Merge pull request #1707 from S1RANN/master

raise other exceptions in antiflood function
eternnoir · Sep 13, 2022 · 8d9dfcf · 8d9dfcf
2 parents a1c77db + a06b4a1
commit 8d9dfcf
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/telebot/util.py b/telebot/util.py
@@ -647,15 +647,15 @@ def antiflood(function: Callable, *args, **kwargs):
     """
     from telebot.apihelper import ApiTelegramException
     from time import sleep
-    msg = None
+
     try:
-        msg = function(*args, **kwargs)
+        return function(*args, **kwargs)
     except ApiTelegramException as ex:
         if ex.error_code == 429:
             sleep(ex.result_json['parameters']['retry_after'])
-            msg = function(*args, **kwargs)
-    finally:
-        return msg
+            return function(*args, **kwargs)
+        else:
+            raise
 
 
 def parse_web_app_data(token: str, raw_init_data: str):