Merge pull request #24 from ltworf/fix_print

Fix print
2020-06-18 17:34:36 +07:00 · 2020-06-18 17:34:36 +07:00 · d3d3c47281
parent 3b1e90b30e a74c17f323
commit d3d3c47281
1 changed files with 3 additions and 6 deletions
--- a/relational/parser.py
+++ b/relational/parser.py
@ -121,14 +121,11 @@ class Node:

    def printtree(self, level: int = 0) -> str:
        '''returns a representation of the tree using indentation'''
-        r = ''
-        for i in range(level):
-            r += '  '
-        r += self.name
-        if self.name in b_operators:
+        r = '  ' * level + self.name
+        if self.name in b_operators and isinstance(self, Binary):
            r += self.left.printtree(level + 1)
            r += self.right.printtree(level + 1)
-        elif self.name in u_operators:
+        elif self.name in u_operators and isinstance(self, Unary):
            r += '\t%s\n' % self.prop
            r += self.child.printtree(level + 1)
        return '\n' + r