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

Fixes #2878

diff --git a/CHANGES.md b/CHANGES.md
index a0b87c780150fc537e7cd1a63cd644644d4cf30f..ac9212e994028e394162a8de5f08f2bc7659a7f8 100644 (file)
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -50,7 +50,8 @@
 
 ### Parser
 
 
 ### 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
 
 
 ### Performance
 

diff --git a/src/blib2to3/Grammar.txt b/src/blib2to3/Grammar.txt
index cf4799f8abe860acff10cb1c0fe84ec5130a3b77..0ce6cf39111566aecb7a6a9d6d0d86d08f5132e1 100644 (file)
--- 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]
 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]
 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 (file)
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 04676c1c2c52a578a94a20845e1089149a4857ca..04eda43d5cfd0435fa79a0dedd677746747e94e2 100644 (file)
--- a/tests/test_format.py
+++ b/tests/test_format.py
@@ -62,6 +62,7 @@ SIMPLE_CASES: List[str] = [
 ]
 
 PY310_CASES: List[str] = [
 ]
 
 PY310_CASES: List[str] = [

+    "starred_for_target",

     "pattern_matching_simple",
     "pattern_matching_complex",
     "pattern_matching_extras",
     "pattern_matching_simple",
     "pattern_matching_complex",
     "pattern_matching_extras",