from pydantic import BaseModel
from datetime import date, datetime
from typing import List, Optional


class DailyRegisteredUsers(BaseModel):
    """Daily registered users data model"""
    date_id: date
    product_id: str = "freeleaps"
    registered_cnt: int
    updated_at: Optional[datetime] = None


class UserRegistrationQuery(BaseModel):
    """Query parameters for user registration data"""
    start_date: date
    end_date: date
    product_id: str = "freeleaps"


class UserRegistrationResponse(BaseModel):
    """Response model for user registration data"""
    dates: List[str]
    counts: List[int]
    total_registrations: int
    query_period: str