You've already forked python-uncompyle6
mirror of
https://github.com/rocky/python-uncompyle6.git
synced 2025-08-03 00:45:53 +08:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
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 AST(spark_AST):
|
|
def isNone(self):
|
|
"""An AST None token. We can't use regular list comparisons
|
|
because AST 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.type)
|
|
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 = str(node)
|
|
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
|