changes

APPData/Router/CompanyRouter.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, Depends
+from fastapi import APIRouter, Depends, HTTPException
 from sqlalchemy.orm import Session
 
 from APPData.Crud import CompanyCrud
@@ -14,7 +14,7 @@ router = APIRouter(
 
 
 @router.get("/create", summary="新建企业")
-def func01(name: str, db: Session = Depends(get_db)):
+def func(name: str, db: Session = Depends(get_db)):
     # 根据name查询企业是否存在
     data = CompanyCrud.get_company_by_name(db=db, name=name)
     if not data:
@@ -24,6 +24,16 @@ def func01(name: str, db: Session = Depends(get_db)):
 
 
 @router.post("/search", summary="查询企业")
-def func02(req_body: CompanySchemas.CompanySearch, db: Session = Depends(get_db)):
+def func(req_body: CompanySchemas.CompanySearch, db: Session = Depends(get_db)):
     company = CompanyCrud.get_company_total(db, name=req_body.name, page=req_body.page, pagesize=req_body.pagesize)
     return company
+
+
+@router.post("/search_name_by_cid", summary="根据企业ID查询企业名称")
+def func(schema: CompanySchemas.ViewCompanyByIDReqBody, db: Session = Depends(get_db)):
+    item = CompanyCrud.get_company_by_id(db=db, _id=schema.cid)
+
+    if not item:
+        raise HTTPException(status_code=404, detail="Company Not Found")
+
+    return {"company_name": item.name}

APPData/Schemas/CompanySchemas.py

@@ -15,4 +15,8 @@ class MacroSearch(BaseModel):
 
 class FinancialSearch(BaseModel):
     cid: str = 'CID00000002'
-    report_date: list = ['2021-12-31', '2020-12-31', '2019-12-31']
+    report_date: list = ['2021-12-31', '2020-12-31', '2019-12-31']
+
+
+class ViewCompanyByIDReqBody(BaseModel):
+    cid: str = None