From: Batuhan Taskaya <isidentical@gmail.com>
Date: Sat, 5 Mar 2022 01:37:16 +0000 (+0300)
Subject: Allow `for`'s target expression to be starred (#2879)
X-Git-Url: https://git.madduck.net/etc/vim.git/commitdiff_plain/6f4976a7ace2fb5c6a5df57c4cb7dcf65eff44c9

Allow `for`'s target expression to be starred (#2879)

Fixes #2878
---

diff --git a/CHANGES.md b/CHANGES.md
index a0b87c7..ac9212e 100644
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -50,7 +50,8 @@
 
 ### Parser
 
-<!-- Changes to the parser or to version autodetection -->
+- Black can now parse starred expressions in the target of `for` and `async for`
+  statements, e.g `for item in *items_1, *items_2: pass` (#2879).
 
 ### Performance
 
diff --git a/src/blib2to3/Grammar.txt b/src/blib2to3/Grammar.txt
index cf4799f..0ce6cf3 100644
--- a/src/blib2to3/Grammar.txt
+++ b/src/blib2to3/Grammar.txt
@@ -109,7 +109,7 @@ compound_stmt: if_stmt | while_stmt | for_stmt | try_stmt | with_stmt | funcdef
 async_stmt: ASYNC (funcdef | with_stmt | for_stmt)
 if_stmt: 'if' namedexpr_test ':' suite ('elif' namedexpr_test ':' suite)* ['else' ':' suite]
 while_stmt: 'while' namedexpr_test ':' suite ['else' ':' suite]
-for_stmt: 'for' exprlist 'in' testlist ':' suite ['else' ':' suite]
+for_stmt: 'for' exprlist 'in' testlist_star_expr ':' suite ['else' ':' suite]
 try_stmt: ('try' ':' suite
            ((except_clause ':' suite)+
 	    ['else' ':' suite]
diff --git a/tests/data/starred_for_target.py b/tests/data/starred_for_target.py
new file mode 100644
index 0000000..8fc8e05
--- /dev/null
+++ b/tests/data/starred_for_target.py
@@ -0,0 +1,27 @@
+for x in *a, *b:
+    print(x)
+
+for x in a, b, *c:
+    print(x)
+
+for x in *a, b, c:
+    print(x)
+
+for x in *a, b, *c:
+    print(x)
+
+async for x in *a, *b:
+    print(x)
+
+async for x in *a, b, *c:
+    print(x)
+
+async for x in a, b, *c:
+    print(x)
+
+async for x in (
+    *loooooooooooooooooooooong,
+    very,
+    *loooooooooooooooooooooooooooooooooooooooooooooooong,
+):
+    print(x)
diff --git a/tests/test_format.py b/tests/test_format.py
index 04676c1..04eda43 100644
--- a/tests/test_format.py
+++ b/tests/test_format.py
@@ -62,6 +62,7 @@ SIMPLE_CASES: List[str] = [
 ]
 
 PY310_CASES: List[str] = [
+    "starred_for_target",
     "pattern_matching_simple",
     "pattern_matching_complex",
     "pattern_matching_extras",