Fix contract.call (#132)

Author: adam-maj

pyproject.toml

@@ -6,7 +6,7 @@ packages = [
   {include = "thirdweb"},
 ]
 readme = "README.md"
-version = "3.0.2"
+version = "3.0.3"
 
 [tool.poetry.dependencies]
 python = ">=3.7.1"

thirdweb/contracts/custom.py

@@ -102,7 +102,51 @@ def __init__(
         self.erc1155 = self._detect_erc_1155()
 
     def call(self, fn: str, *args) -> Any:
-        return self._contract_wrapper.call(fn, *args)
+        func = cast(ContractFunction, getattr(self.functions, fn, None))
+        if func is None:
+            raise Exception(
+                f"Function {fn} not found on contract {self.get_address()}. "
+                + "Check your dashboard for the list of available functions."
+            )
+
+        # We need this to set params properly on func + throws good errors
+        func.args = args
+        func.kwargs = {}
+        func._set_function_info()
+
+        if len(func.abi["inputs"]) != len(args):
+            signature = (
+                "("
+                + ", ".join(
+                    [(i["name"] + ": " + i["type"]) for i in func.abi["inputs"]]
+                )
+                + ")"
+            )
+            raise Exception(
+                f"Function {fn} expects {len(func.arguments)} arguments, "
+                f"but {len(args)} were provided.\nExpected function signature: {signature}"
+            )
+
+        if func.abi["stateMutability"] == "view" or func.abi["stateMutability"] == "pure":
+            return func(*args).call()
+        else:
+            provider = self._contract_wrapper.get_provider()
+            signer = self._contract_wrapper.get_signer()
+
+            if signer is None:
+                raise NoSignerException
+
+            nonce = provider.eth.get_transaction_count(signer.address)  # type: ignore
+
+            tx = func(*args).buildTransaction(
+                TxParams(gas_price=provider.eth.gas_price).as_dict()
+            )
+            tx["nonce"] = nonce
+
+            signed_tx = signer.sign_transaction(tx)  # type: ignore
+            tx_hash = provider.eth.send_raw_transaction(signed_tx.rawTransaction)
+
+            return provider.eth.wait_for_transaction_receipt(tx_hash)
 
     """
     INTERNAL FUNCTIONS