Moved max_size to mixin for reuse

Author: thewhaleking

async_substrate_interface/async_substrate.py

@@ -685,7 +685,7 @@ def __init__(
         self.ws = Websocket(
             url,
             options={
-                "max_size": 2**32,
+                "max_size": self.ws_max_size,
                 "write_limit": 2**16,
             },
         )

async_substrate_interface/sync_substrate.py

@@ -566,12 +566,12 @@ def name(self):
 
     def connect(self, init=False):
         if init is True:
-            return connect(self.chain_endpoint, max_size=2**32)
+            return connect(self.chain_endpoint, max_size=self.ws_max_size)
         else:
             if not self.ws.close_code:
                 return self.ws
             else:
-                return connect(self.chain_endpoint, max_size=2**32)
+                return connect(self.chain_endpoint, max_size=self.ws_max_size)
 
     def get_storage_item(self, module: str, storage_function: str):
         if not self._metadata:

async_substrate_interface/types.py

@@ -344,6 +344,7 @@ class SubstrateMixin(ABC):
     runtime_config: RuntimeConfigurationObject
     type_registry: Optional[dict]
     ss58_format: Optional[int]
+    ws_max_size = 2**32
 
     @property
     def chain(self):