from pydantic import BaseModel
from typing import Optional, Dict, Any

# 导出Excel请求
class ExportExcelRequest(BaseModel):
    section_id: Optional[str] = None
    account_id: Optional[str] = None
    mileage: Optional[str] = None
    work_site: Optional[str] = None
    number: Optional[str] = None
    status: Optional[str] = None

# 导出沉降数据请求
class ExportSettlementRequest(BaseModel):
    project_name: str  # 项目名称（标段）

# 导出水准数据请求（以水准数据为主体）
class ExportLevelDataRequest(BaseModel):
    project_name: str  # 项目名称（标段）

# 导出Excel响应
class ExportExcelResponse(BaseModel):
    code: int
    message: str
    data: Optional[Dict[str, Any]] = None