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

Remove dis3. Fix in 3.x list if not comprehension

Browse Source
This commit is contained in:
rocky
2016-05-28 12:05:38 -04:00
parent bebd85f5e1
commit fca233419f
10 changed files with 31 additions and 442 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
uncompyle6/semantics/pysource.py
View File

@@ -975,7 +975,7 @@ class SourceWalker(GenericASTTraversal, object):
self.prec = 27
n = node[-1]
assert n == 'list_iter'
# find innerst node
# find innermost node
while n == 'list_iter':
n = n[0] # recurse one step
if n == 'list_for': n = n[3]
@@ -1059,31 +1059,38 @@ class SourceWalker(GenericASTTraversal, object):
self.customize(code._customize)
ast = ast[0][0][0][0][0]
try:
n = ast[iter_index]
except:
from trepan.api import debug; debug()
n = ast[iter_index]
assert n == 'list_iter'
## FIXME: I'm not totally sure this is right.
# find innermost node
designator = None
list_if_node = None
while n == 'list_iter':
n = n[0] # recurse one step
if n == 'list_for':
designator = n[2]
if n[2] == 'designator':
designator = n[2]
n = n[3]
elif n in ['list_if', 'list_if_not']:
# FIXME: just a guess
designator = n[1]
list_if_node = n[0]
if n[1] == 'designator':
designator = n[1]
n = n[2]
pass
pass
assert n == 'lc_body', ast
assert designator, "Couldn't find designator in list comprehension"
self.preorder(n[0])
self.write(' for ')
self.preorder(designator)
self.write(' in ')
self.preorder(node[-3])
if list_if_node:
self.write(' if ')
self.preorder(list_if_node)
self.prec = p
def listcomprehension_walk2(self, node):