Upgraded the retry wrapper to wait between retries

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "snek-sploit"
-version = "0.7.0"
+version = "0.7.1"
 description = "Python RPC client for Metasploit Framework"
 authors = [
     "Jiří Rája <[email protected]>"

snek_sploit/lib/context.py

@@ -64,7 +64,7 @@ def _create_arguments(self, call_arguments: list, use_token: bool) -> list:
 
         return arguments
 
-    @retry(tries=3, on_errors=(requests.RequestException,))
+    @retry(attempts=3, on_errors=(requests.RequestException,))
     def call(
         self, endpoint: str, arguments: list = None, use_token: bool = True, timeout: Union[float, tuple] = None
     ) -> RPCResponse:

snek_sploit/util/retry.py

@@ -1,27 +1,31 @@
 from functools import wraps, partial
+import time
 
 
-def retry(func=None, *, tries: int = 1, on_errors: tuple = None):
+def retry(func=None, *, attempts: int = 1, on_errors: tuple = None, wait_on_error: bool = True):
     """
     Retry a function if an error occurs.
     :param func: Original function
-    :param tries: Number of times to retry
+    :param attempts: Number of times to retry
     :param on_errors: On what errors to retry
+    :param wait_on_error: Whether to wait when an error occurs or not
     :return: Wrapper
     """
     if func is None:
-        return partial(retry, tries=tries, on_errors=on_errors)
+        return partial(retry, attempts=attempts, on_errors=on_errors, wait_on_error=wait_on_error)
 
     if not on_errors:
         on_errors = (Exception,)
 
     @wraps(func)
     def wrapper(*args, **kwargs):
-        for i in range(tries + 1):
+        for i in range(attempts):
             try:
                 return func(*args, **kwargs)
             except on_errors as ex:
-                if i == tries:
+                if i + 1 == attempts:
                     raise ex
+                if wait_on_error:
+                    time.sleep(min(i + 1 * 3, 30))
 
     return wrapper