From 50e90226fa3b5719b41bdbc91f6f580a80694760 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=80=9D=E5=B7=9D?= <sichanwong0527@gmail.com>
Date: Sun, 20 Nov 2022 10:54:23 +0800
Subject: [PATCH] changes

---
 APPData/Router/CompanyRouter.py   | 16 +++++++++++++---
 APPData/Schemas/CompanySchemas.py |  6 +++++-
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/APPData/Router/CompanyRouter.py b/APPData/Router/CompanyRouter.py
index 6d18fb0..87d5a8b 100644
--- a/APPData/Router/CompanyRouter.py
+++ b/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}
diff --git a/APPData/Schemas/CompanySchemas.py b/APPData/Schemas/CompanySchemas.py
index 65b571f..9d79ce6 100644
--- a/APPData/Schemas/CompanySchemas.py
+++ b/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']
\ No newline at end of file
+    report_date: list = ['2021-12-31', '2020-12-31', '2019-12-31']
+
+
+class ViewCompanyByIDReqBody(BaseModel):
+    cid: str = None