Magic-Mirror/python-uncompyle6
1
0
Fork 0
mirror of https://github.com/rocky/python-uncompyle6.git synced 2025-02-08 05:57:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add "transfrormd_by" param to SyntaxTree

Browse Source 
This aligns code more with decompyle3
This commit is contained in:
rocky 2022-04-21 05:25:14 -04:00
parent c88d9de316
commit 433d7003e0
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
uncompyle6/parsers/treenode.py
View File

@ -6,9 +6,9 @@ intern = sys.intern
class SyntaxTree(spark_AST):
def __init__(self, *args, **kwargs):
def __init__(self, *args, transformed_by=None, **kwargs):
self.transformed_by = transformed_by
super(SyntaxTree, self).__init__(*args, **kwargs)
self.transformed_by = None
def isNone(self):
"""An SyntaxTree None token. We can't use regular list comparisons

10
uncompyle6/semantics/transform.py
View File

@ -378,8 +378,9 @@ class TreeTransform(GenericASTTraversal, object):
def n_import_from37(self, node):
importlist37 = node[3]
assert importlist37 == "importlist37"
if len(importlist37) == 1:
if importlist37 != "importlist37":
return node
if len(importlist37) == 1 and importlist37 == "importlist37":
alias37 = importlist37[0]
store = alias37[1]
assert store == "store"
@ -394,9 +395,10 @@ class TreeTransform(GenericASTTraversal, object):
# import_as37 ::= LOAD_CONST LOAD_CONST importlist37 store POP_TOP
# 'import_as37': ( '%|import %c as %c\n', 2, -2),
node = SyntaxTree(
"import_as37", [node[0], node[1], import_name_attr, store, node[-1]]
"import_as37",
[node[0], node[1], import_name_attr, store, node[-1]],
transformed_by="n_import_from37",
)
node.transformed_by = "n_import_from37"
pass
pass
return node