from pydantic import BaseModel
from typing import List, Optional
from .Mods.CompanyIndexMain.Schemas import CompanyIndexMainInfo
from .Mods.CompanyBaseInfo.Schemas import CompanyBaseInfoInfo
from .Mods.CompanyPlusProperty.Schemas import CompanyPlusPropertyInfo
from .Mods.CompanyControl.Schemas import CompanyControlInfo
from Utils.SqlAlchemyUtils import QueryParam


class CompanyAllInfo(BaseModel):
    company_id: int
    company_index_main: Optional[CompanyIndexMainInfo]
    company_base_info: Optional[CompanyBaseInfoInfo]
    company_plus_property: Optional[CompanyPlusPropertyInfo]
    company_control: Optional[CompanyControlInfo]


class CompanyAllInfoQueryParams(BaseModel):
    company_index_main: Optional[List[QueryParam]]
    company_base_info: Optional[List[QueryParam]]
    company_plus_property: Optional[List[QueryParam]]
    company_control: Optional[List[QueryParam]]
    page: int
    page_size: int


class CompanyAllInfoQueryReq(CompanyAllInfoQueryParams):
    pass


class CompanyAllInfoQueryRes(BaseModel):
    count: int
    items: List[CompanyAllInfo]