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

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

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

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/auth.py

@@ -3,7 +3,6 @@ from __future__ import annotations
 import typing as t
 from functools import cached_property
 
-
 RUNCACHE_ORG_SCOPE_PREFIX = "runcache:scope:org"
 LEGACY_ORG_SCOPE_PREFIX = "conway:scope:org"
 ORG_SCOPE_PREFIXES = (RUNCACHE_ORG_SCOPE_PREFIX, LEGACY_ORG_SCOPE_PREFIX)
@@ -67,7 +66,7 @@ class Scope:
             raise ValueError(f"Only one organization scope is supported, got multiple: {org_ids}.")
         org_id = org_ids[0]
         if org_id == "*":
-            raise ValueError(f"Wildcard organization ID is not allowed.")
+            raise ValueError("Wildcard organization ID is not allowed.")
         return org_id
 
     @cached_property

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/decorators.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 import enum
 import typing as t
-from dataclasses import fields, Field, dataclass
+from dataclasses import Field, dataclass, fields
 
 from google.protobuf.message import Message
 from typing_extensions import dataclass_transform
@@ -38,7 +38,7 @@ class FieldInfo:
     def create_inner_field_info(self) -> FieldInfo:
         return FieldInfo(
             name=self.name,
-            field_type=[x for x in self.inner_types if x is not type(None)][0],
+            field_type=next(x for x in self.inner_types if x is not type(None)),
             proto_converter={},
         )
 
@@ -321,15 +321,15 @@ def _create_enum_from_proto_method(
         """Convert protobuf enum value to Python enum instance."""
         try:
             return from_proto_map[proto_value]
-        except KeyError:
+        except KeyError as e:
             if hasattr(cls_inner, "_UNSUPPORTED"):
-                return cls_inner._UNSUPPORTED  # ty: ignore[invalid-return-type]
+                return cls_inner._UNSUPPORTED  # ty: ignore[invalid-return-type]  # noqa: SLF001
             valid_values = sorted(from_proto_map.keys())
             proto_enum_name = getattr(proto_enum_cls, "_enum_type", proto_enum_cls).name  # ty: ignore[unresolved-attribute]
             raise ValueError(
                 f"Invalid proto enum value {proto_value} for {proto_enum_name}. "
                 f"Valid values: {valid_values}"
-            )
+            ) from e
 
     return classmethod(from_proto_func)
 
@@ -348,10 +348,10 @@ def _generate_enum_to_proto_method(
         """Convert Python enum instance to protobuf enum value."""
         try:
             return to_proto_map[self]
-        except KeyError:
+        except KeyError as e:
             raise ValueError(
                 f"Invalid enum instance {self} for conversion to proto. "
                 f"This indicates a bug in the proto_enum decorator."
-            )
+            ) from e
 
     return to_proto_func

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/base.py

@@ -5,7 +5,6 @@ import json
 import typing as t
 from dataclasses import dataclass, fields
 
-
 BSDM = t.TypeVar("BSDM", bound="BaseSerDeModel")
 BSDE = t.TypeVar("BSDE", bound="BaseSerDeEnum")
 
