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

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

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

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.23.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]