Module dataclasses_json.cfg

View Source
import functools

from typing import Callable, Dict, Optional, TypeVar, Union

from marshmallow.fields import Field as MarshmallowField

from dataclasses_json.undefined import Undefined, UndefinedParameterError

T = TypeVar("T")

class Exclude:

    """

    Pre-defined constants for exclusion. By default, fields are configured to

    be included.

    """

    ALWAYS: Callable[[T], bool] = lambda _: True

    NEVER: Callable[[T], bool] = lambda _: False

# TODO: add warnings?

class _GlobalConfig:

    def __init__(self):

        self.encoders: Dict[type, Callable] = {}

        self.decoders: Dict[type, Callable] = {}

        self.mm_fields: Dict[type, MarshmallowField] = {}

        # self._json_module = json

    # TODO: #180

    # @property

    # def json_module(self):

    #     return self._json_module

    #

    # @json_module.setter

    # def json_module(self, value):

    #     warnings.warn(f"Now using {value.__name__} module to handle JSON. "

    #                   f"{self._disable_msg}")

    #     self._json_module = value

global_config = _GlobalConfig()

def config(metadata: dict = None, *,

           # TODO: these can be typed more precisely

           # Specifically, a Callable[A, B], where `B` is bound as a JSON type

           encoder: Callable = None,

           decoder: Callable = None,

           mm_field: MarshmallowField = None,

           letter_case: Callable[[str], str] = None,

           undefined: Optional[Union[str, Undefined]] = None,

           field_name: str = None,

           exclude: Optional[Callable[[str, T], bool]] = None,

           ) -> Dict[str, dict]:

    if metadata is None:

        metadata = {}

    lib_metadata = metadata.setdefault('dataclasses_json', {})

    if encoder is not None:

        lib_metadata['encoder'] = encoder

    if decoder is not None:

        lib_metadata['decoder'] = decoder

    if mm_field is not None:

        lib_metadata['mm_field'] = mm_field

    if field_name is not None:

        if letter_case is not None:

            @functools.wraps(letter_case)

            def override(_, _letter_case=letter_case, _field_name=field_name):

                return _letter_case(_field_name)

        else:

            def override(_, _field_name=field_name):

                return _field_name

        letter_case = override

    if letter_case is not None:

        lib_metadata['letter_case'] = letter_case

    if undefined is not None:

        # Get the corresponding action for undefined parameters

        if isinstance(undefined, str):

            if not hasattr(Undefined, undefined.upper()):

                valid_actions = list(action.name for action in Undefined)

                raise UndefinedParameterError(

                    f"Invalid undefined parameter action, "

                    f"must be one of {valid_actions}")

            undefined = Undefined[undefined.upper()]

        lib_metadata['undefined'] = undefined

    if exclude is not None:

        lib_metadata['exclude'] = exclude

    return metadata

Variables

T
global_config

Functions

config

def config(
    metadata: dict = None,
    *,
    encoder: Callable = None,
    decoder: Callable = None,
    mm_field: marshmallow.fields.Field = None,
    letter_case: Callable[[str], str] = None,
    undefined: Union[str, dataclasses_json.undefined.Undefined, NoneType] = None,
    field_name: str = None,
    exclude: Union[Callable[[str, ~T], bool], NoneType] = None
) -> Dict[str, dict]
View Source
def config(metadata: dict = None, *,

           # TODO: these can be typed more precisely

           # Specifically, a Callable[A, B], where `B` is bound as a JSON type

           encoder: Callable = None,

           decoder: Callable = None,

           mm_field: MarshmallowField = None,

           letter_case: Callable[[str], str] = None,

           undefined: Optional[Union[str, Undefined]] = None,

           field_name: str = None,

           exclude: Optional[Callable[[str, T], bool]] = None,

           ) -> Dict[str, dict]:

    if metadata is None:

        metadata = {}

    lib_metadata = metadata.setdefault('dataclasses_json', {})

    if encoder is not None:

        lib_metadata['encoder'] = encoder

    if decoder is not None:

        lib_metadata['decoder'] = decoder

    if mm_field is not None:

        lib_metadata['mm_field'] = mm_field

    if field_name is not None:

        if letter_case is not None:

            @functools.wraps(letter_case)

            def override(_, _letter_case=letter_case, _field_name=field_name):

                return _letter_case(_field_name)

        else:

            def override(_, _field_name=field_name):

                return _field_name

        letter_case = override

    if letter_case is not None:

        lib_metadata['letter_case'] = letter_case

    if undefined is not None:

        # Get the corresponding action for undefined parameters

        if isinstance(undefined, str):

            if not hasattr(Undefined, undefined.upper()):

                valid_actions = list(action.name for action in Undefined)

                raise UndefinedParameterError(

                    f"Invalid undefined parameter action, "

                    f"must be one of {valid_actions}")

            undefined = Undefined[undefined.upper()]

        lib_metadata['undefined'] = undefined

    if exclude is not None:

        lib_metadata['exclude'] = exclude

    return metadata

Classes

Exclude

class Exclude(
    /,
    *args,
    **kwargs
)

Pre-defined constants for exclusion. By default, fields are configured to be included.

View Source
class Exclude:

    """

    Pre-defined constants for exclusion. By default, fields are configured to

    be included.

    """

    ALWAYS: Callable[[T], bool] = lambda _: True

    NEVER: Callable[[T], bool] = lambda _: False

Methods

ALWAYS
def ALWAYS(
    _
)
View Source
    ALWAYS: Callable[[T], bool] = lambda _: True
NEVER
def NEVER(
    _
)
View Source
    NEVER: Callable[[T], bool] = lambda _: False