@@ -69,7 +68,7 @@ def _deserialize_field_value(
     args = t.get_args(field_type)
 
     if args and type(None) in args:
-        field_type = [arg for arg in args if arg is not type(None)][0]
+        field_type = next(arg for arg in args if arg is not type(None))
 
     if hasattr(field_type, "from_dict"):
         return field_type.from_dict(value)  # ty: ignore[call-non-callable]

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/converters.py

@@ -3,8 +3,8 @@ from __future__ import annotations
 import typing as t
 from datetime import timedelta
 
-from query_cache_protobuf.query_cache.struct_pb2 import ListValue, Struct, Value, NULL_VALUE
 from google.protobuf.duration_pb2 import Duration
+from query_cache_protobuf.query_cache.struct_pb2 import NULL_VALUE, ListValue, Struct, Value
 
 if t.TYPE_CHECKING:
     from google.protobuf.internal.containers import MessageMap

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/fields.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import typing as t
-
 from dataclasses import field
 
 

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/services/client_telemetry_service_models.py

@@ -1,20 +1,19 @@
 from __future__ import annotations
 
-from dataclasses import field
-
-
 import typing as t
+from dataclasses import field
 from datetime import timedelta
 
+from query_cache_protobuf.query_cache.services import client_telemetry_service_pb2
+
 from query_cache_common.decorators import proto_dataclass, proto_enum
-from query_cache_common.models.base import BaseSerDeModel, BaseSerDeEnum
+from query_cache_common.models.base import BaseSerDeEnum, BaseSerDeModel
 from query_cache_common.models.converters import (
-    struct_to_dict,
     dict_to_struct,
     duration_to_timedelta,
+    struct_to_dict,
     timedelta_to_duration,
 )
-from query_cache_protobuf.query_cache.services import client_telemetry_service_pb2
 
 
 @proto_dataclass(client_telemetry_service_pb2.SessionStartRequest)

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

@@ -3,6 +3,8 @@ from __future__ import annotations
 import typing as t
 from dataclasses import field
 
+from query_cache_protobuf.query_cache.services import clone_service_pb2
+
 from query_cache_common.decorators import proto_dataclass
 from query_cache_common.models import shared_models
 from query_cache_common.models._typing import TransformedNodesByQuery
@@ -11,7 +13,6 @@ from query_cache_common.models.converters import dict_to_struct, struct_to_dict
 from query_cache_common.models.fields import (
     transformed_nodes_by_query_field,
 )
-from query_cache_protobuf.query_cache.services import clone_service_pb2
 
 
 @proto_dataclass(clone_service_pb2.TableProperties)

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

@@ -3,12 +3,13 @@ from __future__ import annotations
 import typing as t
 from dataclasses import field
 
+from query_cache_protobuf.query_cache.services import execution_service_pb2
+
 from query_cache_common.decorators import proto_dataclass
 from query_cache_common.models import shared_models
-from query_cache_common.models.services import sql_service_models
 from query_cache_common.models.base import BaseSerDeModel
 from query_cache_common.models.converters import dict_to_struct, struct_to_dict
-from query_cache_protobuf.query_cache.services import execution_service_pb2
+from query_cache_common.models.services import sql_service_models
 
 
 @proto_dataclass(execution_service_pb2.ConfirmExecutionRequest)
@@ -126,3 +127,17 @@ class RecordExecutionsRequest(BaseSerDeModel):
 @proto_dataclass(execution_service_pb2.RecordExecutionsResponse)
 class RecordExecutionsResponse(BaseSerDeModel):
     records_stored: int = 0
+
+
+@proto_dataclass(execution_service_pb2.ResolveDeferredRelationsRequest)
+class ResolveDeferredRelationsRequest(BaseSerDeModel):
+    profile_name: str
+    target_name: str
+    project_name: str
+    project_id: t.Optional[str] = None
+    node_unique_ids: t.List[str] = field(default_factory=list)
+
+
+@proto_dataclass(execution_service_pb2.ResolveDeferredRelationsResponse)
+class ResolveDeferredRelationsResponse(BaseSerDeModel):
+    fqn_by_unique_id: t.Dict[str, str] = field(default_factory=dict)

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/services/explain_service_models.py

@@ -3,10 +3,11 @@ from __future__ import annotations
 import typing as t
 from dataclasses import field
 
+from query_cache_protobuf.query_cache.services import explain_service_pb2
+
 from query_cache_common.decorators import proto_dataclass, proto_enum
 from query_cache_common.models.base import BaseSerDeEnum, BaseSerDeModel
 from query_cache_common.models.shared_models import SubmitSQLResultType
-from query_cache_protobuf.query_cache.services import explain_service_pb2
 
 
 @proto_enum(explain_service_pb2.ExplainMarker)

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

@@ -3,6 +3,8 @@ from __future__ import annotations
 import typing as t
 from dataclasses import field
 
+from query_cache_protobuf.query_cache.services import sql_service_pb2
+
 from query_cache_common.decorators import proto_dataclass
 from query_cache_common.models import shared_models
 from query_cache_common.models._typing import TransformedNodesByQuery
@@ -12,7 +14,6 @@ from query_cache_common.models.fields import (
     transformed_nodes_by_query_field,
 )
 from query_cache_common.models.services.clone_service_models import TableProperties
-from query_cache_protobuf.query_cache.services import sql_service_pb2
 
 
 @proto_dataclass(sql_service_pb2.SubmitValuesRequest)

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/services/state_selector_service_models.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
 import typing as t
-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 BaseSerDeModel, BaseSerDeEnum
 
 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):

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/shared_models.py

@@ -2,10 +2,11 @@ from __future__ import annotations
 
 import typing as t
 
-from query_cache_common.decorators import proto_dataclass, proto_enum
-from query_cache_common.models.base import BaseSerDeModel, BaseSerDeEnum
 from query_cache_protobuf.query_cache import shared_pb2
 
+from query_cache_common.decorators import proto_dataclass, proto_enum
+from query_cache_common.models.base import BaseSerDeEnum, BaseSerDeModel
+
 
 @proto_enum(shared_pb2.ModelExecutionType)
 class ModelExecutionType(BaseSerDeEnum):
@@ -118,3 +119,4 @@ class DbtNodeState(BaseSerDeModel):
     node_contract_hash: t.Optional[str] = None
     node_persisted_descriptions_hash: t.Optional[str] = None
     project_id: t.Optional[str] = None
+    node_database_representation: t.Optional[str] = None

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/utils.py

@@ -1,14 +1,16 @@
 from __future__ import annotations
 
 import datetime
-import humanize
 import time
 import typing as t
 
+import humanize
 
 if t.TYPE_CHECKING:
     from sqlglot.expressions import (
         Create as CreateStatement,
+    )
+    from sqlglot.expressions import (
         Expression as SqlglotExpression,
     )
 

{query_cache_common-1.20.0 → query_cache_common-1.22.0}/src/query_cache_common/models/services/client_validation_service_models.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
+from query_cache_protobuf.query_cache.services import client_validation_service_pb2
 
 from query_cache_common.decorators import proto_dataclass
 from query_cache_common.models.base import BaseSerDeModel
-from query_cache_protobuf.query_cache.services import client_validation_service_pb2
 
 
 @proto_dataclass(client_validation_service_pb2.ValidateClientVersionRequest)