jinx/python-uncompyle6
1
Fork 0
mirror of https://github.com/rocky/python-uncompyle6.git synced 2025-08-04 09:22:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix bugs in using pysource from fragments.

Browse Source
This commit is contained in:
rocky
2015-12-25 10:47:07 -05:00
parent 1d533cbb23
commit 39f0f7440b
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
uncompyle6/semantics/fragments.py
View File

@@ -70,6 +70,7 @@ class Traverser(pysource.Walker, object):
'f': StringIO(),
'indent': '',
}
self.version = version
self.p = get_python_parser(version, dict(debug_parser))
self.showast = showast
self.__params = params
@@ -79,6 +80,7 @@ class Traverser(pysource.Walker, object):
self.return_none = False
self.mod_globs = set()
self.currentclass = None
self.classes = []
self.pending_newlines = 0
self.hide_internal = False

9
uncompyle6/semantics/pysource.py
View File

@@ -1480,7 +1480,7 @@ class Walker(GenericASTTraversal, object):
self.print_(self.indent, 'global ', g)
self.mod_globs -= all_globals
rn = ('None' in code.co_names) and not find_none(ast)
self.gen_source(ast, code._customize, isLambda=isLambda, returnNone=rn)
self.gen_source(ast, code.co_name, code._customize, isLambda=isLambda, returnNone=rn)
code._tokens = None; code._customize = None # save memory
def build_class(self, code):
@@ -1530,15 +1530,16 @@ class Walker(GenericASTTraversal, object):
for g in find_globals(ast, set()):
self.print_(indent, 'global ', g)
self.gen_source(ast, code._customize)
self.gen_source(ast, code.co_name, code._customize)
code._tokens = None; code._customize = None # save memory
self.classes.pop(-1)
def gen_source(self, ast, customize, isLambda=0, returnNone=False):
def gen_source(self, ast, name, customize, isLambda=0, returnNone=False):
"""convert AST to source code"""
rn = self.return_none
self.return_none = returnNone
self.name = name
# if code would be empty, append 'pass'
if len(ast) == 0:
self.print_(self.indent, 'pass')
@@ -1631,7 +1632,7 @@ def deparse_code(version, co, out=sys.stdout, showasm=False, showast=False,
pass
# What we've been waiting for: Generate source from AST!
deparsed.gen_source(deparsed.ast, customize)
deparsed.gen_source(deparsed.ast, co.co_name, customize)
for g in deparsed.mod_globs:
deparsed.write('# global %s ## Warning: Unused global' % g)