utils/tags.py

"""Module with Airflow tag definitions."""

from enum import Enum, member


class InvalidTagError(AttributeError):
    pass


class Tag(Enum):
    """Enum containing available Airflow tags."""

    def __getattr__(self, item: str) -> str:
        """
        Simplifies accessing enum values.

        Instead of Tag.ImpactTier.value.tier_1.value we can
        just use Tag.ImpactTier.tier_1.
        Simplify accessing enum values.

        Instead of Tag.ImpactTier.value.tier_1.value we can just use
        Tag.ImpactTier.tier_1.

        # source: https://newbedev.com/enum-of-enums-in-python
        """

        if item == "_value_":
            raise InvalidTagError

        try:
            ret_val = getattr(self.value, item).value
        except AttributeError as _err:
            raise InvalidTagError() from _err

        return ret_val

    @member
    class ImpactTier(Enum):
        """Valid options for Impact tier tag."""

        tier_1: str = "impact/tier_1"
        tier_2: str = "impact/tier_2"
        tier_3: str = "impact/tier_3"

    @member
    class Triage(Enum):
        """Tag for conveying information to the engineer on triage."""

        confidential: str = "triage/confidential"
        record_only: str = "triage/record_only"
        no_triage: str = "triage/no_triage"

    @member
    class Repo(Enum):
        """Valid options for Repo tag."""

        bqetl: str = "repo/bigquery-etl"
        airflow: str = "repo/telemetry-airflow"
        private_bqetl: str = "repo/private-bigquery-etl"