Change AST to SyntaxTree in many places

uncompyle6/parsers/treenode.py

@@ -0,0 +1,47 @@
+import sys
+from uncompyle6 import PYTHON3
+from uncompyle6.scanners.tok import NoneToken
+from spark_parser.ast import AST as spark_AST
+
+if PYTHON3:
+    intern = sys.intern
+
+class SyntaxTree(spark_AST):
+    def isNone(self):
+        """An SyntaxTree None token. We can't use regular list comparisons
+        because SyntaxTree token offsets might be different"""
+        return len(self.data) == 1 and NoneToken == self.data[0]
+
+    def __repr__(self):
+        return self.__repr1__('', None)
+
+    def __repr1__(self, indent, sibNum=None):
+        rv = str(self.kind)
+        if sibNum is not None:
+            rv = "%2d. %s" % (sibNum, rv)
+        enumerate_children = False
+        if len(self) > 1:
+            rv += " (%d)" % (len(self))
+            enumerate_children = True
+        rv = indent + rv
+        indent += '    '
+        i = 0
+        for node in self:
+            if hasattr(node, '__repr1__'):
+                if enumerate_children:
+                    child =  node.__repr1__(indent, i)
+                else:
+                    child = node.__repr1__(indent, None)
+            else:
+                inst = node.format(line_prefix='L.')
+                if inst.startswith("\n"):
+                    # Nuke leading \n
+                    inst = inst[1:]
+                if enumerate_children:
+                    child = indent + "%2d. %s" % (i, inst)
+                else:
+                    child = indent + inst
+                pass
+            rv += "\n" + child
+            i += 1
+        return rv