from pydantic import BaseModel
from typing import Optional, List


class QueryStatsData(BaseModel):
    total: int = 0
    avg_time_ms: int = 0
    slow_count: int = 0
    type_distribution: List[dict] = []


class DashboardStats(BaseModel):
    ticket_total: int = 0
    ticket_pending: int = 0
    ticket_month: int = 0
    ticket_executing: int = 0
    instance_total: int = 0
    instance_online: int = 0
    user_total: int = 0
    queries_today: int = 0
    query_stats: Optional[QueryStatsData] = None


class TrendPoint(BaseModel):
    date: str
    submitted: int = 0
    success: int = 0
    failed: int = 0


class TrendData(BaseModel):
    items: List[TrendPoint] = []