mirror of
https://github.com/rocky/python-uncompyle6.git
synced 2024-11-23 21:20:06 +00:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
import sys
|
|
from uncompyle6.scanners.tok import NoneToken
|
|
from spark_parser.ast import AST as spark_AST
|
|
|
|
intern = sys.intern
|
|
|
|
|
|
class SyntaxTree(spark_AST):
|
|
def __init__(self, *args, transformed_by=None, **kwargs):
|
|
self.transformed_by = transformed_by
|
|
super(SyntaxTree, self).__init__(*args, **kwargs)
|
|
|
|
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
|
|
if self.transformed_by is not None:
|
|
if self.transformed_by is True:
|
|
rv += " (transformed)"
|
|
else:
|
|
rv += " (transformed by %s)" % self.transformed_by
|
|
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="")
|
|
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
|
|
|
|
def first_child(self):
|
|
if len(self) > 0:
|
|
child = self[0]
|
|
if not isinstance(child, SyntaxTree):
|
|
return child
|
|
return child.first_child()
|
|
return self
|
|
|
|
def last_child(self):
|
|
if len(self) > 0:
|
|
child = self[-1]
|
|
if not isinstance(child, SyntaxTree):
|
|
return child
|
|
return child.last_child()
|
|
return self
|