Added usage of decorator for contract instances autoupdate (#21)

dkg/providers/blockchain.py

@@ -117,7 +117,7 @@ def make_json_rpc_request(self, endpoint: str, args: dict[str, Any] = {}) -> Any
     def handle_updated_contract(func):
         @wraps(func)
         def wrapper(self, *args, **kwargs):
-            contract_name = kwargs.get("contract_name") or (args[0] if args else None)
+            contract_name = kwargs.get("contract") or (args[0] if args else None)
 
             try:
                 return func(self, *args, **kwargs)
@@ -133,6 +133,7 @@ def wrapper(self, *args, **kwargs):
 
         return wrapper
 
+    @handle_updated_contract
     def call_function(
         self,
         contract: str,
@@ -240,14 +241,10 @@ def _update_contract_instance(self, contract: str):
         self.contracts[contract] = self.w3.eth.contract(
             address=(
                 self.contracts["Hub"]
-                .functions.getContractAddress(
-                    contract if contract != "ERC20Token" else "Token"
-                )
-                .call()
+                .functions.getContractAddress(contract).call()
                 if not contract.endswith("AssetStorage")
                 else self.contracts["Hub"]
-                .functions.getAssetStorageAddress(contract)
-                .call()
+                .functions.getAssetStorageAddress(contract).call()
             ),
             abi=self.abi[contract],
         )

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dkg"
-version = "0.1.0-beta.3"
+version = "0.1.0-beta.4"
 description = "Python library for interacting with the OriginTrail Decentralized Knowledge Graph"
 authors = ["Uladzislau Hubar <hubar.uladzislau@gmail.com>"]
 license = "Apache-2.0"