structured_data._adt.annotations

Complexity

Total Complexity

10

Size/Duplication

Total Lines	41
Duplicated Lines	0 %

Importance

Changes

0

3 Functions

Rating	Name	Duplication	Size	Complexity
A	sum_args_from_annotations()	0	8	2
A	product_args_from_annotations()	0	14	5
A	_all_annotations()	0	4	3

"""Helper functions for processing annotations."""

import inspect
import sys
import typing

from .. import _nillable_write
from . import ctor


def _all_annotations(cls: type) -> typing.Iterator[typing.Tuple[type, str, typing.Any]]:
    for superclass in reversed(cls.__mro__):
        for key, value in vars(superclass).get("__annotations__", {}).items():
            yield (superclass, key, value)


def sum_args_from_annotations(cls: type) -> typing.Dict[str, typing.Tuple]:
    """Return the constructor data for Sum classes."""
    args: typing.Dict[str, typing.Tuple] = {}
    for superclass, key, value in _all_annotations(cls):
        _nillable_write.nillable_write(
            args, key, ctor.get_args(value, vars(sys.modules[superclass.__module__]))
        )
    return args


def product_args_from_annotations(cls: type) -> typing.Dict[str, typing.Any]:
    """Return the field data for Product classes."""
    args: typing.Dict[str, typing.Any] = {}
    for superclass, key, value in _all_annotations(cls):
        if (
            value == "None"
            or ctor.annotation_is_classvar(
                value, vars(sys.modules[superclass.__module__])
            )
            or inspect.isdatadescriptor(inspect.getattr_static(cls, key, None))
        ):
            value = None
        _nillable_write.nillable_write(args, key, value)
    return args