From 0acc0b6bb008f16ab27f5f78a280ceb70009c187 Mon Sep 17 00:00:00 2001 From: Benjamin Himes Date: Tue, 15 Apr 2025 17:55:31 +0200 Subject: [PATCH] Adds additional exception for catching --- async_substrate_interface/async_substrate.py | 3 ++- async_substrate_interface/errors.py | 4 ++++ async_substrate_interface/sync_substrate.py | 3 ++- 3 files changed, 8 insertions(+), 2 deletions(-) diff --git a/async_substrate_interface/async_substrate.py b/async_substrate_interface/async_substrate.py index e8a95aa..c40d628 100644 --- a/async_substrate_interface/async_substrate.py +++ b/async_substrate_interface/async_substrate.py @@ -39,6 +39,7 @@ SubstrateRequestException, ExtrinsicNotFound, BlockNotFound, + MaxRetriesExceeded, ) from async_substrate_interface.types import ( ScaleObj, @@ -1910,7 +1911,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) diff --git a/async_substrate_interface/errors.py b/async_substrate_interface/errors.py index 9de753b..98114fe 100644 --- a/async_substrate_interface/errors.py +++ b/async_substrate_interface/errors.py @@ -8,6 +8,10 @@ class SubstrateRequestException(Exception): pass +class MaxRetriesExceeded(SubstrateRequestException): + pass + + class StorageFunctionNotFound(ValueError): pass diff --git a/async_substrate_interface/sync_substrate.py b/async_substrate_interface/sync_substrate.py index f463f2f..8a3fa16 100644 --- a/async_substrate_interface/sync_substrate.py +++ b/async_substrate_interface/sync_substrate.py @@ -21,6 +21,7 @@ ExtrinsicNotFound, SubstrateRequestException, BlockNotFound, + MaxRetriesExceeded, ) from async_substrate_interface.types import ( SubstrateMixin, @@ -1611,7 +1612,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,