from pydantic import BaseModel


class CompanySearch(BaseModel):
    name: str = None
    page: int = 1
    pagesize: int = 10


class MacroSearch(BaseModel):
    provinces: str = '四川省'
    region: str = '成都市'
    year: int = 2021


class FinancialSearch(BaseModel):
    cid: str = 'CID00000002'
    report_date: list = ['2021-12-31', '2020-12-31', '2019-12-31']


class ViewCompanyByIDReqBody(BaseModel):
    cid: str = None