Dynamically pulls the info for Vec<AccountId> from the metadata

Author: thewhaleking

async_substrate_interface/async_substrate.py

@@ -717,6 +717,7 @@ def __init__(
         self.metadata_version_hex = "0x0f000000"  # v15
         self.reload_type_registry()
         self._initializing = False
+        self.registry_type_map = {}
 
     async def __aenter__(self):
         await self.initialize()
@@ -806,6 +807,15 @@ async def load_registry(self):
             metadata_option_bytes
         )
         self.registry = PortableRegistry.from_metadata_v15(self.metadata_v15)
+        registry_type_map = {}
+        for i in json.loads(self.registry.registry)["types"]:
+            for variants in (
+                i.get("type").get("def", {}).get("variant", {}).get("variants", [{}])
+            ):
+                for field in variants.get("fields", [{}]):
+                    if field.get("type") and field.get("typeName"):
+                        field["typeName"] = registry_type_map[field["type"]]
+        self.registry_type_map = registry_type_map
 
     async def _load_registry_at_block(self, block_hash: str) -> MetadataV15:
         # Should be called for any block that fails decoding.

async_substrate_interface/sync_substrate.py

@@ -516,6 +516,7 @@ def __init__(
         self.metadata_version_hex = "0x0f000000"  # v15
         self.reload_type_registry()
         self.ws = self.connect(init=True)
+        self.registry_type_map = {}
         if not _mock:
             self.initialize()
 
@@ -610,6 +611,15 @@ def load_registry(self):
             metadata_option_bytes
         )
         self.registry = PortableRegistry.from_metadata_v15(self.metadata_v15)
+        registry_type_map = {}
+        for i in json.loads(self.registry.registry)["types"]:
+            for variants in (
+                i.get("type").get("def", {}).get("variant", {}).get("variants", [{}])
+            ):
+                for field in variants.get("fields", [{}]):
+                    if field.get("type") and field.get("typeName"):
+                        field["typeName"] = registry_type_map[field["type"]]
+        self.registry_type_map = registry_type_map
 
     def _load_registry_at_block(self, block_hash: str) -> MetadataV15:
         # Should be called for any block that fails decoding.

async_substrate_interface/types.py

@@ -345,6 +345,7 @@ class SubstrateMixin(ABC):
     type_registry: Optional[dict]
     ss58_format: Optional[int]
     ws_max_size = 2**32
+    registry_type_map: dict[str, int]
 
     @property
     def chain(self):
@@ -722,12 +723,19 @@ def _encode_scale(self, type_string, value: Any) -> bytes:
         if value is None:
             result = b"\x00"
         else:
+            try:
+                vec_acct_id = (
+                    f"scale_info::{self.registry_type_map['Vec<T::AccountId>']}"
+                )
+            except KeyError:
+                vec_acct_id = "scale_info::152"
+
             if type_string == "scale_info::0":  # Is an AccountId
                 # encode string into AccountId
                 ## AccountId is a composite type with one, unnamed field
                 return bytes.fromhex(ss58_decode(value, SS58_FORMAT))
 
-            elif type_string == "scale_info::151":  # Vec<AccountId>
+            elif type_string == vec_acct_id:  # Vec<AccountId>
                 if not isinstance(value, (list, tuple)):
                     value = [value]