Add archive node for RetrySyncSubstrate

Author: thewhaleking

async_substrate_interface/substrate_addons.py

@@ -117,13 +117,17 @@ def __init__(
         max_retries: int = 5,
         retry_timeout: float = 60.0,
         _mock: bool = False,
+        archive_nodes: Optional[list[str]] = None,
     ):
         fallback_chains = fallback_chains or []
         self.fallback_chains = (
             iter(fallback_chains)
             if not retry_forever
             else cycle(fallback_chains + [url])
         )
+        self.archive_nodes = (
+            iter(archive_nodes) if not retry_forever else cycle(archive_nodes)
+        )
         self.use_remote_preset = use_remote_preset
         self.chain_name = chain_name
         self._mock = _mock
@@ -174,20 +178,32 @@ def _retry(self, method, *args, **kwargs):
             EOFError,
             ConnectionClosed,
             TimeoutError,
+            socket.gaierror,
+            StateDiscardedError,
         ) as e:
+            use_archive = isinstance(e, StateDiscardedError)
             try:
-                self._reinstantiate_substrate(e)
+                self._reinstantiate_substrate(e, use_archive=use_archive)
                 return method_(*args, **kwargs)
             except StopIteration:
                 logger.error(
                     f"Max retries exceeded with {self.url}. No more fallback chains."
                 )
                 raise MaxRetriesExceeded
 
-    def _reinstantiate_substrate(self, e: Optional[Exception] = None) -> None:
-        next_network = next(self.fallback_chains)
+    def _reinstantiate_substrate(
+        self, e: Optional[Exception] = None, use_archive: bool = False
+    ) -> None:
+        if use_archive:
+            bh = getattr(e, "block_hash", "Unknown Block Hash")
+            logger.info(
+                f"Attempt made to {bh} failed for state discarded. Attempting to switch to archive node."
+            )
+            next_network = next(self.archive_nodes)
+        else:
+            next_network = next(self.fallback_chains)
         self.ws.close()
-        if e.__class__ == MaxRetriesExceeded:
+        if isinstance(e, MaxRetriesExceeded):
             logger.error(
                 f"Max retries exceeded with {self.url}. Retrying with {next_network}."
             )
@@ -287,7 +303,7 @@ async def _reinstantiate_substrate(
             next_network = next(self.archive_nodes)
         else:
             next_network = next(self.fallback_chains)
-        if e.__class__ == MaxRetriesExceeded:
+        if isinstance(e, MaxRetriesExceeded):
             logger.error(
                 f"Max retries exceeded with {self.url}. Retrying with {next_network}."
             )