expose visibility/stateMutability for functionDef objects #4

Author: tintinweb

solidity_parser/__main__.py

@@ -59,7 +59,12 @@
             ## functions
             print(("\t" * level) + "=== functions")
             level += 2
-            for name in contract_object.functions.keys():
-                print(("\t" * level) + "* " + str(name))
+            for name, funcObj in contract_object.functions.items():
+                txtAttribs = []
+                if funcObj.visibility:
+                    txtAttribs.append(funcObj.visibility)
+                if funcObj.stateMutability:
+                    txtAttribs.append(funcObj.stateMutability)
+                print(("\t" * level) + "* " + str(name) + "\t\t (" + ','.join(txtAttribs)+ ")")
             level -= 2
 

solidity_parser/parser.py

@@ -1146,6 +1146,9 @@ class FunctionObject(object):
 
                 def __init__(self, node):
                     self._node = node
+                    if(node.type=="FunctionDefinition"):
+                        self.visibility = node.visibility
+                        self.stateMutability = node.stateMutability
                     self.arguments = {}
                     self.returns = {}
                     self.declarations = {}