import os
from pydantic import BaseModel, Field
from typing import Optional, List
from datetime import datetime


class AlertCreate(BaseModel):
    camera_id: str = Field(..., description="摄像头标识")
    roi_id: Optional[str] = Field(None, description="ROI区域标识")
    alert_type: str = Field(..., description="告警类型")
    algorithm: Optional[str] = Field(None, description="算法名称")
    confidence: Optional[int] = Field(None, ge=0, le=100, description="置信度")
    duration_minutes: Optional[int] = Field(None, ge=0, description="离岗时长(分钟)")
    trigger_time: datetime = Field(..., description="触发时间")
    message: Optional[str] = Field(None, description="告警消息")


class AlertResponse(BaseModel):
    id: int
    alert_no: str
    camera_id: str
    roi_id: Optional[str]
    alert_type: str
    algorithm: Optional[str]
    confidence: Optional[int]
    duration_minutes: Optional[int]
    trigger_time: Optional[str]
    message: Optional[str]
    snapshot_url: Optional[str]
    status: str
    handle_remark: Optional[str]
    handled_by: Optional[str]
    handled_at: Optional[str]
    ai_analysis: Optional[dict]
    created_at: Optional[str]
    updated_at: Optional[str]

    class Config:
        from_attributes = True


class AlertListResponse(BaseModel):
    alerts: List[AlertResponse]
    total: int
    page: int
    page_size: int


class AlertHandleRequest(BaseModel):
    status: str = Field(..., description="处理状态: confirmed/ignored/resolved")
    remark: Optional[str] = Field(None, description="处理备注")


class AlertStatisticsResponse(BaseModel):
    total: int
    pending: int
    confirmed: int
    ignored: int
    resolved: int
    by_type: dict
    by_level: dict


class HealthResponse(BaseModel):
    status: str
    database: str