from datetime import datetime

from pydantic import BaseModel


class SubscriptionCreate(BaseModel):
    tenant_id: str
    plan_id: str
    limits: dict | None = None
    status: str = "TRIAL"


class SubscriptionUpdate(BaseModel):
    plan_id: str | None = None
    limits: dict | None = None
    status: str | None = None
    ended_at: datetime | None = None


class SubscriptionResponse(BaseModel):
    id: str
    tenant_id: str
    plan_id: str
    limits: dict | None
    status: str
    started_at: datetime
    ended_at: datetime | None

    class Config:
        from_attributes = True