pytrilogy 0.0.3.0__py3-none-any.whl → 0.0.3.1__py3-none-any.whl

{pytrilogy-0.0.3.0.dist-info → pytrilogy-0.0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pytrilogy
-Version: 0.0.3.0
+Version: 0.0.3.1
 Summary: Declarative, typed query language that compiles to SQL.
 Home-page: 
 Author: 

{pytrilogy-0.0.3.0.dist-info → pytrilogy-0.0.3.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-trilogy/__init__.py,sha256=NutBQIuxEOak6wIOXS3HLb_AXutozJpl1V0dd6o3D-Y,302
+trilogy/__init__.py,sha256=KLYdc650VOIzjRZdH4baZEtSeBPReI5LTzjij3xENfE,302
 trilogy/compiler.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 trilogy/constants.py,sha256=qZ1d0hoKPPV2HHCoFwPYTVB7b6bXjpWvXd3lE-zEhy8,1494
 trilogy/engine.py,sha256=yOPnR7XCjWG82Gym_LLZBkYKKJdLCvqdCyt8zguNcnM,1103
@@ -17,14 +17,14 @@ trilogy/core/functions.py,sha256=7Pq9jYSJd45L2pxT7AI-_rXVZmeLnmTPp8d1lA4z4Vk,244
 trilogy/core/graph_models.py,sha256=z17EoO8oky2QOuO6E2aMWoVNKEVJFhLdsQZOhC4fNLU,2079
 trilogy/core/internal.py,sha256=iicDBlC6nM8d7e7jqzf_ZOmpUsW8yrr2AA8AqEiLx-s,1577
 trilogy/core/optimization.py,sha256=xGO8piVsLrpqrx-Aid_Y56_5slSv4eZmlP64hCHRiEc,7957
-trilogy/core/query_processor.py,sha256=qh-72MBOpnhIrmBQW4_lLCvJS-H3Hq_vuy--mWamw7U,19336
+trilogy/core/query_processor.py,sha256=HyDxBhQsD9KX-Y7pYznlpCAW6AvI76RqPTQNa1mreoE,19450
 trilogy/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 trilogy/core/models/author.py,sha256=Ai-AFBsr4sbm3T0WE3TxFwB73foGg7-uKUTyQtoBEXQ,67093
-trilogy/core/models/build.py,sha256=uy9sJUHTnsY9sdpyzA8egPENnyTU9fryCKz3p8H0b4I,55562
+trilogy/core/models/build.py,sha256=kiq31T8LtUtgmT37m617Q2MlMvQTuAxJzwb6947EiWU,56127
 trilogy/core/models/build_environment.py,sha256=8UggvlPU708GZWYPJMc_ou2r7M3TY2g69eqGvz03YX0,5528
 trilogy/core/models/core.py,sha256=yie1uuq62uOQ5fjob9NMJbdvQPrCErXUT7JTCuYRyjI,9697
 trilogy/core/models/datasource.py,sha256=c0tGxyH2WwTmAD047tr69U0a6GNVf-ug26H68yii7DA,9257
-trilogy/core/models/environment.py,sha256=E7vdrhN0DR6_UKpxRP1KB3w9_GEcCfsKffQ4uv0_CzA,24779
+trilogy/core/models/environment.py,sha256=QSl-H6nwarzKbQgNRjtwDKMJtA4F_GVQpRs-NMNt-6Q,24983
 trilogy/core/models/execute.py,sha256=ABylFQgtavjjCfFkEsFdUwfMB4UBQLHjdzQ9E67QlAE,33521
 trilogy/core/optimizations/__init__.py,sha256=EBanqTXEzf1ZEYjAneIWoIcxtMDite5-n2dQ5xcfUtg,356
 trilogy/core/optimizations/base_optimization.py,sha256=gzDOKImoFn36k7XBD3ysEYDnbnb6vdVIztUfFQZsGnM,513
@@ -32,7 +32,7 @@ trilogy/core/optimizations/inline_constant.py,sha256=lvNTIXaLNkw3HseJyXyDNk5R52d
 trilogy/core/optimizations/inline_datasource.py,sha256=AHuTGh2x0GQ8usOe0NiFncfTFQ_KogdgDl4uucmhIbI,4241
 trilogy/core/optimizations/predicate_pushdown.py,sha256=g4AYE8Aw_iMlAh68TjNXGP754NTurrDduFECkUjoBnc,9399
 trilogy/core/processing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-trilogy/core/processing/concept_strategies_v3.py,sha256=StDlgfoUIeuh2IoZlRlLgkJp5hA8polgdMmOzL_wLCo,39265
+trilogy/core/processing/concept_strategies_v3.py,sha256=hzClGgWxfqxnx4Nc0iIr_DwOybAnuk5RozCijqsZUg4,39168
 trilogy/core/processing/graph_utils.py,sha256=8QUVrkE9j-9C1AyrCb1nQEh8daCe0u1HuXl-Te85lag,1205
 trilogy/core/processing/utility.py,sha256=Oc5tLGeDDpzhbfo2ZcF8ex1kez-NcJDMcG2Lm5BjS4c,20548
 trilogy/core/processing/node_generators/__init__.py,sha256=s_YV1OYc336DuS9591259qjI_K_CtOCuhkf4t2aOgYs,733
@@ -43,12 +43,12 @@ trilogy/core/processing/node_generators/group_node.py,sha256=94uoGZWvBKJ1eqjbDHC
 trilogy/core/processing/node_generators/group_to_node.py,sha256=E5bEjovSx422d_MlAUCDFdY4P2WJVp61BmWwltkhzA8,3095
 trilogy/core/processing/node_generators/multiselect_node.py,sha256=z9FQOxxUvxW31a0TckFfAvnuvU8vP1GyN224RTbXUAk,7114
 trilogy/core/processing/node_generators/node_merge_node.py,sha256=sv55oynfqgpHEpo1OEtVDri-5fywzPhDlR85qaWikvY,16195
-trilogy/core/processing/node_generators/rowset_node.py,sha256=x3rrdUkywAFiaYRk8lv-ra8KAdmBa8dLXj0FO55lz08,6213
+trilogy/core/processing/node_generators/rowset_node.py,sha256=lVjwxO71HVZ5De82bo7YYILVQ7cYeXdCzIloLCfpyLE,6112
 trilogy/core/processing/node_generators/select_merge_node.py,sha256=VHCPMbnKfg7AOfoYa6PKxpNni-j5JEfliNUiltmZhds,19698
 trilogy/core/processing/node_generators/select_node.py,sha256=Y-zO0AFkTrpi2LyebjpyHU7WWANr7nKZSS9rY7DH4Wo,1888
 trilogy/core/processing/node_generators/union_node.py,sha256=zuMSmgF170vzlp2BBQEhKbqUMjVl2xQDqUB82Dhv-VU,2536
 trilogy/core/processing/node_generators/unnest_node.py,sha256=cOEKnMRzXUW3bwmiOlgn3E1-B38osng0dh2pDykwITY,2410
-trilogy/core/processing/node_generators/window_node.py,sha256=_W_bF-8nYi1fjeZ7rXg54Sf7SRDyjDOwk7S_RBoZCWM,3514
+trilogy/core/processing/node_generators/window_node.py,sha256=6KoxhmpVOTN3HGWT0FIS96xqlm2Inouw9VL2D_0kSg0,3481
 trilogy/core/processing/node_generators/select_helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 trilogy/core/processing/node_generators/select_helpers/datasource_injection.py,sha256=GMW07bb6hXurhF0hZLYoMAKSIS65tat5hwBjvqqPeSA,6516
 trilogy/core/processing/nodes/__init__.py,sha256=DqPG3Y8vl5-UTeox6hn1EE6iwPIJpsM-XeZALHSgLZQ,5058
@@ -66,7 +66,7 @@ trilogy/core/statements/build.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 trilogy/core/statements/common.py,sha256=KxEmz2ySySyZ6CTPzn0fJl5NX2KOk1RPyuUSwWhnK1g,759
 trilogy/core/statements/execute.py,sha256=cSlvpHFOqpiZ89pPZ5GDp9Hu6j6uj-5_h21FWm_L-KM,1248
 trilogy/dialect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-trilogy/dialect/base.py,sha256=uvow18PxtFdabf59x0mAXbazV9bqGAfbBxcCAe8JaFs,40212
+trilogy/dialect/base.py,sha256=H6eebtFV09edKbBpukxxB0xBftolXFhvXbshbZhNeUI,40290
 trilogy/dialect/bigquery.py,sha256=mKC3zoEU232h9RtIXJjqiZ72lWH8a6S28p6wAZKrAfg,2952
 trilogy/dialect/common.py,sha256=cbTo_vamdp8pj9spSjGSH-bSZpy4FpNJ12k5vMvyT2Y,3942
 trilogy/dialect/config.py,sha256=UiBY2tBbNk9owx-zxP_3lN9lErEUXhXIU_bcXA18AvU,2992
@@ -76,24 +76,24 @@ trilogy/dialect/postgres.py,sha256=VH4EB4myjIeZTHeFU6vK00GxY9c53rCBjg2mLbdaCEE,3
 trilogy/dialect/presto.py,sha256=bAxaDcLL21fivPg7hmBd3HJmd0yYJdPdwNgNA5ga7DE,3391
 trilogy/dialect/snowflake.py,sha256=wmao9p26jX5yIX5SC8sRAZTXkPGTvq6ixO693QTfhz8,2989
 trilogy/dialect/sql_server.py,sha256=IN91uEM0MpsiVAlsYC89uMQkiVn0i86B8Tst6v9uFkU,3129
-trilogy/hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+trilogy/hooks/__init__.py,sha256=T3SF3phuUDPLXKGRVE_Lf9mzuwoXWyaLolncR_1kY30,144
 trilogy/hooks/base_hook.py,sha256=I_l-NBMNC7hKTDx1JgHZPVOOCvLQ36m2oIGaR5EUMXY,1180
 trilogy/hooks/graph_hook.py,sha256=c-vC-IXoJ_jDmKQjxQyIxyXPOuUcLIURB573gCsAfzQ,2940
-trilogy/hooks/query_debugger.py,sha256=qds0xnW1cxH6Nv9U1L3gqSqD9OrE3TDQKEjyHyuzkFI,5558
+trilogy/hooks/query_debugger.py,sha256=1npRjww94sPV5RRBBlLqMJRaFkH9vhEY6o828MeoEcw,5583
 trilogy/metadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 trilogy/parsing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-trilogy/parsing/common.py,sha256=ti_2s3EEMAz5NwmctchiB0DpVPVusTehc2l3bZQdMQI,20500
+trilogy/parsing/common.py,sha256=b_TGP-cSV21Y8INI77DhD3c4kg0vpJ5TZUuzU-JcdaU,20594
 trilogy/parsing/config.py,sha256=Z-DaefdKhPDmSXLgg5V4pebhSB0h590vI0_VtHnlukI,111
 trilogy/parsing/exceptions.py,sha256=92E5i2frv5hj9wxObJZsZqj5T6bglvPzvdvco_vW1Zk,38
 trilogy/parsing/helpers.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-trilogy/parsing/parse_engine.py,sha256=7WU9jWH8tc6PSgBo_FcDA51QuhytzPTrZYFagCRw9Ec,53869
+trilogy/parsing/parse_engine.py,sha256=rtlwl-WzcPCd5pm0WhXHwe8JGOJCuQm_EAnRj6qG93w,54318
 trilogy/parsing/render.py,sha256=o_XuQWhcwx1lD9eGVqkqZEwkmQK0HdmWWokGBtdeH4I,17837
 trilogy/parsing/trilogy.lark,sha256=EazfEvYPuvkPkNjUnVzFi0uD9baavugbSI8CyfawShk,12573
 trilogy/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 trilogy/scripts/trilogy.py,sha256=1L0XrH4mVHRt1C9T1HnaDv2_kYEfbWTb5_-cBBke79w,3774
-pytrilogy-0.0.3.0.dist-info/LICENSE.md,sha256=5ZRvtTyCCFwz1THxDTjAu3Lidds9WjPvvzgVwPSYNDo,1042
-pytrilogy-0.0.3.0.dist-info/METADATA,sha256=z1DV_FX4ZThTyGzGCqQwyg78lSPcjDyOLO3NypXMJXk,8983
-pytrilogy-0.0.3.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-pytrilogy-0.0.3.0.dist-info/entry_points.txt,sha256=0petKryjvvtEfTlbZC1AuMFumH_WQ9v8A19LvoS6G6c,54
-pytrilogy-0.0.3.0.dist-info/top_level.txt,sha256=cAy__NW_eMAa_yT9UnUNlZLFfxcg6eimUAZ184cdNiE,8
-pytrilogy-0.0.3.0.dist-info/RECORD,,
+pytrilogy-0.0.3.1.dist-info/LICENSE.md,sha256=5ZRvtTyCCFwz1THxDTjAu3Lidds9WjPvvzgVwPSYNDo,1042
+pytrilogy-0.0.3.1.dist-info/METADATA,sha256=HAkTZhYCZL5M6_WSWKakxXEHP5xqxSMMgKhM2E6HndE,8983
+pytrilogy-0.0.3.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pytrilogy-0.0.3.1.dist-info/entry_points.txt,sha256=0petKryjvvtEfTlbZC1AuMFumH_WQ9v8A19LvoS6G6c,54
+pytrilogy-0.0.3.1.dist-info/top_level.txt,sha256=cAy__NW_eMAa_yT9UnUNlZLFfxcg6eimUAZ184cdNiE,8
+pytrilogy-0.0.3.1.dist-info/RECORD,,

trilogy/__init__.py

@@ -4,6 +4,6 @@ from trilogy.dialect.enums import Dialects
 from trilogy.executor import Executor
 from trilogy.parser import parse
 
-__version__ = "0.0.3.0"
+__version__ = "0.0.3.1"
 
 __all__ = ["parse", "Executor", "Dialects", "Environment", "CONFIG"]

trilogy/core/models/build.py

@@ -242,6 +242,9 @@ class BuildGrain(BaseModel):
     components: set[str] = Field(default_factory=set)
     where_clause: Optional[BuildWhereClause] = None
 
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
     def without_condition(self):
         return BuildGrain(components=self.components)
 
@@ -1171,7 +1174,7 @@ class BuildFilterItem(BuildConceptArgs, BaseModel):
 
 class BuildRowsetLineage(BuildConceptArgs, BaseModel):
     name: str
-    derived_concepts: List[BuildConcept]
+    derived_concepts: List[str]
     select: SelectLineage | MultiSelectLineage
 
 
@@ -1222,8 +1225,8 @@ class BuildSelectLineage(BaseModel):
     limit: Optional[int] = None
     meta: Metadata = Field(default_factory=lambda: Metadata())
     grain: BuildGrain = Field(default_factory=BuildGrain)
-    where_clause: WhereClause | BuildWhereClause | None = Field(default=None)
-    having_clause: HavingClause | BuildHavingClause | None = Field(default=None)
+    where_clause: BuildWhereClause | None = Field(default=None)
+    having_clause: BuildHavingClause | None = Field(default=None)
 
     @property
     def output_components(self) -> List[BuildConcept]:
@@ -1447,7 +1450,9 @@ class Factory:
     ):
         self.grain = grain or Grain()
         self.environment = environment
-        self.local_concepts: dict[str, BuildConcept] = local_concepts or {}
+        self.local_concepts: dict[str, BuildConcept] = (
+            {} if local_concepts is None else local_concepts
+        )
 
     @singledispatchmethod
     def build(self, base):
@@ -1494,7 +1499,7 @@ class Factory:
     @build.register
     def _(self, base: Function) -> BuildFunction:
 
-        new = BuildFunction(
+        new = BuildFunction.model_construct(
             operator=base.operator,
             arguments=[self.build(c) for c in base.arguments],
             output_datatype=base.output_datatype,
@@ -1515,14 +1520,14 @@ class Factory:
 
     @build.register
     def _(self, base: CaseWhen) -> BuildCaseWhen:
-        return BuildCaseWhen(
+        return BuildCaseWhen.model_construct(
             comparison=self.build(base.comparison),
             expr=(self.build(base.expr)),
         )
 
     @build.register
     def _(self, base: CaseElse) -> BuildCaseElse:
-        return BuildCaseElse(expr=self.build(base.expr))
+        return BuildCaseElse.model_construct(expr=self.build(base.expr))
 
     @build.register
     def _(self, base: Concept) -> BuildConcept:
@@ -1537,7 +1542,7 @@ class Factory:
             derivation, final_grain, build_lineage
         )
         is_aggregate = Concept.calculate_is_aggregate(build_lineage)
-        rval = BuildConcept(
+        rval = BuildConcept.model_construct(
             name=base.name,
             datatype=base.datatype,
             purpose=base.purpose,
@@ -1565,12 +1570,12 @@ class Factory:
         else:
             by = [self.build(x) for x in base.by]
         parent = self.build(base.function)
-        return BuildAggregateWrapper(function=parent, by=by)
+        return BuildAggregateWrapper.model_construct(function=parent, by=by)
 
     @build.register
     def _(self, base: ColumnAssignment) -> BuildColumnAssignment:
 
-        return BuildColumnAssignment(
+        return BuildColumnAssignment.model_construct(
             alias=(
                 self.build(base.alias)
                 if isinstance(base.alias, Function)
@@ -1584,26 +1589,31 @@ class Factory:
 
     @build.register
     def _(self, base: OrderBy) -> BuildOrderBy:
-        return BuildOrderBy(items=[self.build(x) for x in base.items])
+        return BuildOrderBy.model_construct(items=[self.build(x) for x in base.items])
 
     @build.register
     def _(self, base: OrderItem) -> BuildOrderItem:
-        return BuildOrderItem(
+        return BuildOrderItem.model_construct(
             expr=(self.build(base.expr)),
             order=base.order,
         )
 
     @build.register
     def _(self, base: WhereClause) -> BuildWhereClause:
-        return BuildWhereClause(conditional=self.build(base.conditional))
+        return BuildWhereClause.model_construct(
+            conditional=self.build(base.conditional)
+        )
 
     @build.register
     def _(self, base: HavingClause) -> BuildHavingClause:
-        return BuildHavingClause(conditional=self.build(base.conditional))
+        return BuildHavingClause.model_construct(
+            conditional=self.build(base.conditional)
+        )
 
     @build.register
     def _(self, base: WindowItem) -> BuildWindowItem:
-        return BuildWindowItem(
+
+        return BuildWindowItem.model_construct(
             type=base.type,
             content=self.build(base.content),
             order_by=[self.build(x) for x in base.order_by],
@@ -1613,7 +1623,7 @@ class Factory:
 
     @build.register
     def _(self, base: Conditional) -> BuildConditional:
-        return BuildConditional(
+        return BuildConditional.model_construct(
             left=(self.build(base.left)),
             right=(self.build(base.right)),
             operator=base.operator,
@@ -1621,7 +1631,7 @@ class Factory:
 
     @build.register
     def _(self, base: SubselectComparison) -> BuildSubselectComparison:
-        return BuildSubselectComparison(
+        return BuildSubselectComparison.model_construct(
             left=(self.build(base.left)),
             right=(self.build(base.right)),
             operator=base.operator,
@@ -1629,7 +1639,7 @@ class Factory:
 
     @build.register
     def _(self, base: Comparison) -> BuildComparison:
-        return BuildComparison(
+        return BuildComparison.model_construct(
             left=(self.build(base.left)),
             right=(self.build(base.right)),
             operator=base.operator,
@@ -1637,7 +1647,7 @@ class Factory:
 
     @build.register
     def _(self, base: AlignItem) -> BuildAlignItem:
-        return BuildAlignItem(
+        return BuildAlignItem.model_construct(
             alias=base.alias,
             concepts=[self.build(x) for x in base.concepts],
             namespace=base.namespace,
@@ -1645,7 +1655,9 @@ class Factory:
 
     @build.register
     def _(self, base: AlignClause) -> BuildAlignClause:
-        return BuildAlignClause(items=[self.build(x) for x in base.items])
+        return BuildAlignClause.model_construct(
+            items=[self.build(x) for x in base.items]
+        )
 
     @build.register
     def _(self, base: RowsetItem):
@@ -1663,7 +1675,7 @@ class Factory:
     def _(self, base: RowsetLineage) -> BuildRowsetLineage:
         out = BuildRowsetLineage(
             name=base.name,
-            derived_concepts=[],
+            derived_concepts=[x.address for x in base.derived_concepts],
             select=base.select,
         )
         return out
@@ -1675,17 +1687,19 @@ class Factory:
             where = factory.build(base.where_clause)
         else:
             where = None
-        return BuildGrain(components=base.components, where_clause=where)
+        return BuildGrain.model_construct(
+            components=base.components, where_clause=where
+        )
 
     @build.register
     def _(self, base: FilterItem) -> BuildFilterItem:
-        return BuildFilterItem(
+        return BuildFilterItem.model_construct(
             content=self.build(base.content), where=self.build(base.where)
         )
 
     @build.register
     def _(self, base: Parenthetical) -> BuildParenthetical:
-        return BuildParenthetical(content=(self.build(base.content)))
+        return BuildParenthetical.model_construct(content=(self.build(base.content)))
 
     @build.register
     def _(self, base: SelectLineage) -> BuildSelectLineage:
@@ -1753,7 +1767,7 @@ class Factory:
         derived_base = []
         for k in base.derived_concepts:
             base_concept = self.environment.concepts[k]
-            x = BuildConcept(
+            x = BuildConcept.model_construct(
                 name=base_concept.name,
                 datatype=base_concept.datatype,
                 purpose=base_concept.purpose,
@@ -1778,7 +1792,7 @@ class Factory:
             local_concepts=local_build_cache,
         )
         where_factory = Factory(environment=self.environment)
-        lineage = BuildMultiSelectLineage(
+        lineage = BuildMultiSelectLineage.model_construct(
             # we don't build selects here; they'll be built automatically in query discovery
             selects=base.selects,
             grain=final_grain,
@@ -1831,7 +1845,7 @@ class Factory:
         factory = Factory(
             grain=base.grain, environment=self.environment, local_concepts=local_cache
         )
-        return BuildDatasource(
+        return BuildDatasource.model_construct(
             name=base.name,
             columns=[factory.build(c) for c in base.columns],
             address=base.address,

trilogy/core/models/environment.py

@@ -39,6 +39,7 @@ from trilogy.core.models.author import (
     Concept,
     ConceptRef,
     Function,
+    SelectLineage,
     UndefinedConcept,
     UndefinedConceptFull,
     address_with_namespace,
@@ -189,6 +190,7 @@ class Environment(BaseModel):
     ] = Field(default_factory=EnvironmentDatasourceDict)
     functions: Dict[str, Function] = Field(default_factory=dict)
     data_types: Dict[str, DataType] = Field(default_factory=dict)
+    named_statements: Dict[str, SelectLineage] = Field(default_factory=dict)
     imports: Dict[str, list[Import]] = Field(
         default_factory=lambda: defaultdict(list)  # type: ignore
     )
@@ -217,6 +219,9 @@ class Environment(BaseModel):
 
         return Factory(self, local_concepts=local_concepts).build(self)
 
+    def add_rowset(self, name: str, lineage: SelectLineage):
+        self.named_statements[name] = lineage
+
     def duplicate(self):
         return Environment.model_construct(
             datasources=self.datasources.duplicate(),

trilogy/core/processing/concept_strategies_v3.py

@@ -779,8 +779,6 @@ def _search_concepts(
     else:
 
         completion_mandatory = mandatory_list
-    if "date.month_seq" in completion_mandatory:
-        raise SyntaxError(completion_mandatory)
     attempted: set[str] = set()
 
     found: set[str] = set()

trilogy/core/processing/node_generators/rowset_node.py

@@ -54,17 +54,15 @@ def gen_rowset_node(
     rowset_relevant: list[BuildConcept] = [
         v
         for v in environment.concepts.values()
-        if isinstance(v.lineage, BuildRowsetItem)
-        and v.lineage.rowset.name == rowset.name
+        if v.address in lineage.rowset.derived_concepts
     ]
-    # logger.info(
-    #     f"{padding(depth)}{LOGGER_PREFIX} rowset relevant nodes are {rowset_relevant}"
-    # )
+
     select_hidden = node.hidden_concepts
     rowset_hidden = [
         x
         for x in rowset_relevant
-        if isinstance(x.lineage, BuildRowsetItem)
+        if x.address in lineage.rowset.derived_concepts
+        and isinstance(x.lineage, BuildRowsetItem)
         and x.lineage.content.address in select_hidden
     ]
     additional_relevant = [

trilogy/core/processing/node_generators/window_node.py

@@ -19,7 +19,6 @@ def resolve_window_parent_concepts(
     if not isinstance(concept.lineage, WINDOW_TYPES):
         raise ValueError
     base = []
-    logger.info(concept.lineage)
     if concept.lineage.over:
         base += concept.lineage.over
     if concept.lineage.order_by:

trilogy/core/query_processor.py

@@ -12,6 +12,8 @@ from trilogy.core.models.build import (
     BuildConcept,
     BuildConditional,
     BuildDatasource,
+    BuildMultiSelectLineage,
+    BuildSelectLineage,
     Factory,
 )
 from trilogy.core.models.environment import Environment
@@ -373,7 +375,10 @@ def get_query_node(
         raise ValueError(f"Statement has no output components {statement}")
 
     history = history or History(base_environment=environment)
-    build_statement = Factory(environment=environment).build(statement)
+    build_statement: BuildSelectLineage | BuildMultiSelectLineage = Factory(
+        environment=environment
+    ).build(statement)
+
     # build_statement = statement
     build_environment = environment.materialize_for_select(
         build_statement.local_concepts

trilogy/dialect/base.py

@@ -84,7 +84,9 @@ CONDITIONAL_ITEMS = (BuildConditional,)
 
 
 def INVALID_REFERENCE_STRING(x: Any, callsite: str = ""):
-    # raise SyntaxError(x)
+    # if CONFIG.validate_missing:
+    #     raise SyntaxError(f"INVALID_REFERENCE_BUG_{callsite}<{x}>")
+
     return f"INVALID_REFERENCE_BUG_{callsite}<{x}>"
 
 

trilogy/hooks/__init__.py

@@ -0,0 +1,4 @@
+from trilogy.hooks.graph_hook import GraphHook
+from trilogy.hooks.query_debugger import DebuggingHook
+
+__all__ = ["DebuggingHook", "GraphHook"]

trilogy/hooks/query_debugger.py

@@ -12,7 +12,6 @@ from trilogy.core.models.execute import (
 )
 from trilogy.core.processing.nodes import StrategyNode
 from trilogy.core.statements.author import SelectStatement
-from trilogy.dialect.bigquery import BigqueryDialect
 from trilogy.hooks.base_hook import BaseHook
 
 
@@ -22,9 +21,6 @@ class PrintMode(Enum):
     FULL = 3
 
 
-renderer = BigqueryDialect()
-
-
 class DebuggingHook(BaseHook):
     def __init__(
         self,
@@ -46,6 +42,9 @@ class DebuggingHook(BaseHook):
         self.process_other = PrintMode(process_other)
         self.messages: list[str] = []
         self.uuid = uuid4()
+        from trilogy.dialect.bigquery import BigqueryDialect
+
+        self.renderer = BigqueryDialect()
 
     def print(self, *args):
         merged = " ".join([str(x) for x in args])
@@ -123,7 +122,7 @@ class DebuggingHook(BaseHook):
         self.print(
             "  " * depth, input.name, "->", input.group_to_grain, "->", select_statement
         )
-        sql = renderer.render_cte(input).statement
+        sql = self.renderer.render_cte(input).statement
         for line in sql.split("\n"):
             logger.debug("  " * (depth) + line)
         if isinstance(input, CTE):

trilogy/parsing/common.py

@@ -503,6 +503,11 @@ def rowset_to_concepts(rowset: RowsetDerivationStatement, environment: Environme
         if isinstance(orig_concept.lineage, FilterItem):
             if orig_concept.lineage.where == rowset.select.where_clause:
                 name = environment.concepts[orig_concept.lineage.content.address].name
+        base_namespace = (
+            f"{rowset.name}.{orig_concept.namespace}"
+            if orig_concept.namespace != rowset.namespace
+            else rowset.name
+        )
 
         new_concept = Concept(
             name=name,
@@ -512,11 +517,7 @@ def rowset_to_concepts(rowset: RowsetDerivationStatement, environment: Environme
             grain=orig_concept.grain,
             # TODO: add proper metadata
             metadata=Metadata(concept_source=ConceptSource.CTE),
-            namespace=(
-                f"{rowset.name}.{orig_concept.namespace}"
-                if orig_concept.namespace != rowset.namespace
-                else rowset.name
-            ),
+            namespace=base_namespace,
             keys=orig_concept.keys,
             derivation=Derivation.ROWSET,
             granularity=orig_concept.granularity,
@@ -538,6 +539,9 @@ def rowset_to_concepts(rowset: RowsetDerivationStatement, environment: Environme
     default_grain = Grain.from_concepts([*pre_output])
     # remap everything to the properties of the rowset
     for x in pre_output:
+        print("xxxx")
+        print(x.address)
+        print(x.grain)
         if x.keys:
             if all([k in orig for k in x.keys]):
                 x.keys = set([orig[k].address if k in orig else k for k in x.keys])

trilogy/parsing/parse_engine.py

@@ -576,11 +576,24 @@ class ParseToObjects(Transformer):
             select=select,
             namespace=self.environment.namespace or DEFAULT_NAMESPACE,
         )
+
+        # clean up current definitions
+        # to_delete = set()
+        # if output.name in self.environment.named_statements:
+        #     for k, v in self.environment.concepts.items():
+        #         if v.derivation == Derivation.ROWSET and v.lineage.rowset.name == name:
+        #             to_delete.add(k)
+        # for k in to_delete:
+        #     self.environment.concepts.pop(k)
+
         for new_concept in rowset_to_concepts(output, self.environment):
             if new_concept.metadata:
                 new_concept.metadata.line_number = meta.line
-            # output.select.local_concepts[new_concept.address] = new_concept
-            self.environment.add_concept(new_concept)
+            self.environment.add_concept(new_concept, force=True)
+
+        self.environment.add_rowset(
+            output.name, output.select.as_lineage(self.environment)
+        )
         return output
 
     @v_args(meta=True)