Bump mashumaro from 2.9 to 3.0.3 in /core (#5118)

* add pyyaml as a setup requirement

.changes/unreleased/Under the Hood-20220720-115226.yaml

@@ -0,0 +1,7 @@
+kind: Under the Hood
+body: Update mashumaro to 3.0.3
+time: 2022-07-20T11:52:26.210876-04:00
+custom:
+  Author: gshank
+  Issue: "4940"
+  PR: "5118"

core/dbt/contracts/graph/manifest.py

@@ -1,7 +1,7 @@
 import enum
 from dataclasses import dataclass, field
 from itertools import chain, islice
-from mashumaro import DataClassMessagePackMixin
+from mashumaro.mixins.msgpack import DataClassMessagePackMixin
 from multiprocessing.synchronize import Lock
 from typing import (
     Dict,

core/setup.py

@@ -56,7 +56,7 @@
         "hologram>=0.0.14,<=0.0.15",
         "isodate>=0.6,<0.7",
         "logbook>=1.5,<1.6",
-        "mashumaro==2.9",
+        "mashumaro[msgpack]==3.0.3",
         "minimal-snowplow-tracker==0.0.2",
         "networkx>=2.3,<2.8.1;python_version<'3.8'",
         "networkx>=2.3,<3;python_version>='3.8'",
@@ -69,6 +69,7 @@
         "requests<3.0.0",
         "idna>=2.5,<4",
         "cffi>=1.9,<2.0.0",
+        "pyyaml>=6.0",
     ],
     zip_safe=False,
     classifiers=[

third-party-stubs/mashumaro/__init__.pyi

@@ -1,5 +1,4 @@
 from mashumaro.exceptions import MissingField as MissingField
-from mashumaro.serializer.base.dict import DataClassDictMixin as DataClassDictMixin
-from mashumaro.serializer.json import DataClassJSONMixin as DataClassJSONMixin
-from mashumaro.serializer.msgpack import DataClassMessagePackMixin as DataClassMessagePackMixin
-from mashumaro.serializer.yaml import DataClassYAMLMixin as DataClassYAMLMixin
+from mashumaro.helper import field_options as field_options, pass_through as pass_through
+from mashumaro.mixins.dict import DataClassDictMixin as DataClassDictMixin
+from mashumaro.mixins.msgpack import DataClassMessagePackMixin as DataClassMessagePackMixin

third-party-stubs/mashumaro/config.pyi

@@ -1,10 +1,31 @@
-from mashumaro.types import SerializationStrategy as SerializationStrategy
-from typing import Any, Callable, Dict, List, Union
+from mashumaro.dialect import Dialect
+from mashumaro.types import SerializationStrategy
+from typing import Any, Callable, Dict, List, Optional, Type, Union
+from mashumaro.core.const import PEP_586_COMPATIBLE
 
+if PEP_586_COMPATIBLE:
+    from typing import Literal  # type: ignore
+else:
+    from typing_extensions import Literal  # type: ignore
+
+TO_DICT_ADD_BY_ALIAS_FLAG: str
 TO_DICT_ADD_OMIT_NONE_FLAG: str
+ADD_DIALECT_SUPPORT: str
+
+CodeGenerationOption = Literal[
+    "TO_DICT_ADD_BY_ALIAS_FLAG",
+    "TO_DICT_ADD_OMIT_NONE_FLAG",
+    "ADD_DIALECT_SUPPORT",
+]
+
 SerializationStrategyValueType = Union[SerializationStrategy, Dict[str, Union[str, Callable]]]
 
 class BaseConfig:
     debug: bool = ...
     code_generation_options: List[str] = ...
     serialization_strategy: Dict[Any, SerializationStrategyValueType] = ...
+    aliases: Dict[str, str] = ...
+    serialize_by_alias: bool = ...
+    namedtuple_as_dict: bool = ...
+    allow_postponed_evaluation: bool = ...
+    dialect: Optional[Type[Dialect]] = ...

third-party-stubs/mashumaro/core/const.pyi

@@ -0,0 +1,13 @@
+from typing import Any
+
+PY_36: Any
+PY_37: Any
+PY_38: Any
+PY_39: Any
+PY_310: Any
+PY_37_MIN: Any
+PY_38_MIN: Any
+PY_39_MIN: Any
+PY_310_MIN = PY_310
+PEP_585_COMPATIBLE = PY_39_MIN
+PEP_586_COMPATIBLE = PY_38_MIN

third-party-stubs/mashumaro/core/helpers.pyi

@@ -0,0 +1,3 @@
+import datetime
+
+def parse_timezone(s: str) -> datetime.timezone: ...

third-party-stubs/mashumaro/core/meta/builder.pyi

@@ -0,0 +1,115 @@
+from mashumaro.core.helpers import *
+import types
+import typing
+from base64 import decodebytes as decodebytes, encodebytes as encodebytes
+from dataclasses import Field
+from mashumaro.config import (
+    ADD_DIALECT_SUPPORT as ADD_DIALECT_SUPPORT,
+    BaseConfig as BaseConfig,
+    TO_DICT_ADD_BY_ALIAS_FLAG as TO_DICT_ADD_BY_ALIAS_FLAG,
+    TO_DICT_ADD_OMIT_NONE_FLAG as TO_DICT_ADD_OMIT_NONE_FLAG,
+)
+from mashumaro.core.const import PY_39_MIN as PY_39_MIN
+from mashumaro.core.meta.helpers import (
+    get_args as get_args,
+    get_class_that_defines_field as get_class_that_defines_field,
+    get_class_that_defines_method as get_class_that_defines_method,
+    get_literal_values as get_literal_values,
+    get_name_error_name as get_name_error_name,
+    get_type_origin as get_type_origin,
+    is_class_var as is_class_var,
+    is_dataclass_dict_mixin as is_dataclass_dict_mixin,
+    is_dataclass_dict_mixin_subclass as is_dataclass_dict_mixin_subclass,
+    is_dialect_subclass as is_dialect_subclass,
+    is_generic as is_generic,
+    is_init_var as is_init_var,
+    is_literal as is_literal,
+    is_named_tuple as is_named_tuple,
+    is_new_type as is_new_type,
+    is_optional as is_optional,
+    is_special_typing_primitive as is_special_typing_primitive,
+    is_type_var as is_type_var,
+    is_type_var_any as is_type_var_any,
+    is_typed_dict as is_typed_dict,
+    is_union as is_union,
+    not_none_type_arg as not_none_type_arg,
+    resolve_type_vars as resolve_type_vars,
+    type_name as type_name,
+)
+from mashumaro.core.meta.patch import patch_fromisoformat as patch_fromisoformat
+from mashumaro.dialect import Dialect as Dialect
+from mashumaro.exceptions import (
+    BadDialect as BadDialect,
+    BadHookSignature as BadHookSignature,
+    InvalidFieldValue as InvalidFieldValue,
+    MissingField as MissingField,
+    ThirdPartyModuleNotFoundError as ThirdPartyModuleNotFoundError,
+    UnresolvedTypeReferenceError as UnresolvedTypeReferenceError,
+    UnserializableDataError as UnserializableDataError,
+    UnserializableField as UnserializableField,
+    UnsupportedDeserializationEngine as UnsupportedDeserializationEngine,
+    UnsupportedSerializationEngine as UnsupportedSerializationEngine,
+)
+from mashumaro.helper import pass_through as pass_through
+from mashumaro.types import (
+    GenericSerializableType as GenericSerializableType,
+    SerializableType as SerializableType,
+    SerializationStrategy as SerializationStrategy,
+)
+from typing import Any
+
+NoneType: Any
+__PRE_SERIALIZE__: str
+__PRE_DESERIALIZE__: str
+__POST_SERIALIZE__: str
+__POST_DESERIALIZE__: str
+
+class CodeLines:
+    def __init__(self) -> None: ...
+    def append(self, line: str) -> None: ...
+    def indent(self) -> typing.Generator[None, None, None]: ...
+    def as_text(self) -> str: ...
+    def reset(self) -> None: ...
+
+class CodeBuilder:
+    cls: Any = ...
+    lines: Any = ...
+    globals: Any = ...
+    type_vars: Any = ...
+    field_classes: Any = ...
+    initial_arg_types: Any = ...
+    dialect: Any = ...
+    allow_postponed_evaluation: Any = ...
+    def __init__(
+        self,
+        cls: Any,
+        arg_types: typing.Tuple = ...,
+        dialect: typing.Optional[typing.Type[Dialect]] = ...,
+        first_method: str = ...,
+        allow_postponed_evaluation: bool = ...,
+    ) -> None: ...
+    def reset(self) -> None: ...
+    @property
+    def namespace(self) -> typing.Dict[typing.Any, typing.Any]: ...
+    @property
+    def annotations(self) -> typing.Dict[str, typing.Any]: ...
+    @property
+    def field_types(self) -> typing.Dict[str, typing.Any]: ...
+    @property
+    def dataclass_fields(self) -> typing.Dict[str, Field]: ...
+    @property
+    def metadatas(self) -> typing.Dict[str, typing.Mapping[str, typing.Any]]: ...
+    def get_field_default(self, name: str) -> typing.Any: ...
+    def ensure_module_imported(self, module: types.ModuleType) -> None: ...
+    def add_line(self, line: str) -> None: ...
+    def indent(self) -> typing.Generator[None, None, None]: ...
+    def compile(self) -> None: ...
+    def get_declared_hook(self, method_name: str) -> typing.Any: ...
+    def add_from_dict(self) -> None: ...
+    def get_config(self, cls: Any = ...) -> typing.Type[BaseConfig]: ...
+    def get_to_dict_flags(self, cls: Any = ...) -> str: ...
+    def get_from_dict_flags(self, cls: Any = ...) -> str: ...
+    def get_to_dict_default_flag_values(self, cls: Any = ...) -> str: ...
+    def get_from_dict_default_flag_values(self, cls: Any = ...) -> str: ...
+    def is_code_generation_option_enabled(self, option: str, cls: Any = ...) -> bool: ...
+    def add_to_dict(self) -> None: ...

third-party-stubs/mashumaro/core/meta/helpers.pyi

@@ -0,0 +1,37 @@
+import typing
+from typing import Any
+
+def get_type_origin(t: Any): ...
+def get_generic_name(t: Any, short: bool = ...) -> str: ...
+def get_args(t: typing.Any) -> typing.Tuple[typing.Any, ...]: ...
+def get_literal_values(t: typing.Any) -> Any: ...
+def type_name(
+    t: typing.Any,
+    short: bool = ...,
+    type_vars: typing.Dict[str, typing.Any] = ...,
+    is_type_origin: bool = ...,
+    none_type_as_none: bool = ...,
+) -> str: ...
+def is_special_typing_primitive(t: Any) -> bool: ...
+def is_generic(t: Any): ...
+def is_typed_dict(t: Any) -> bool: ...
+def is_named_tuple(t: Any) -> bool: ...
+def is_new_type(t: Any) -> bool: ...
+def is_union(t: Any): ...
+def is_optional(t: Any, type_vars: typing.Dict[str, typing.Any] = ...) -> bool: ...
+def is_annotated(t: Any) -> bool: ...
+def is_literal(t: Any) -> bool: ...
+def not_none_type_arg(
+    args: typing.Tuple[typing.Any, ...], type_vars: typing.Dict[str, typing.Any] = ...
+) -> Any: ...
+def is_type_var(t: Any) -> bool: ...
+def is_type_var_any(t: Any) -> bool: ...
+def is_class_var(t: Any) -> bool: ...
+def is_init_var(t: Any) -> bool: ...
+def get_class_that_defines_method(method_name: Any, cls: Any): ...
+def get_class_that_defines_field(field_name: Any, cls: Any): ...
+def is_dataclass_dict_mixin(t: Any) -> bool: ...
+def is_dataclass_dict_mixin_subclass(t: Any) -> bool: ...
+def resolve_type_vars(cls, arg_types: Any = ..., is_cls_created: bool = ...): ...
+def get_name_error_name(e: NameError) -> str: ...
+def is_dialect_subclass(t: Any) -> bool: ...

third-party-stubs/mashumaro/dialect.pyi

@@ -0,0 +1,7 @@
+from mashumaro.types import SerializationStrategy
+from typing import Any, Callable, Dict, Union
+
+SerializationStrategyValueType = Union[SerializationStrategy, Dict[str, Union[str, Callable]]]
+
+class Dialect:
+    serialization_strategy: Dict[Any, SerializationStrategyValueType] = ...

third-party-stubs/mashumaro/dialects/msgpack.pyi

@@ -0,0 +1,5 @@
+from mashumaro.dialect import Dialect
+from typing import Any
+
+class MessagePackDialect(Dialect):
+    serialization_strategy: Any = ...

third-party-stubs/mashumaro/exceptions.pyi

@@ -1,4 +1,4 @@
-from mashumaro.meta.helpers import type_name as type_name
+from mashumaro.core.meta.helpers import type_name as type_name
 from typing import Any, Optional
 
 class MissingField(LookupError):
@@ -26,6 +26,16 @@ class UnserializableField(UnserializableDataError):
     @property
     def holder_class_name(self): ...
 
+class UnsupportedSerializationEngine(UnserializableField):
+    def __init__(
+        self, field_name: Any, field_type: Any, holder_class: Any, engine: Any
+    ) -> None: ...
+
+class UnsupportedDeserializationEngine(UnserializableField):
+    def __init__(
+        self, field_name: Any, field_type: Any, holder_class: Any, engine: Any
+    ) -> None: ...
+
 class InvalidFieldValue(ValueError):
     field_name: Any = ...
     field_type: Any = ...
@@ -44,3 +54,22 @@ class InvalidFieldValue(ValueError):
     def field_type_name(self): ...
     @property
     def holder_class_name(self): ...
+
+class BadHookSignature(TypeError): ...
+
+class ThirdPartyModuleNotFoundError(ModuleNotFoundError):
+    module_name: Any = ...
+    field_name: Any = ...
+    holder_class: Any = ...
+    def __init__(self, module_name: Any, field_name: Any, holder_class: Any) -> None: ...
+    @property
+    def holder_class_name(self): ...
+
+class UnresolvedTypeReferenceError(NameError):
+    holder_class: Any = ...
+    name: Any = ...
+    def __init__(self, holder_class: Any, unresolved_type_name: Any) -> None: ...
+    @property
+    def holder_class_name(self): ...
+
+class BadDialect(ValueError): ...

third-party-stubs/mashumaro/helper.pyi

@@ -0,0 +1,24 @@
+from mashumaro.types import SerializationStrategy
+from typing import Any, Callable, Optional, Union
+from typing_extensions import Literal
+
+NamedTupleDeserializationEngine = Literal["as_dict", "as_list"]
+DateTimeDeserializationEngine = Literal["ciso8601", "pendulum"]
+AnyDeserializationEngine = Literal[NamedTupleDeserializationEngine, DateTimeDeserializationEngine]
+
+NamedTupleSerializationEngine = Literal["as_dict", "as_list"]
+AnySerializationEngine = NamedTupleSerializationEngine
+
+def field_options(
+    serialize: Optional[Union[AnySerializationEngine, Callable[[Any], Any]]] = ...,
+    deserialize: Optional[Union[AnyDeserializationEngine, Callable[[Any], Any]]] = ...,
+    serialization_strategy: Optional[SerializationStrategy] = ...,
+    alias: Optional[str] = ...,
+) -> Any: ...
+
+class _PassThrough(SerializationStrategy):
+    def __call__(self, *args: Any, **kwargs: Any) -> None: ...
+    def serialize(self, value: Any): ...
+    def deserialize(self, value: Any): ...
+
+pass_through: Any

third-party-stubs/mashumaro/mixins/dict.pyi

@@ -0,0 +1,15 @@
+from typing import Any, Dict, Mapping, Type, TypeVar
+
+T = TypeVar("T", bound="DataClassDictMixin")
+
+class DataClassDictMixin:
+    def __init_subclass__(cls: Type[T], **kwargs: Any) -> None: ...
+    def to_dict(self, omit_none: bool = True, validate: bool = False) -> dict: ...
+    @classmethod
+    def from_dict(cls, d: Mapping, validate=True) -> Any: ...
+    @classmethod
+    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: ...
+    def __post_serialize__(self, d: Dict[Any, Any]) -> Dict[Any, Any]: ...

third-party-stubs/mashumaro/mixins/json.pyi

@@ -0,0 +1,19 @@
+from mashumaro.mixins.dict import DataClassDictMixin as DataClassDictMixin
+from typing import Any, Dict, Type, TypeVar, Union
+from typing_extensions import Protocol as Protocol
+
+EncodedData = Union[str, bytes, bytearray]
+T = TypeVar("T", bound="DataClassJSONMixin")
+
+class Encoder:
+    def __call__(self, obj: Any, **kwargs: Any) -> EncodedData: ...
+
+class Decoder:
+    def __call__(self, s: EncodedData, **kwargs: Any) -> Dict[Any, Any]: ...
+
+class DataClassJSONMixin(DataClassDictMixin):
+    def to_json(self, encoder: Encoder = ..., **to_dict_kwargs: Any) -> EncodedData: ...
+    @classmethod
+    def from_json(
+        cls: Type[T], data: EncodedData, decoder: Decoder = ..., **from_dict_kwargs: Any
+    ) -> T: ...