Add support for form-from comprehensions

b0o · Dec 8, 2024 · be91184 · be91184
1 parent a1c9c26
commit be91184
Show file tree

Hide file tree

Showing 5 changed files with 233,113 additions and 211,387 deletions.
diff --git a/grammar.js b/grammar.js
@@ -780,12 +780,26 @@ module.exports = grammar(Python, {
         ),
       ),
 
+    _comprehension_for_clause: $ =>
+      choice($.for_in_clause,
+        seq(optional("async"),
+          "for",
+          $.for_from_loop),
+      ),
+
+    _comprehension_clauses: $ => seq(
+      $._comprehension_for_clause,
+      repeat(choice(
+        $._comprehension_for_clause,
+        $.if_clause,
+      )),
+    ),
 
     for_in_loop: $ =>
       seq(
         field("left", $._left_hand_side),
         "in",
-        field("right", $._expressions),
+        field("right", $._expression_within_for_in_clause),
       ),
 
     for_from_relation: $ =>

diff --git a/src/grammar.json b/src/grammar.json
diff --git a/src/node-types.json b/src/node-types.json