from enum import Enum

from pydantic import BaseModel

from config import settings


class SubscriptionDuration(Enum):
    MONTH = "month"
    THREE_MONTHS = "three_months"
    SIX_MONTHS = "six_months"
    YEAR = "year"


class SubscriptionPlan(BaseModel):
    devices: int = settings.min_devices
    duration: SubscriptionDuration = SubscriptionDuration.MONTH
    whitelists: bool = False