Treat match and case as soft keywords in lambda assignments (RustPyth…

…on#4623)
zeroxer · Mar 4, 2023 · 1871a16 · 1871a16
1 parent 4e19be7
commit 1871a16
Show file tree

Hide file tree

Showing 3 changed files with 243 additions and 1 deletion.
diff --git a/compiler/parser/src/parser.rs b/compiler/parser/src/parser.rs
@@ -604,6 +604,8 @@ match match:
     case 1: pass
     case 2:
         pass
+match = lambda query: query == event
+print(match(12))
 "#,
             "<test>",
         )

diff --git a/compiler/parser/src/snapshots/rustpython_parser__parser__tests__match_as_identifier.snap b/compiler/parser/src/snapshots/rustpython_parser__parser__tests__match_as_identifier.snap
diff --git a/compiler/parser/src/soft_keywords.rs b/compiler/parser/src/soft_keywords.rs
@@ -59,11 +59,15 @@ where
                     let mut nesting = 0;
                     let mut first = true;
                     let mut seen_colon = false;
+                    let mut seen_lambda = false;
                     while let Some(Ok((_, tok, _))) = self.underlying.peek() {
                         match tok {
                             Tok::Newline => break,
+                            Tok::Lambda if nesting == 0 => seen_lambda = true,
                             Tok::Colon if nesting == 0 => {
-                                if !first {
+                                if seen_lambda {
+                                    seen_lambda = false;
+                                } else if !first {
                                     seen_colon = true;
                                 }
                             }