Don't cache local chain

Author: thewhaleking

async_substrate_interface/utils/cache.py

@@ -4,9 +4,10 @@
 import sqlite3
 import asyncstdlib as a
 
+USE_CACHE = True if os.getenv("NO_CACHE") != "1" else False
 CACHE_LOCATION = (
     os.path.expanduser("~/.cache/async-substrate_interface")
-    if os.getenv("NO_CACHE") != "1"
+    if USE_CACHE
     else ":memory:"
 )
 
@@ -18,6 +19,10 @@ def _get_table_name(func):
     return func.__qualname__.replace(".", "_")
 
 
+def _check_if_local(chain: str) -> bool:
+    return any([x in chain for x in ["127.0.0.1", "localhost", "0.0.0.0"]])
+
+
 def _create_table(c, conn, table_name):
     c.execute(
         f"CREATE TABLE IF NOT EXISTS {table_name} (key BLOB PRIMARY KEY, value BLOB, chain TEXT)"
@@ -61,14 +66,16 @@ def inner(self, *args, **kwargs):
         c = conn.cursor()
         key = pickle.dumps((args, kwargs))
         chain = self.url
-
-        result = _retrieve_from_cache(c, table_name, key, chain)
-        if result is not None:
-            return result
+        if not (local_chain := _check_if_local(chain)) or not USE_CACHE:
+            result = _retrieve_from_cache(c, table_name, key, chain)
+            if result is not None:
+                return result
 
         # If not in DB, call func and store in DB
         result = func(self, *args, **kwargs)
-        _insert_into_cache(c, conn, table_name, key, result, chain)
+
+        if not local_chain or not USE_CACHE:
+            _insert_into_cache(c, conn, table_name, key, result, chain)
 
         return result
 
@@ -87,13 +94,15 @@ async def inner(self, *args, **kwargs):
         key = pickle.dumps((args, kwargs))
         chain = self.url
 
-        result = _retrieve_from_cache(c, table_name, key, chain)
-        if result is not None:
-            return result
+        if not (local_chain := _check_if_local(chain)) or not USE_CACHE:
+            result = _retrieve_from_cache(c, table_name, key, chain)
+            if result is not None:
+                return result
 
         # If not in DB, call func and store in DB
         result = await func(self, *args, **kwargs)
-        _insert_into_cache(c, conn, table_name, key, result, chain)
+        if not local_chain or not USE_CACHE:
+            _insert_into_cache(c, conn, table_name, key, result, chain)
 
         return result