dbt-common 1.0.3__tar.gz → 1.0.4__tar.gz

{dbt_common-1.0.3 → dbt_common-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dbt-common
-Version: 1.0.3
+Version: 1.0.4
 Summary: The shared common utilities that dbt-core and adapter implementations use
 Project-URL: Homepage, https://github.com/dbt-labs/dbt-common
 Project-URL: Repository, https://github.com/dbt-labs/dbt-common.git
@@ -30,7 +30,7 @@ Requires-Dist: isodate<0.7,>=0.6
 Requires-Dist: jinja2<4,>=3.1.3
 Requires-Dist: jsonschema<5.0,>=4.0
 Requires-Dist: mashumaro[msgpack]<4.0,>=3.9
-Requires-Dist: pathspec<0.12,>=0.9
+Requires-Dist: pathspec<0.13,>=0.9
 Requires-Dist: protobuf<5.0.0,>=4.0.0
 Requires-Dist: python-dateutil<3.0,>=2.0
 Requires-Dist: requests<3.0.0

dbt_common-1.0.4/dbt_common/__about__.py

@@ -0,0 +1 @@
+version = "1.0.4"

{dbt_common-1.0.3 → dbt_common-1.0.4}/dbt_common/contracts/config/properties.py

@@ -1,10 +1,10 @@
 from dataclasses import dataclass, field
-from typing import Dict, Any
+from typing import Dict, Any, Optional
 
-from dbt_common.dataclass_schema import ExtensibleDbtClassMixin
+from dbt_common.dataclass_schema import ExtensibleDbtClassMixin, dbtClassMixin
 
 
-class AdditionalPropertiesMixin:
+class AdditionalPropertiesMixin(dbtClassMixin):
     """Make this class an extensible property.
 
     The underlying class definition must include a type definition for a field
@@ -41,8 +41,8 @@ class AdditionalPropertiesMixin:
         data = super().__pre_deserialize__(data)
         return data
 
-    def __post_serialize__(self, dct):
-        data = super().__post_serialize__(dct)
+    def __post_serialize__(self, dct: Dict, context: Optional[Dict] = None):
+        data = super().__post_serialize__(dct, context)
         data.update(self.extra)
         if "_extra" in data:
             del data["_extra"]

{dbt_common-1.0.3 → dbt_common-1.0.4}/dbt_common/dataclass_schema.py

@@ -7,11 +7,17 @@ from datetime import datetime
 from dateutil.parser import parse
 
 # type: ignore
-from mashumaro import DataClassDictMixin
-from mashumaro.config import TO_DICT_ADD_OMIT_NONE_FLAG, BaseConfig as MashBaseConfig
+from mashumaro.config import (
+    TO_DICT_ADD_OMIT_NONE_FLAG,
+    ADD_SERIALIZATION_CONTEXT,
+    BaseConfig as MashBaseConfig,
+)
 from mashumaro.types import SerializableType, SerializationStrategy
 from mashumaro.jsonschema import build_json_schema
 
+# following includes DataClassDictMixin
+from mashumaro.mixins.msgpack import DataClassMessagePackMixin
+
 import functools
 
 
@@ -34,6 +40,7 @@ class DateTimeSerialization(SerializationStrategy):
 class dbtMashConfig(MashBaseConfig):
     code_generation_options = [
         TO_DICT_ADD_OMIT_NONE_FLAG,
+        ADD_SERIALIZATION_CONTEXT,
     ]
     serialization_strategy = {
         datetime: DateTimeSerialization(),
@@ -47,7 +54,8 @@ class dbtMashConfig(MashBaseConfig):
 
 # This class pulls in DataClassDictMixin from Mashumaro. The 'to_dict'
 # and 'from_dict' methods come from Mashumaro.
-class dbtClassMixin(DataClassDictMixin):
+# Note: DataClassMessagePackMixin inherits from DataClassDictMixin
+class dbtClassMixin(DataClassMessagePackMixin):
     """Convert and validate JSON schemas.
 
     The Mixin adds methods to generate a JSON schema and
@@ -73,7 +81,7 @@ class dbtClassMixin(DataClassDictMixin):
     # This is called by the mashumaro to_dict in order to handle
     # nested classes. We no longer do any munging here, but leaving here
     # so that subclasses can leave super() in place for possible future needs.
-    def __post_serialize__(self, data):
+    def __post_serialize__(self, data, context: Optional[Dict]):
         return data
 
     @classmethod

{dbt_common-1.0.3 → dbt_common-1.0.4}/pyproject.toml

@@ -34,7 +34,7 @@ dependencies = [
   "jsonschema>=4.0,<5.0",
   "Jinja2>=3.1.3,<4",
   "mashumaro[msgpack]>=3.9,<4.0",
-  "pathspec>=0.9,<0.12",
+  "pathspec>=0.9,<0.13",
   "protobuf>=4.0.0,<5.0.0",
   "python-dateutil>=2.0,<3.0",
   "requests<3.0.0",  # needs to match dbt-core

{dbt_common-1.0.3 → dbt_common-1.0.4}/third-party-stubs/mashumaro/jsonschema/models.pyi

@@ -106,8 +106,8 @@ class JSONSchema(DataClassJSONMixin):
         serialize_by_alias: bool
         aliases: Incomplete
         serialization_strategy: Incomplete
-    def __pre_serialize__(self) -> JSONSchema: ...
-    def __post_serialize__(self, d: Dict[Any, Any]) -> Dict[Any, Any]: ...
+    def __pre_serialize__(self, context: Optional[Dict]) -> JSONSchema: ...
+    def __post_serialize__(self, d: Dict[Any, Any], context: Optional[Dict]) -> Dict[Any, Any]: ...
     def __init__(
         self,
         schema,

{dbt_common-1.0.3 → dbt_common-1.0.4}/third-party-stubs/mashumaro/mixins/dict.pyi

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Mapping, Type, TypeVar
+from typing import Any, Dict, Mapping, Type, TypeVar, Optional
 
 T = TypeVar("T", bound="DataClassDictMixin")
 
@@ -11,5 +11,5 @@ class DataClassDictMixin:
     def __pre_deserialize__(cls: Type[T], d: Dict[Any, Any]) -> Dict[Any, Any]: ...
     @classmethod
     def __post_deserialize__(cls: Type[T], obj: T) -> T: ...
-    def __pre_serialize__(self: T) -> T: ...
-    def __post_serialize__(self, d: Dict[Any, Any]) -> Dict[Any, Any]: ...
+    def __pre_serialize__(self: T, context: Optional[Dict]) -> T: ...
+    def __post_serialize__(self, d: Dict[Any, Any], context: Optional[Dict]) -> Dict[Any, Any]: ...

dbt_common-1.0.3/dbt_common/__about__.py

@@ -1 +0,0 @@
-version = "1.0.3"