import enum
from typing import Literal


class MessageChannel(enum.StrEnum):
    SMS = "SMS"
    WHATSAPP = "WHATSAPP"

    def as_literal(self) -> Literal["SMS", "WHATSAPP"]:
        return "SMS" if self == MessageChannel.SMS else "WHATSAPP"


class CampaignStatus(enum.StrEnum):
    DRAFT = "DRAFT"
    SCHEDULED = "SCHEDULED"
    IN_PROGRESS = "IN_PROGRESS"
    CANCELLED = "CANCELLED"
    SENT = "SENT"


class CampaignCancelReason(enum.StrEnum):
    NO_FANS = "NO_FANS"
    CHANNEL_NOT_ALLOWED = "CHANNEL_NOT_ALLOWED"