from datetime import datetime

from pydantic import BaseModel, Field


class SubscriptionData(BaseModel):
    devices: int | None = Field(None)
    expires_at: datetime | None = Field(None)
    addon_ids: list[str] = Field(default_factory=list)
    subscription_link: str | None = Field(None)
    has_subscription: bool = Field()


class UserInfo(BaseModel):
    username: str | None = Field(None)
    telegram_id: str | None = Field(None)
    referal_code: str = Field()