from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session

from .Mods.CompanyBaseInfo import Router as CompanyBaseInfoRouter
from .Mods.CompanyControl import Router as CompanyControlRouter
from .Mods.CompanyPlusProperty import Router as CompanyPlusPropertyRouter
from .Mods.CompanyIndexMain import Router as CompanyIndexMainRouter
from .Mods.CompanyShareholder import Router as CompanyShareholderRouter
from .Mods.CompanyEquityPenetration import Router as CompanyEquityPenetrationRouter
from Context.common import auth_util, common_db
from . import Crud, Schemas

router = APIRouter(
    prefix="/company_base_info",
    dependencies=[Depends(auth_util.token_data_depend)]
)
router.include_router(CompanyBaseInfoRouter.router)
router.include_router(CompanyControlRouter.router)
router.include_router(CompanyPlusPropertyRouter.router)
router.include_router(CompanyIndexMainRouter.router)
router.include_router(CompanyShareholderRouter.router)
router.include_router(CompanyEquityPenetrationRouter.router)


@router.post("/company_all_info_query", summary="城投详细信息联合查询", tags=["城投详细信息"],
             response_model=Schemas.CompanyAllInfoQueryRes)
def company_all_info_query(req: Schemas.CompanyAllInfoQueryReq, db: Session = Depends(common_db.get_db)):
    count, items = Crud.company_all_info_query(db, req)
    return Schemas.CompanyAllInfoQueryRes(count=count, items=items)