Merge pull request #76 from astral-sh/match-case-end-location

RustPython · May 31, 2023 · fe25708 · fe25708
2 parents ae3a477 + 342cd19
commit fe25708
Show file tree

Hide file tree

Showing 2 changed files with 1,203 additions and 1,229 deletions.
diff --git a/parser/src/python.lalrpop b/parser/src/python.lalrpop
@@ -387,7 +387,9 @@ MatchStatement: ast::Stmt = {
 }
 
 MatchCase: ast::MatchCase = {
-    <start:@L> "case" <pattern:Patterns> <guard:(Guard)?> ":" <body:Suite> <end:@R> => {
+    <start:@L> "case" <pattern:Patterns> <guard:(Guard)?> ":" <body:Suite> => {
+        // SAFETY: `body` is never empty because it is non-optional and `Suite` matches one or more statements.
+        let end = body.last().unwrap().end();
         ast::MatchCase {
             pattern,
             guard: guard.map(Box::new),