Source code for czech_plus.models

"""Module for our models."""
import enum
import typing


[docs]class MultiValueEnum(enum.Enum):
    """Base class for Enum with multiple values."""

    def __new__(cls, *values):
        """Construct new instance of the class."""
        obj = object.__new__(cls)
        for other_value in values:
            cls._value2member_map_[other_value] = obj
        return obj

[docs]    def __repr__(self):
        """Nice string representation of the class."""
        return "<%s.%s: %s>" % (
            self.__class__.__name__,
            self._name_,
            ", ".join([repr(v) for v in self._value_]),
        )


@enum.unique
[docs]class Gender(enum.Enum):
    """Gender of the noun."""

[docs]    M = "ten"
    """Masculine."""
[docs]    F = "ta"
    """Female."""
[docs]    N = "to"
    """Neuter."""
[docs]    mM = "mM"
    """Masculine plural."""
[docs]    mF = "mF"
    """Female plural."""
[docs]    mN = "mN"
    """Neuter plural."""
    # Special genders
[docs]    A = "ten/ta"
    """Animate (masculine/female).

    Look at ``recepční`` gender.
    """


@enum.unique
[docs]class Case(MultiValueEnum):
    """Case of the noun word.

    Value of the item is question, you can also use numbers (from 1 to 7) as aliases.
    """

[docs]    nominative = "kdo? co?", 1
[docs]    genitive = "koho? čeho?", 2
[docs]    dative = "komu? čemu?", 3
[docs]    accusative = "koho? co?", 4
[docs]    vocative = "voláme", 5
[docs]    locative = "kom? čem?", 6
[docs]    instrumental = "kým? čím?", 7

    @property
[docs]    def questions(self) -> str:
        r"""Get questions of the case.

        Just an alias to :attr:`Case.value`\ [0].
        """
        return typing.cast(str, self.value[0])

    @property
[docs]    def number(self) -> int:
        r"""Get number of the case.

        Just an alias to :attr:`Case.value`\ [1].
        """
        return typing.cast(int, self.value[1])