diff --git a/App/Crud/RatingFlowCrud.py b/App/Crud/RatingFlowCrud.py index 7bdade2..cddd37d 100644 --- a/App/Crud/RatingFlowCrud.py +++ b/App/Crud/RatingFlowCrud.py @@ -63,7 +63,8 @@ def get_rating_flow_all_nodes(_id: str, db: Session, mongodb: MongoHelper): __dict.update({"node_num": obj.node_num}) __dict.update({"node_name": obj.node_name}) __dict.update({"status": obj.status}) - __dict.update({"content": mongodb.find_data_by_id(dbname="打分评级流程", sheet=obj.node_name, _id=obj.obj_id)}) + __dict.update( + {"content": mongodb.find_data_by_id(dbname="打分评级流程", sheet=obj.node_name, _id=obj.obj_id)}) __list.append(__dict) return __list diff --git a/App/Router/RatingFlowRouter.py b/App/Router/RatingFlowRouter.py index 401908b..6d939e4 100644 --- a/App/Router/RatingFlowRouter.py +++ b/App/Router/RatingFlowRouter.py @@ -117,6 +117,16 @@ def func(schema: RatingFlowSchema.CreateRatingFlowReqBodys, token: str = Header( return {"Info": "Success"} +@router.post("/search", summary="根据评级ID查询", tags=["评级流程"]) +def func(schema: RatingFlowSchema.SearchFlowReqBody, db: Session = Depends(get_db)): + # 获取流程; rf_item: 流程对象 + rf_item = Crud.get_rating_flow(db=db, rating_flow_id=schema.id) + if not rf_item: + raise HTTPException(status_code=404, detail="评级流程不存在") + + return rf_item + + @router.post("/add_participant", summary='新增参与人', tags=["评级流程"]) def func(schema: RatingFlowSchema.AddParticipantReqBody, db: Session = Depends(get_db)): rf_item = Crud.get_rating_flow(db=db, rating_flow_id=schema.rating_flow_id) diff --git a/App/Schemas/RatingFlowSchema.py b/App/Schemas/RatingFlowSchema.py index be7a86b..07b9bc7 100644 --- a/App/Schemas/RatingFlowSchema.py +++ b/App/Schemas/RatingFlowSchema.py @@ -108,3 +108,7 @@ class ChangeStatusReqBody(BaseModel): class RatingFlowListReqBody(BaseModel): page: int = 1 pagesize: int = 10 + + +class SearchFlowReqBody(BaseModel): + id: str = "评级流程ID"