Merge pull request #96 from opentensor/feat/thewhaleking/new-exception-for-max-retries

Adds additional exception for catching

Author: thewhaleking

async_substrate_interface/async_substrate.py

@@ -38,6 +38,7 @@
     SubstrateRequestException,
     ExtrinsicNotFound,
     BlockNotFound,
+    MaxRetriesExceeded,
 )
 from async_substrate_interface.types import (
     ScaleObj,
@@ -1914,7 +1915,7 @@ async def _make_rpc_request(
                         logger.warning(
                             f"Timed out waiting for RPC requests {attempt} times. Exiting."
                         )
-                        raise SubstrateRequestException("Max retries reached.")
+                        raise MaxRetriesExceeded("Max retries reached.")
                     else:
                         self.ws.last_received = time.time()
                         await self.ws.connect(force=True)

async_substrate_interface/errors.py

@@ -8,6 +8,10 @@ class SubstrateRequestException(Exception):
     pass
 
 
+class MaxRetriesExceeded(SubstrateRequestException):
+    pass
+
+
 class StorageFunctionNotFound(ValueError):
     pass
 

async_substrate_interface/sync_substrate.py

@@ -20,6 +20,7 @@
     ExtrinsicNotFound,
     SubstrateRequestException,
     BlockNotFound,
+    MaxRetriesExceeded,
 )
 from async_substrate_interface.types import (
     SubstrateMixin,
@@ -1615,7 +1616,7 @@ def _make_rpc_request(
                     logger.warning(
                         f"Timed out waiting for RPC requests {attempt} times. Exiting."
                     )
-                    raise SubstrateRequestException("Max retries reached.")
+                    raise MaxRetriesExceeded("Max retries reached.")
                 else:
                     return self._make_rpc_request(
                         payloads,