Add Flood Wait Exception (#42)

Co-authored-by: Kirill <iam@python273.pw>

Author: VarnaX 279

telegraph/api.py

@@ -3,7 +3,7 @@
 
 import requests
 
-from .exceptions import TelegraphException
+from .exceptions import TelegraphException, RetryAfterError
 from .utils import html_to_nodes, nodes_to_html
 
 
@@ -34,7 +34,12 @@ def method(self, method, values=None, path=''):
         if response.get('ok'):
             return response['result']
 
-        raise TelegraphException(response.get('error'))
+        error = response.get('error')
+        if isinstance(error, str) and error.startswith('FLOOD_WAIT_'):
+            retry_after = int(error.rsplit('_', 1)[-1])
+            raise RetryAfterError(retry_after)
+        else:
+            raise TelegraphException(error)
 
 
 class Telegraph(object):

telegraph/exceptions.py

@@ -14,3 +14,11 @@ class NotAllowedTag(ParsingException):
 
 class InvalidHTML(ParsingException):
     pass
+
+
+class RetryAfterError(TelegraphException):
+
+    def __init__(self, retry_after: int):
+        self.retry_after = retry_after
+        super().__init__(f'Flood control exceeded. Retry in {retry_after} seconds')
+ 

telegraph/upload.py

@@ -2,7 +2,7 @@
 
 import requests
 
-from .exceptions import TelegraphException
+from .exceptions import TelegraphException, RetryAfterError
 
 
 def upload_file(f):
@@ -24,7 +24,11 @@ def upload_file(f):
         error = response.get('error')
 
     if error:
-        raise TelegraphException(error)
+        if isinstance(error, str) and error.startswith('FLOOD_WAIT_'):
+            retry_after = int(error.rsplit('_',1)[-1])
+            raise RetryAfterError(retry_after)
+        else:
+            raise TelegraphException(error)
 
     return [i['src'] for i in response]