Skip to content

Commit addddf8

Browse files
committed
Correct getting code node on mkfunc
1 parent f4d21d3 commit addddf8

File tree

1 file changed

+2
-13
lines changed

1 file changed

+2
-13
lines changed

uncompyle6/semantics/fragments.py

+2-13
Original file line numberDiff line numberDiff line change
@@ -80,14 +80,13 @@
8080
from uncompyle6.semantics.check_ast import checker
8181
from uncompyle6.semantics.consts import (
8282
INDENT_PER_LEVEL,
83-
MAP,
8483
NONE,
8584
PASS,
8685
PRECEDENCE,
8786
TABLE_DIRECT,
88-
TABLE_R,
8987
escape,
9088
)
89+
from uncompyle6.semantics.helper import find_code_node
9190
from uncompyle6.semantics.pysource import (
9291
DEFAULT_DEBUG_OPTS,
9392
TREE_DEFAULT_DEBUG,
@@ -597,17 +596,7 @@ def n_alias(self, node):
597596
def n_mkfunc(self, node):
598597
start = len(self.f.getvalue())
599598

600-
if self.version >= (3, 3) or node[-2] == "kwargs":
601-
# LOAD_CONST code object ..
602-
# LOAD_CONST 'x0' if >= 3.3
603-
# MAKE_FUNCTION ..
604-
code_node = node[-3]
605-
elif node[-2] == "expr":
606-
code_node = node[-2][0]
607-
else:
608-
# LOAD_CONST code object ..
609-
# MAKE_FUNCTION ..
610-
code_node = node[-2]
599+
code_node = find_code_node(node, -2)
611600
func_name = code_node.attr.co_name
612601
self.write(func_name)
613602
self.set_pos_info(code_node, start, len(self.f.getvalue()))

0 commit comments

Comments
 (0)