Improve node.__str__ (fix #509)

5 years ago · b4a5df2388
parent 0d08823c77
commit b4a5df2388
1 changed files with 6 additions and 1 deletions
--- a/slither/core/cfg/node.py
+++ b/slither/core/cfg/node.py
@ -914,7 +914,12 @@ class Node(SourceMapping, ChildFunction):
    ###################################################################################

    def __str__(self):
-        txt = NodeType.str(self._node_type) + ' ' + str(self.expression)
+        additional_info = ''
+        if self.expression:
+            additional_info += ' ' + str(self.expression)
+        elif self.variable_declaration:
+            additional_info += ' ' + str(self.variable_declaration)
+        txt = NodeType.str(self._node_type) + additional_info
        return txt

    # endregion