from datetime import datetime
from typing import Optional, List

from pydantic import BaseModel

from ...mods.rate_serve.schemas import RateServeQuery, RateServeInfo, RateServeId, RateServeInfoLite
from ...schemas import NodeState
from ...mods.rate_serve_status_log.schemas import RateServeStatusLogAllRes


class RateServeApplyReq(BaseModel):
    file_id: str


class RateServeApplyRes(RateServeId):
    pass


class RateServeQueryReq(RateServeQuery):
    pass


class RateServeQueryRes(BaseModel):
    count: int
    items: List[RateServeInfoLite]
    pass


class RateServeGetReq(RateServeId):
    pass


class RateServeGetRes(RateServeInfoLite):
    pass


class RateSheetUploadRes(BaseModel):
    id: str
    path: str
    type: Optional[str]
    sub_type: Optional[str]
    user_id: Optional[str]


class RateServeCancelReq(RateServeId):
    remark: Optional[str] = None


class RateServeStatusLogsGetReq(RateServeId):
    pass


class RateServeStatusLogsGetRes(RateServeStatusLogAllRes):
    pass