add: signature from abi to tester funcs

Author: heikoheiko

ethereum/abi.py

@@ -20,6 +20,7 @@ def __init__(self, full_signature):
             full_signature = json_decode(full_signature)
         for sig_item in full_signature:
             encode_types = [f['type'] for f in sig_item['inputs']]
+            signature = [(f['type'], f['name']) for f in sig_item['inputs']]
             name = sig_item['name']
             if '(' in name:
                 name = name[:name.find('(')]
@@ -41,7 +42,8 @@ def __init__(self, full_signature):
                     "prefix": prefix,
                     "encode_types": encode_types,
                     "decode_types": decode_types,
-                    "is_unknown_type": is_unknown_type
+                    "is_unknown_type": is_unknown_type,
+                    "signature": signature
                 }
             elif sig_item['type'] == 'event':
                 prefix = big_endian_to_int(utils.sha3(sig))

ethereum/tester.py

@@ -9,8 +9,9 @@
 from ethereum.slogging import LogRecorder, configure_logging, set_level
 from ethereum.utils import to_string
 from ethereum._solidity import get_solidity
+from ethereum import slogging
 import rlp
-from rlp.utils import decode_hex, encode_hex, ascii_chr
+from rlp.utils import ascii_chr
 
 serpent = None
 
@@ -141,8 +142,23 @@ def kall(*args, **kwargs):
                             return outdata
                     return kall
 
-                for f in self._translator.function_data:
-                    vars(self)[f] = kall_factory(f)
+                for fname in self._translator.function_data:
+                    func = kall_factory(fname)
+                    # create wrapper with signature
+                    # in IPython: mycontract.yeah?
+                    # Definition:  mycontract.yeah(uint32_number, uint256_counter, bool_really)
+                    signature = self._translator.function_data[fname]['signature']
+                    fsig = ', '.join(name for typ, name in signature)
+                    if fsig:
+                        fsig += ', '
+                    wfunc = 'def %s(%s**kargs): return func(%s**kargs)' % (fname, fsig, fsig)
+                    wfunc_code = compile(wfunc, 'na', 'exec')
+                    fakeglobals = {}
+                    eval(wfunc_code, {'func': func}, fakeglobals)  # evil, use in tester only!
+                    func = fakeglobals[fname]
+                    func.__doc__ = '%s(%s)' % (fname, ', '.join(('%s %s' % x) for x in signature))
+
+                    setattr(self, fname, func)
 
         return _abi_contract(me, code, sender, endowment, language)