update contract

Author: serderovsh

tronapi/base/contracts.py

@@ -6,15 +6,23 @@
 
 import functools
 
-from eth_utils import to_hex
 from hexbytes import HexBytes
 
+from eth_utils import (
+    to_hex,
+    function_abi_to_4byte_selector
+)
 from tronapi.base.abi import (
     filter_by_name,
     filter_by_encodability,
     filter_by_argument_count,
     get_fallback_func_abi,
-    abi_to_signature, get_abi_input_types, check_if_arguments_can_be_encoded, map_abi_data)
+    abi_to_signature,
+    get_abi_input_types,
+    check_if_arguments_can_be_encoded,
+    map_abi_data,
+    merge_args_and_kwargs
+)
 
 from tronapi.base.function_identifiers import FallbackFn
 from tronapi.base.normalizers import abi_address_to_hex, abi_bytes_to_bytes, abi_string_to_text
@@ -126,3 +134,19 @@ def encode_abi(tron, abi, arguments, data=None):
         return to_hex(HexBytes(data) + encoded_arguments)
     else:
         return encode_hex(encoded_arguments)
+
+
+def get_function_info(fn_name, contract_abi=None, fn_abi=None, args=None, kwargs=None):
+    if args is None:
+        args = tuple()
+    if kwargs is None:
+        kwargs = {}
+
+    if fn_abi is None:
+        fn_abi = find_matching_fn_abi(contract_abi, fn_name, args, kwargs)
+
+    fn_selector = encode_hex(function_abi_to_4byte_selector(fn_abi))
+
+    fn_arguments = merge_args_and_kwargs(fn_abi, args, kwargs)
+
+    return fn_abi, fn_selector, fn_arguments

tronapi/contract.py

@@ -15,7 +15,7 @@
     fallback_func_abi_exists,
     check_if_arguments_can_be_encoded
 )
-from tronapi.base.contracts import find_matching_fn_abi
+from tronapi.base.contracts import find_matching_fn_abi, encode_abi, get_function_info
 from tronapi.base.datatypes import PropertyCheckingFactory
 from tronapi.base.decorators import (
     combomethod,
@@ -241,6 +241,20 @@ def constructor(cls):
                                    cls.abi,
                                    cls.bytecode)
 
+    @combomethod
+    def encodeABI(cls, fn_name, args=None, kwargs=None, data=None):
+        """Encodes the arguments using the Tron ABI for the contract function
+        that matches the given name and arguments..
+        """
+        fn_abi, fn_selector, fn_arguments = get_function_info(
+            fn_name, contract_abi=cls.abi, args=args, kwargs=kwargs,
+        )
+
+        if data is None:
+            data = fn_selector
+
+        return encode_abi(cls.tron, fn_abi, fn_arguments, data)
+
     @staticmethod
     def get_fallback_function(abi, tron, address=None):
         if abi and fallback_func_abi_exists(abi):