changes
This commit is contained in:
parent
209a924a57
commit
50e90226fa
|
@ -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}
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue