update contract

Author: serderovsh

tronapi/common/contracts.py

@@ -70,13 +70,16 @@ def find_matching_fn_abi(abi, fn_identifier=None, args=None, kwargs=None):
         encoding_filter,
     ])
     function_candidates = pipe(abi, *filters)
+
     if len(function_candidates) == 1:
         return function_candidates[0]
     else:
         matching_identifiers = name_filter(abi)
         matching_function_signatures = [abi_to_signature(func) for func in matching_identifiers]
+
         arg_count_matches = len(arg_count_filter(matching_identifiers))
         encoding_matches = len(encoding_filter(matching_identifiers))
+
         if arg_count_matches == 0:
             diagnosis = "\nFunction invocation failed due to improper number of arguments."
         elif encoding_matches == 0:

tronapi/contract.py

@@ -208,7 +208,8 @@ def factory(cls, tron, class_name=None, **kwargs):
             class_name or cls.__name__,
             (cls,),
             kwargs,
-            normalizers=normalizers)
+            normalizers=normalizers
+        )
 
         setattr(contract, 'functions', ContractFunctions(contract.abi, contract.tron))
         setattr(contract, 'fallback', Contract.get_fallback_function(contract.abi, contract.tron))

tronapi/trx.py

@@ -805,14 +805,18 @@ def get_contract(self, contract_address):
         })
 
     def contract(self, address=None, **kwargs):
-        contract_factory_class = kwargs.pop('contract_factory_class',
-                                            self.default_contract_factory)
-        contract_factory = contract_factory_class.factory(self.tron, **kwargs)
+        """Work with a contract
+
+        Args:
+            address (str): TRON Address
+            **kwargs (any): details (bytecode, abi)
+        """
+        factory_class = kwargs.pop('contract_factory_class', self.default_contract_factory)
+        contract_factory = factory_class.factory(self.tron, **kwargs)
 
         if address:
             return contract_factory(address)
-        else:
-            return contract_factory
+        return contract_factory
 
     def validate_address(self, address, _is_hex=False):
         """Validate address