Source code for svarog.checks

from typing import _GenericAlias  # type: ignore
from typing import Any
from typing import Dict
from typing import List
from typing import Mapping
from typing import MutableSequence
from typing import Sequence
from typing import Type
from typing import TypeVar
from typing import Union

from .compat import _SpecialGenericAlias
from .compat import get_args
from .compat import get_origin

T = TypeVar("T")


def has_annotated_init(type_: Type[T]) -> bool:
    return hasattr(type_, "__init__") and hasattr(type_.__init__, "__annotations__")


def is_union(type_: Union) -> bool:
    return (
        type_ is Union or isinstance(type_, _GenericAlias) and type_.__origin__ is Union
    )


def is_list(type_: Any) -> bool:
    return type_ is List or (
        isinstance(type_, (_GenericAlias, _SpecialGenericAlias))
        and issubclass(get_origin(type_), Sequence)  # type: ignore
    )


def is_bare(type_: Any) -> bool:
    args = get_args(type_)
    return (
        args == get_args(List)
        or args == get_args(Sequence)
        or args == get_args(Mapping)
        or args == get_args(Dict)
        or args == get_args(MutableSequence)
    )


def is_mapping(type_: Any) -> bool:
    return type_ is Mapping or (
        type_.__class__ is _GenericAlias and issubclass(type_.__origin__, Mapping)
    )