from __future__ import annotations

from enum import StrEnum


class NotificationType(StrEnum):
    """通知类型。"""

    SYSTEM = "system"
    CREDIT = "credit"
    VIDEO = "video"
    GENERATION = "generation"
    PAYMENT = "payment"
    RECHARGE = "recharge"
    UNKNOWN = "unknown"


NOTIFICATION_TYPE_LABELS = {
    NotificationType.SYSTEM.value: "系统通知",
    NotificationType.CREDIT.value: "积分通知",
    NotificationType.VIDEO.value: "视频通知",
    NotificationType.GENERATION.value: "生成通知",
    NotificationType.PAYMENT.value: "支付通知",
    NotificationType.RECHARGE.value: "充值通知",
    NotificationType.UNKNOWN.value: "未知通知",
}
