from pydantic import BaseModel
from typing import Optional, List

class WorkAreaCreate(BaseModel):
    """工区创建"""
    department_id: str
    parent_id: Optional[str] = None
    type: Optional[str] = None
    name: Optional[str] = None

class WorkAreaResponse(BaseModel):
    """工区响应"""
    id: int
    department_id: Optional[str] = None
    parent_id: Optional[str] = None
    type: Optional[str] = None
    name: Optional[str] = None

class WorkAreaBatchImport(BaseModel):
    """工区批量导入"""
    account_id: int
    data: List[WorkAreaCreate]

class WorkAreaQuery(BaseModel):
    """工区查询参数"""
    account_id: int
    department_id: Optional[str] = None
    parent_id: Optional[str] = None
    type: Optional[str] = None
    name: Optional[str] = None
    page: int = 1
    page_size: int = 20