pbi-parsers 0.7.21__py3-none-any.whl → 0.8.0__py3-none-any.whl

pbi_parsers/__init__.py

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

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.

{pbi_parsers-0.7.21.dist-info → pbi_parsers-0.8.0.dist-info}/METADATA

@@ -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.dist-info → pbi_parsers-0.8.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pbi_parsers/__init__.py,sha256=XYRPT6QSy184iWUMvcvZjM8YxFQb7VBbETk4M3-OF3M,92
+pbi_parsers/__init__.py,sha256=cFx9LLUu1MM9UFcbhuA1oauYK8JiPDwU0ohEoJnrI_I,91
 pbi_parsers/base/__init__.py,sha256=U7QpzFFD9A4wK3ZHin6xg5fPgTca0y5KC-O7nrZ-flM,115
 pbi_parsers/base/lexer.py,sha256=5iOkdYzJ9wGwz7I4rDlc7slrRNUPzi0oFFkDxN1d62M,4180
 pbi_parsers/base/tokens.py,sha256=slIVl4673xXomqHMgrn1ApHs8YRvcC2yQgs1zfSpe1U,2220
@@ -10,7 +10,7 @@ pbi_parsers/dax/parser.py,sha256=QLKrIBcxZ26TGhTHpeKcTGEHEHUDLC6IgpxxrdJzdek,182
 pbi_parsers/dax/tokens.py,sha256=nY1laCbL8vwALpJ4jcd8k4lAscqOwqdw3dFuj4_KKVk,1234
 pbi_parsers/dax/utils.py,sha256=OURPa-b6Ldn0_KKXPdLIPA3Zdc12OfbbFd2X5SocCek,4402
 pbi_parsers/dax/exprs/__init__.py,sha256=OUfiXzZYp5HkTPE9x595MMxpsgG1IvsED8p8spAKZuk,3432
-pbi_parsers/dax/exprs/_base.py,sha256=KIlP2AxtZVJGy49ANZfotYnBKt9-EDKOuTx2k48k9cc,1634
+pbi_parsers/dax/exprs/_base.py,sha256=bMHLICgAUOqAKl_S9d6V8kk62cqM1jynUY2-gBJlEcs,2732
 pbi_parsers/dax/exprs/_utils.py,sha256=BxxRCtsqpL9t330ZfBnkm1lq7B_ejAoG89Ot2bLYJig,2123
 pbi_parsers/dax/exprs/add_sub.py,sha256=O-1DGXr4F4WWG1PyL1xra_pFR0ZxFo-_y-e6SioHrQc,2241
 pbi_parsers/dax/exprs/add_sub_unary.py,sha256=OhLF_AFxatvbI7X_Tj3_wQ0gjsnmuwkMkmcCzsECsK4,2003
@@ -73,7 +73,7 @@ pbi_parsers/pq/exprs/statement.py,sha256=JSg48pGAU3Ka2pt4lzVsYlVOqGeF_ARGm8Ajf0l
 pbi_parsers/pq/exprs/try_expr.py,sha256=UcnqfA-t9S1LVrKqeNUT8n4JJcO-ZQZoJrxAdjJ-GMA,1692
 pbi_parsers/pq/exprs/type_expr.py,sha256=hH5ubrIJaxwQsopNJHUZ4ByS1rHEgv2Tf8ocYqSukXM,2570
 pbi_parsers/pq/exprs/variable.py,sha256=wp4t0QHIGA264sXnWp7XVe1H8MJzMIOaoLNBQe-dfNk,1602
-pbi_parsers-0.7.21.dist-info/METADATA,sha256=EbXAfjLz-llL3r1V_cmYmX2aLZ-3Djh77b2Dsa96v50,2907
-pbi_parsers-0.7.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-pbi_parsers-0.7.21.dist-info/licenses/LICENSE,sha256=Sn0IfXOE4B0iL9lZXmGmRuTGyJeCtefxcfws0bLjp2g,1072
-pbi_parsers-0.7.21.dist-info/RECORD,,
+pbi_parsers-0.8.0.dist-info/METADATA,sha256=o3D15LUYr8IlrK-TNjptPgLFlSvwHF7BqmkXfOPFJtA,2906
+pbi_parsers-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+pbi_parsers-0.8.0.dist-info/licenses/LICENSE,sha256=Sn0IfXOE4B0iL9lZXmGmRuTGyJeCtefxcfws0bLjp2g,1072
+pbi_parsers-0.8.0.dist-info/RECORD,,