from datetime import datetime, date
from pydantic import BaseModel
from typing import Optional, List


class CompanyRateInfo(BaseModel):
    id: Optional[int] = None
    company_name: Optional[str] = None
    file_md: Optional[str] = None
    rate_time: Optional[datetime] = None
    rate_state: Optional[str] = None
    rate_level: Optional[str] = None
    badge: Optional[str] = None
    publish_state: Optional[str] = None
    rate_data: Optional[str] = None
    result_data: Optional[str] = None
    create_time: Optional[datetime] = None
    file_data: Optional[str] = None
    wj_data: Optional[str] = None
    file_data_state: Optional[str] = None
    api_data: Optional[str] = None
    api_data_state: Optional[str] = None
    man_data: Optional[str] = None
    man_data_state: Optional[str] = None
    mgr_data: Optional[str] = None
    mgr_data_state: Optional[str] = None
    rate_scope: Optional[float] = None


class CompanyRateReq(BaseModel):
    id: int


class CompanyRateAddInfo(BaseModel):
    company_name: Optional[str] = None
    file_md: Optional[str] = None
    rate_time: Optional[datetime] = None
    rate_state: Optional[str] = None
    rate_level: Optional[str] = None
    badge: Optional[str] = None
    publish_state: Optional[str] = None
    rate_data: Optional[str] = None
    result_data: Optional[str] = None
    create_time: Optional[datetime] = None
    file_data: Optional[str] = None
    wj_data: Optional[str] = None
    file_data_state: Optional[str] = None
    api_data: Optional[str] = None
    api_data_state: Optional[str] = None
    man_data: Optional[str] = None
    man_data_state: Optional[str] = None
    mgr_data: Optional[str] = None
    mgr_data_state: Optional[str] = None
    rate_scope: Optional[float] = None


class CompanyRateAddReq(CompanyRateAddInfo):
    pass

class GetReport(BaseModel):
    id:int
class CompanyRateFileDataLoadReq(BaseModel):
    id: int
    file_md: str


class CompanyRateAddRes(CompanyRateInfo):
    pass


class CompanyRateUpdateInfo(CompanyRateInfo):
    pass


class CompanyRateUpdateReq(CompanyRateUpdateInfo):
    pass


class CompanyRateUpdateRes(CompanyRateInfo):
    pass


class CompanyRateQuery(CompanyRateInfo):
    page: Optional[int] = None
    page_size: Optional[int] = None


class CompanyRateQueryReq(CompanyRateQuery):
    pass


class CompanyRateGetReq(BaseModel):
    id: int


class CompanyRateGetRes(CompanyRateInfo):
    pass


class CompanyRateQueryRes(BaseModel):
    count: int
    items: List[CompanyRateInfo]


class CompanyRateDeleteReq(BaseModel):
    id: int

#######################