Add Tree.find_token() method

lark-parser · Sep 11, 2024 · bd40098 · bd40098
1 parent 5faea92
commit bd40098
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lark/tree.py b/lark/tree.py
@@ -171,6 +171,10 @@ def find_data(self, data: str) -> 'Iterator[Tree[_Leaf_T]]':
 
 ###}
 
+    def find_token(self, typ: str) -> Iterator[_Leaf_T]:
+        """Returns all tokens whose type equals the given typ."""
+        return self.scan_values(lambda v: isinstance(v, Token) and v.type == typ)
+
     def expand_kids_by_data(self, *data_values):
         """Expand (inline) children with any of the given data values. Returns True if anything changed"""
         changed = False