DataMate/runtime/datamate-python/app/module/annotation/schema/mapping.py

from pydantic import Field, BaseModel
from typing import Optional, TYPE_CHECKING
from datetime import datetime

from app.module.shared.schema import BaseResponseModel
from app.module.shared.schema import StandardResponse

if TYPE_CHECKING:
    from .template import AnnotationTemplateResponse


class DatasetMappingCreateRequest(BaseModel):
    """数据集映射 创建 请求模型

    Accept both snake_case and camelCase field names from frontend JSON by
    declaring explicit aliases. Frontend sends `datasetId`, `name`,
    `description`, `templateId` (camelCase), so provide aliases so pydantic will map them
    to the internal attributes used in the service code (dataset_id, name,
    description, template_id).
    """
    dataset_id: str = Field(..., alias="datasetId", description="源数据集ID")
    name: Optional[str] = Field(None, alias="name", description="标注项目名称")
    description: Optional[str] = Field(None, alias="description", description="标注项目描述")
    template_id: Optional[str] = Field(None, alias="templateId", description="标注模板ID")
    label_config: Optional[str] = Field(None, alias="labelConfig", description="Label Studio XML配置")
    segmentation_enabled: Optional[bool] = Field(
        None,
        alias="segmentationEnabled",
        description="是否启用文本分段",
    )

    class Config:
        # allow population by field name when constructing model programmatically
        validate_by_name = True

class DatasetMappingCreateResponse(BaseResponseModel):
    """数据集映射 创建 响应模型"""
    id: str = Field(..., description="映射UUID")
    labeling_project_id: str = Field(..., description="Label Studio项目ID")
    labeling_project_name: str = Field(..., description="Label Studio项目名称")

class DatasetMappingUpdateRequest(BaseModel):
    """数据集映射 更新 请求模型
    
    支持更新的字段：
    - name: 标注项目名称
    - description: 标注项目描述
    - template_id: 标注模板ID
    - label_config: Label Studio XML配置
    """
    name: Optional[str] = Field(None, alias="name", description="标注项目名称")
    description: Optional[str] = Field(None, alias="description", description="标注项目描述")
    template_id: Optional[str] = Field(None, alias="templateId", description="标注模板ID")
    label_config: Optional[str] = Field(None, alias="labelConfig", description="Label Studio XML配置")

    class Config:
        validate_by_name = True

class DatasetMappingResponse(BaseModel):
    """数据集映射 查询 响应模型"""
    id: str = Field(..., description="映射UUID")
    dataset_id: str = Field(..., alias="datasetId", description="源数据集ID")
    dataset_name: Optional[str] = Field(None, alias="datasetName", description="数据集名称")
    dataset_type: Optional[str] = Field(None, alias="datasetType", description="数据集类型")
    labeling_project_id: str = Field(..., alias="labelingProjectId", description="标注项目ID")
    name: Optional[str] = Field(None, description="标注项目名称")
    description: Optional[str] = Field(None, description="标注项目描述")
    template_id: Optional[str] = Field(None, alias="templateId", description="关联的模板ID")
    labeling_type: Optional[str] = Field(None, alias="labelingType", description="标注类型")
    template: Optional['AnnotationTemplateResponse'] = Field(None, description="关联的标注模板详情")
    label_config: Optional[str] = Field(None, alias="labelConfig", description="实际使用的 Label Studio XML 配置")
    segmentation_enabled: Optional[bool] = Field(
        None,
        alias="segmentationEnabled",
        description="是否启用文本分段",
    )
    total_count: int = Field(0, alias="totalCount", description="数据集总数据量")
    annotated_count: int = Field(0, alias="annotatedCount", description="已标注数据量")
    in_progress_count: int = Field(0, alias="inProgressCount", description="分段标注中数据量")
    created_at: datetime = Field(..., alias="createdAt", description="创建时间")
    updated_at: Optional[datetime] = Field(None, alias="updatedAt", description="更新时间")
    deleted_at: Optional[datetime] = Field(None, alias="deletedAt", description="删除时间")

    class Config:
        from_attributes = True
        populate_by_name = True
        
class DeleteDatasetResponse(BaseResponseModel):
    """删除数据集响应模型"""
    id: str = Field(..., description="映射UUID")
    status: str = Field(..., description="删除状态")