Start rolling in LOAD_ARG for 3.7+

This commit is contained in:
rocky
2022-05-05 21:21:59 -04:00
parent 8576117d00
commit f5043408ec
7 changed files with 323 additions and 62 deletions

View File

@@ -182,9 +182,11 @@ class ComprehensionMixin:
self.write(" in ")
if node[2] == "expr":
iter_expr = node[2]
elif node[3] == "get_aiter":
iter_expr = node[3]
else:
iter_expr = node[-3]
assert iter_expr == "expr"
assert iter_expr in ("expr", "get_aiter"), iter_expr
self.preorder(iter_expr)
self.preorder(tree[iter_index])
self.prec = p