Merge pull request #308 from fubuloubu/gas_cost_abi

Modification of abi format to include gas cost estimate for functions…

viper/compiler.py

@@ -12,11 +12,6 @@ def compile(code, *args, **kwargs):
     return compile_lll.assembly_to_evm(compile_lll.compile_to_assembly(lll))
 
 
-def mk_full_signature(code, *args, **kwargs):
-    o = parser.mk_full_signature(parser.parse(code))
-    return o
-
-
 def gas_estimate(origcode, *args, **kwargs):
     o = {}
     code = optimizer.optimize(parser.parse_to_lll(origcode))
@@ -32,6 +27,18 @@ def gas_estimate(origcode, *args, **kwargs):
     return o
 
 
+def mk_full_signature(code, *args, **kwargs):
+    abi = parser.mk_full_signature(parser.parse(code))
+    # Add gas estimates for each function to ABI
+    gas_estimates = gas_estimate(code)
+    for idx, func in enumerate(abi):
+        func_name = func['name'].split('(')[0]
+        # Skip __init__, has no estimate
+        if func_name != '__init__':
+            abi[idx]['gas'] = gas_estimates[func_name]
+    return abi
+
+
 # Dummy object, as some tools expect this interface
 class Compiler(object):