support for star expression in lists

Author: Mic92

baron/grammator_data_structures.py

@@ -132,11 +132,21 @@ def listmaker_one(pack):
         (test,) = pack
         return [test]
 
+    @pg.production("listmaker : star_expr")
+    def listmaker_one_star(pack):
+        (test,) = pack
+        return [test]
+
     @pg.production("listmaker : test comma listmaker")
     def listmaker_more(pack):
         (test, comma, listmaker) = pack
         return [test, comma] + listmaker
 
+    @pg.production("listmaker : star_expr comma listmaker")
+    def listmaker_more_start(pack):
+        (test, comma, listmaker) = pack
+        return [test, comma] + listmaker
+
     @pg.production("atom : LEFT_BRACKET dictmaker RIGHT_BRACKET")
     def dict(pack):
         (left_bracket, dictmaker, right_bracket,) = pack

tests/test_grammator_data_structures.py

@@ -220,6 +220,74 @@ def test_list_more():
         }
     ])
 
+def test_list_star_one():
+    "[*a]"
+    parse_simple([
+        ('LEFT_SQUARE_BRACKET', '[', [], []),
+        ('STAR', '*', [], []),
+        ('NAME', 'a'),
+        ('RIGHT_SQUARE_BRACKET', ']', [], []),
+    ], [
+        {
+            "type": "list",
+            "first_formatting": [],
+            "second_formatting": [],
+            "third_formatting": [],
+            "fourth_formatting": [],
+            "value": [
+                {
+                    "formatting": [],
+                    "type": "star_expression",
+                    "value": {
+                        "type": "name",
+                        "value": "a",
+                    }
+                }
+            ],
+        }
+    ])
+
+def test_list_star_more():
+    "[*a, 'foo']"
+    parse_simple([
+        ('LEFT_SQUARE_BRACKET', '[', [], []),
+        ('STAR', '*', [], []),
+        ('NAME', 'a'),
+        ('COMMA', ',', [], [('SPACE', ' ')]),
+        ('STRING', "'foo'"),
+        ('RIGHT_SQUARE_BRACKET', ']', [], []),
+    ], [
+        {
+            "type": "list",
+            "first_formatting": [],
+            "second_formatting": [],
+            "third_formatting": [],
+            "fourth_formatting": [],
+            "value": [
+                {
+                    "type": "star_expression",
+                    "formatting": [],
+                    "value": {
+                        "type": "name",
+                        "value": "a",
+                    }
+                },
+                {
+                    "first_formatting": [],
+                    "second_formatting": [{"type": "space", "value": " "}],
+                    "type": "comma",
+                },
+                {
+                    "first_formatting": [],
+                    "second_formatting": [],
+                    "type": "string",
+                    "value": "'foo'",
+                }
+            ],
+        }
+    ])
+
+
 
 def test_dict_empty():
     "{ }"
@@ -301,6 +369,7 @@ def test_dict_one_double_star():
 )
 
 
+
 def test_dict_more_colon():
     "{a: b, b: c, c: d}"
     parse_simple([