zack3d/pylingual
1
0
Fork 0
mirror of https://github.com/syssec-utd/pylingual.git synced 2026-05-10 18:39:03 -07:00
Code Issues Packages Projects Releases Wiki Activity

3.14 Loop changes

Browse Source
This commit is contained in:
Xinlong Hu
2025-12-16 18:32:04 -06:00
parent 78fef3e7f0
commit ed22e7a30c
1 changed files with 63 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pylingual/control_flow_reconstruction/templates/Loop.py
+63 -2
View File
@@ -12,6 +12,7 @@ from ..utils import (
is_not_type,
no_back_edges,
versions_below,
versions_except,
versions_from,
ending_instructions,
has_no_lines,
@@ -29,7 +30,7 @@ if TYPE_CHECKING:
@register_template(0, 1)
@register_template(0, 1, *versions_below(3, 14))
class ForLoop(ControlFlowTemplate):
template = T(
for_iter=~N("for_body", "tail"),
@@ -46,6 +47,23 @@ class ForLoop(ControlFlowTemplate):
{for_body}
"""
@register_template(0, 1, *versions_from(3, 14))
class ForLoop3_14(ControlFlowTemplate):
template = T(
for_iter=~N("for_body", "tail").with_cond(with_top_level_instructions("FOR_ITER")),
for_body=~N("for_iter").with_in_deg(1),
tail=N.tail().with_cond(is_not_type(LoopElse)),
)
try_match = make_try_match({EdgeKind.Fall: "tail"}, "for_iter", "for_body")
@to_indented_source
def to_indented_source():
"""
{for_iter}
{for_body}
"""
@register_template(0, 1)
class ForElseLoop(ControlFlowTemplate):
@@ -85,7 +103,7 @@ class LoopedReturn(ControlFlowTemplate):
{for_body}
"""
@register_template(0, 2, *versions_below(3, 10))
@register_template(0, 2, *versions_except((3, 10), (3, 11), (3, 12), (3, 13)))
class SelfLoop3_6(ControlFlowTemplate):
template = T(
loop_body=~N("loop_body", None)
@@ -244,6 +262,49 @@ class WhileIfElseLoop(ControlFlowTemplate):
"""
@register_template(1, 39, *versions_from(3, 14))
class WhileLoop3_14(ControlFlowTemplate):
template = T(
if_header=~N("if_body", "else_body").with_cond(without_top_level_instructions("WITH_EXCEPT_START", "CHECK_EXC_MATCH", "FOR_ITER", "NOP")),
else_body=~N("tail.").with_cond(without_top_level_instructions("RERAISE", "END_FINALLY")).with_in_deg(1),
if_body=~N("if_header").with_in_deg(1),
tail=N.tail(),
)
try_match = make_try_match({EdgeKind.Fall: "tail"}, "if_header", "if_body", "else_body")
@to_indented_source
def to_indented_source():
"""
{if_header}
{if_body}
{else_body?else:}
{else_body}
"""
@register_template(1, 39, *versions_from(3, 14))
class WhileTrueLoop3_14(ControlFlowTemplate):
template = T(
if_header=~N("if_body", "else_body").with_cond(without_top_level_instructions("WITH_EXCEPT_START", "CHECK_EXC_MATCH", "FOR_ITER")).with_cond(starting_instructions("NOP")),
else_body=~N("tail.").with_cond(without_top_level_instructions("RERAISE", "END_FINALLY")).with_in_deg(1),
if_body=~N("if_header").with_in_deg(1),
tail=N.tail(),
)
try_match = make_try_match({EdgeKind.Fall: "tail"}, "if_header", "if_body", "else_body")
@to_indented_source
def to_indented_source():
"""
while True:
{if_header}
{else_body}
{if_body?else:}
{if_body}
"""
@register_template(0, 3)
class InlinedComprehensionTemplate(ControlFlowTemplate):
template = T(