query-cache-common 1.22.0__tar.gz → 1.24.0__tar.gz

{query_cache_common-1.22.0 → query_cache_common-1.24.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: query-cache-common
-Version: 1.22.0
+Version: 1.24.0
 Summary: Common code for Query Cache shared across client and server
 License: 
                                          Apache License

{query_cache_common-1.22.0 → query_cache_common-1.24.0}/src/query_cache_common/models/services/clone_service_models.py

@@ -33,6 +33,7 @@ class CloneRequest(BaseSerDeModel):
     clone_source_table_type: t.Optional[str] = None
     table_properties: t.Optional[TableProperties] = None
     clone_chain_depth_limit: t.Optional[int] = None
+    table_namespace: t.Optional[str] = None
 
 
 @proto_dataclass(clone_service_pb2.ReadyToCloneResponse)

{query_cache_common-1.22.0 → query_cache_common-1.24.0}/src/query_cache_common/models/services/execution_service_models.py

@@ -51,6 +51,7 @@ class SQLExecution(BaseSerDeModel):
     dbt_node_state: t.Optional[shared_models.DbtNodeState] = None
     default_schema: t.Optional[str] = None
     from_speculative_submit: bool = False
+    table_namespace: t.Optional[str] = None
 
     @classmethod
     def from_submit_sql_request(
@@ -68,6 +69,7 @@ class SQLExecution(BaseSerDeModel):
             labels=req.labels,
             dbt_node_state=req.dbt_node_state,
             default_schema=req.default_schema,
+            table_namespace=req.table_namespace,
         )
 
 
@@ -80,6 +82,7 @@ class ValuesExecution(BaseSerDeModel):
     semantic_extras: t.Dict[str, str] = field(default_factory=dict)
     labels: t.Dict[str, str] = field(default_factory=dict)
     dbt_node_state: t.Optional[shared_models.DbtNodeState] = None
+    table_namespace: t.Optional[str] = None
 
     @classmethod
     def from_submit_values_request(
@@ -93,6 +96,7 @@ class ValuesExecution(BaseSerDeModel):
             semantic_extras=req.semantic_extras,
             labels=req.labels,
             dbt_node_state=req.dbt_node_state,
+            table_namespace=req.table_namespace,
         )
 
 

query_cache_common-1.24.0/src/query_cache_common/models/services/selector_service_models.py

@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+import typing as t
+
+from query_cache_protobuf.query_cache.services import selector_service_pb2
+
+from query_cache_common.decorators import proto_dataclass, proto_enum
+from query_cache_common.models.base import BaseSerDeEnum, BaseSerDeModel
+
+
+@proto_enum(selector_service_pb2.SelectorCriteria)
+class SelectorCriteria(BaseSerDeEnum):
+    SC_UNSPECIFIED = "SC_UNSPECIFIED"
+    MODIFIED = "MODIFIED"
+    NEW = "NEW"
+    UNMODIFIED = "UNMODIFIED"
+    OLD = "OLD"
+    BODY = "BODY"
+    CONFIGS = "CONFIGS"
+    RELATION = "RELATION"
+    PERSISTED_DESCRIPTIONS = "PERSISTED_DESCRIPTIONS"
+    MACROS = "MACROS"
+    CONTRACT = "CONTRACT"
+
+
+@proto_dataclass(selector_service_pb2.DbtNodeData)
+class DbtNodeData(BaseSerDeModel):
+    node_unique_id: str
+    node_hash: str
+    node_body_hash: t.Optional[str] = None
+    node_configs_hash: t.Optional[str] = None
+    node_persisted_descriptions_hash: t.Optional[str] = None
+    node_macros_hash: t.Optional[str] = None
+    node_contract_hash: t.Optional[str] = None
+    node_database_representation: t.Optional[str] = None
+
+
+@proto_dataclass(selector_service_pb2.SelectorRequest)
+class SelectorRequest(BaseSerDeModel):
+    target: str
+    project_id: str
+    nodes: t.List[DbtNodeData]
+    selector_criteria: SelectorCriteria
+
+
+@proto_dataclass(selector_service_pb2.SelectorResponse)
+class SelectorResponse(BaseSerDeModel):
+    node_unique_ids: t.List[str]

{query_cache_common-1.22.0 → query_cache_common-1.24.0}/src/query_cache_common/models/services/sql_service_models.py

@@ -29,6 +29,7 @@ class SubmitValuesRequest(BaseSerDeModel):
     clone_table_properties: t.Optional[TableProperties] = None
     clone_chain_depth_limit: t.Optional[int] = None
     dbt_node_state: t.Optional[shared_models.DbtNodeState] = None
+    table_namespace: t.Optional[str] = None
 
 
 @proto_dataclass(sql_service_pb2.SubmitEnrichedSQLRequest)
@@ -52,6 +53,7 @@ class SubmitEnrichedSQLRequest(BaseSerDeModel):
     dbt_node_state: t.Optional[shared_models.DbtNodeState] = None
     default_schema: t.Optional[str] = None
     compare_unrendered_code: bool = False
+    table_namespace: t.Optional[str] = None
 
 
 @proto_dataclass(sql_service_pb2.QueryHashMatchMetadataInfo)

query_cache_common-1.22.0/src/query_cache_common/models/services/state_selector_service_models.py

@@ -1,40 +0,0 @@
-from __future__ import annotations
-
-import typing as t
-
-from query_cache_protobuf.query_cache.services import state_selector_service_pb2
-
-from query_cache_common.decorators import proto_dataclass, proto_enum
-from query_cache_common.models import shared_models
-from query_cache_common.models.base import BaseSerDeEnum, BaseSerDeModel
-
-
-@proto_dataclass(state_selector_service_pb2.StateSelectorRequest)
-class StateSelectorRequest(BaseSerDeModel):
-    target: str
-    nodes: t.List[shared_models.DbtNodeState]
-    selector_criteria: SelectorCriteria
-    project_id: t.Optional[str] = None
-    project_name: t.Optional[str] = None
-
-
-@proto_enum(state_selector_service_pb2.SelectorCriteria)
-class SelectorCriteria(BaseSerDeEnum):
-    MODIFIED = "MODIFIED"
-    NEW = "NEW"
-    UNMODIFIED = "UNMODIFIED"
-    OLD = "OLD"
-    BODY = "BODY"
-    CONFIGS = "CONFIGS"
-    RELATION = "RELATION"
-    PERSISTED_DESCRIPTIONS = "PERSISTED_DESCRIPTIONS"
-    MACROS = "MACROS"
-    CONTRACT = "CONTRACT"
-
-
-@proto_dataclass(state_selector_service_pb2.StateSelectorResponse)
-class StateSelectorResponse(BaseSerDeModel):
-    modified_ids: t.List[str]
-    new_ids: t.List[str]
-    unmodified_ids: t.List[str]
-    old_ids: t.List[str]