"""标注平台数据模型。"""

from pydantic import BaseModel


class AnnotationProjectItem(BaseModel):
    project_id: str
    project_name: str
    description: str = ""
    project_type: str = ""       # "image" | "text"
    task_type: str = ""
    status: str = ""
    created_by: str = ""
    created_at: str = ""
    updated_at: str = ""
    task_count: int = 0
    completed_task_count: int = 0


class AnnotationProjectListResponse(BaseModel):
    items: list[AnnotationProjectItem]
    total: int
    page: int
    page_size: int


class AnnotationProjectDetailResponse(BaseModel):
    project_id: str
    project_name: str
    description: str = ""
    project_type: str = ""
    task_type: str = ""
    status: str = ""
    created_by: str = ""
    created_at: str = ""
    updated_at: str = ""
    task_count: int = 0
    completed_task_count: int = 0
    assigned_task_count: int = 0
    completion_percentage: float = 0.0


class ProjectImportResponse(BaseModel):
    project_id: str
    project_name: str
    format: str
    total_exported: int
    dataset_id: str
    dataset_name: str