feat: add stream.peek_until

vberlier · Jun 14, 2021 · 8b0475c · 8b0475c
1 parent 02bdabd
commit 8b0475c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tokenstream/stream.py b/tokenstream/stream.py
@@ -282,6 +282,15 @@ def peek(self, n: int = 1) -> Optional[Token]:
 
         return token
 
+    def peek_until(self, *patterns: TokenPattern) -> Iterator[Token]:
+        """Collect tokens until one of the given patterns matches."""
+        while token := self.peek():
+            if token.match(*patterns):
+                next(self)
+                return
+            yield token
+        raise self.emit_error(UnexpectedEOF(patterns))
+
     @overload
     def collect(self) -> Iterator[Token]:
         ...