pbi-parsers 0.7.21__tar.gz → 0.8.0__tar.gz

{pbi_parsers-0.7.21 → pbi_parsers-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pbi_parsers
-Version: 0.7.21
+Version: 0.8.0
 Summary: Power BI lexer, parsers, and formatters for DAX and M (Power Query) languages
 Project-URL: Homepage, https://github.com/douglassimonsen/pbi_parsers
 Project-URL: Documentation, https://douglassimonsen.github.io/pbi_parsers/

{pbi_parsers-0.7.21 → pbi_parsers-0.8.0}/pbi_parsers/__init__.py

@@ -1,6 +1,6 @@
 from . import dax, pq
 
-__version__ = "0.7.21"
+__version__ = "0.8.0"
 
 
 __all__ = [

{pbi_parsers-0.7.21 → pbi_parsers-0.8.0}/pbi_parsers/dax/exprs/_base.py

@@ -1,4 +1,5 @@
-from typing import TYPE_CHECKING, Any
+from collections.abc import Callable
+from typing import TYPE_CHECKING, Any, TypeVar
 
 from pbi_parsers.dax.tokens import TokenType
 
@@ -6,6 +7,9 @@ if TYPE_CHECKING:
     from pbi_parsers.dax.parser import Parser
 
 
+T = TypeVar("T", bound="Expression")
+
+
 class Expression:
     pre_comments: list[Any] = []
     post_comments: list[Any] = []
@@ -23,6 +27,37 @@ class Expression:
         msg = "This method should be implemented by subclasses."
         raise NotImplementedError(msg)
 
+    def find(self, cls_type: type[T] | tuple[type[T], ...], attributes: Callable[[T], bool] | None = None) -> T:
+        attributes = attributes or (lambda _x: True)
+
+        if isinstance(self, cls_type) and attributes(self):
+            return self
+
+        for child in self.children():
+            try:
+                return child.find(cls_type, attributes)
+            except ValueError:
+                continue
+
+        msg = f"Matching {cls_type} not found in expression tree."
+        raise ValueError(msg)
+
+    def find_all(
+        self,
+        cls_type: type[T] | tuple[type[T], ...],
+        attributes: Callable[[T], bool] | None = None,
+    ) -> list[T]:
+        attributes = attributes or (lambda _x: True)
+        results: list[T] = []
+
+        if isinstance(self, cls_type) and attributes(self):
+            results.append(self)
+
+        for child in self.children():
+            results.extend(child.find_all(cls_type, attributes))
+
+        return results
+
     @classmethod
     def match(cls, parser: "Parser") -> "Expression | None":
         """Attempt to match the current tokens to this expression type.