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

Possibly linemap improvements

Browse Source
This commit is contained in:
rocky
2018-01-26 01:56:41 -05:00
parent b552c413f9
commit 5e13077fd2
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
uncompyle6/main.py
View File

@@ -69,7 +69,7 @@ def decompile(
code_objects = code_objects, code_objects = code_objects,
is_pypy = is_pypy, is_pypy = is_pypy,
) )
header_count = 1+len(sys_version_lines) header_count = 3+len(sys_version_lines)
linemap = [(line_no, deparsed.source_linemap[line_no]+header_count) linemap = [(line_no, deparsed.source_linemap[line_no]+header_count)
for line_no in for line_no in
sorted(deparsed.source_linemap.keys())] sorted(deparsed.source_linemap.keys())]

7
uncompyle6/semantics/linemap.py
View File

@@ -27,6 +27,13 @@ class LineMapWalker(SourceWalker):
self.source_linemap[self.current_line_number] = node.linestart self.source_linemap[self.current_line_number] = node.linestart
return super(LineMapWalker, self).default(node) return super(LineMapWalker, self).default(node)
def n_LOAD_CONST(self, node):
if hasattr(node, 'linestart'):
if node.linestart:
self.source_linemap[self.current_line_number] = node.linestart
return super(LineMapWalker, self).n_LOAD_CONST(node)
class LineMapFragmentWalker(fragments.FragmentsWalker, LineMapWalker): class LineMapFragmentWalker(fragments.FragmentsWalker, LineMapWalker):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(LineMapFragmentWalker, self).__init__(*args, **kwargs) super(LineMapFragmentWalker, self).__init__(*args, **kwargs)