Clean up debug

Author: thewhaleking

async_substrate_interface/async_substrate.py

@@ -4183,12 +4183,11 @@ async def close(self):
         """
         try:
             await self.runtime_cache.dump_to_disk(self.url)
-            print(4276)
             await self.ws.shutdown()
         except AttributeError:
             pass
-        for db in AsyncSqliteDB._instances.values():
-            await db.close()
+        db = AsyncSqliteDB(self.url)
+        await db.close()
 
     @async_sql_lru_cache(maxsize=SUBSTRATE_CACHE_METHOD_SIZE)
     async def get_parent_block_hash(self, block_hash):

async_substrate_interface/utils/cache.py

@@ -41,7 +41,6 @@ def __new__(cls, chain_endpoint: str):
     async def close(self):
         async with self._lock:
             if self._db:
-                print(44)
                 await self._db.close()
 
     async def _create_if_not_exists(self, chain: str, table_name: str):
@@ -164,22 +163,21 @@ async def dump_runtime_cache(
                 local_chain = await self._create_if_not_exists(chain, table)
                 if local_chain:
                     return None
-
+                serialized_mapping = {}
                 for key, value in mapping.items():
                     if not isinstance(value, (str, int)):
-                        serialized_runtime = pickle.dumps(value.serialize())
+                        serialized_value = pickle.dumps(value.serialize())
                     else:
-                        serialized_runtime = pickle.dumps(value)
-
-                    await self._db.execute(
-                        f"INSERT OR REPLACE INTO {table} (key, value, chain) VALUES (?,?,?)",
-                        (key, serialized_runtime, chain),
-                    )
-
-                # await self._db.executemany(
-                #     f"INSERT OR REPLACE INTO {table} (key, value, chain) VALUES (?,?,?)",
-                #     [(key, pickle.dumps(runtime.serialize()), chain) for key, runtime in mapping.items()],
-                # )
+                        serialized_value = pickle.dumps(value)
+                    serialized_mapping[key] = serialized_value
+
+                await self._db.executemany(
+                    f"INSERT OR REPLACE INTO {table} (key, value, chain) VALUES (?,?,?)",
+                    [
+                        (key, serialized_value_, chain)
+                        for key, serialized_value_ in serialized_mapping.items()
+                    ],
+                )
 
             await self._db.commit()
 

pyproject.toml

@@ -8,7 +8,7 @@ keywords = ["substrate", "development", "bittensor"]
 
 dependencies = [
   "wheel",
-  "aiosqlite>=0.21.0,<1.0.0"
+  "aiosqlite>=0.21.0,<1.0.0",
   "bt-decode==v0.8.0",
   "scalecodec~=1.2.11",
   "websockets>=14.1",