Added retry decorator; Bumped version

pyproject.toml

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

snek_sploit/lib/context.py

@@ -4,6 +4,7 @@
 from typing import Union, List, Dict
 
 from snek_sploit.util import constants, exceptions
+from snek_sploit.util.retry import retry
 
 
 ResponseDict = Dict[Union[int, str, bytes], Union[int, str, bytes, list, dict, bool]]
@@ -63,6 +64,7 @@ def _create_arguments(self, call_arguments: list, use_token: bool) -> list:
 
         return arguments
 
+    @retry(tries=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

@@ -0,0 +1,27 @@
+from functools import wraps, partial
+
+
+def retry(func=None, *, tries: int = 1, on_errors: tuple = None):
+    """
+    Retry a function if an error occurs.
+    :param func: Original function
+    :param tries: Number of times to retry
+    :param on_errors: On what errors to retry
+    :return: Wrapper
+    """
+    if func is None:
+        return partial(retry, tries=tries, on_errors=on_errors)
+
+    if not on_errors:
+        on_errors = (Exception,)
+
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        for i in range(tries + 1):
+            try:
+                return func(*args, **kwargs)
+            except on_errors as ex:
+                if i == tries:
+                    raise ex
+
+    return wrapper