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

from utils.pydantic_utils import AllOptional


class CompanySubmitInfo(BaseModel):
    id: Optional[int]
    company_name: Optional[str]
    file_md: Optional[str]
    create_time: Optional[datetime]


class CompanySubmitAddInfo(BaseModel):
    company_name: Optional[str]
    file_md: Optional[str]
    create_time: Optional[datetime]


class CompanySubmitAddReq(CompanySubmitAddInfo):
    pass


class CompanySubmitAddRes(CompanySubmitInfo):
    pass


class CompanySubmitUpdateInfo(CompanySubmitInfo):
    pass


class CompanySubmitUpdateReq(CompanySubmitUpdateInfo):
    pass


class CompanySubmitUpdateRes(CompanySubmitInfo):
    pass


class CompanySubmitInfoOptional(CompanySubmitInfo, metaclass=AllOptional):
    pass


class CompanySubmitQuery(CompanySubmitInfoOptional):
    pass


class CompanySubmitQueryReq(CompanySubmitQuery):
    pass


class CompanySubmitGetReq(BaseModel):
    id: int


class CompanySubmitGetRes(CompanySubmitInfo):
    pass


class CompanySubmitQueryRes(BaseModel):
    count: int
    items: List[CompanySubmitInfo]


class CompanySubmitDeleteReq(BaseModel):
    id: int

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