changes 更改节点状态接口

App/Crud/RatingFlowCrud.py

@@ -92,3 +92,10 @@ def edit_rating_status(db, rating_flow_id, status):
     db.query(RatingFlow).filter_by(id=rating_flow_id).update(data)
     db.commit()
     return db.query(RatingFlow).filter_by(id=rating_flow_id).first()
+
+
+def change_rating_status(db, _id: str, status: bool):
+    update_data = {"status": status}
+    db.query(RatingFlowNode).filter_by(id=_id).update(update_data)
+    db.commit()
+    return db.query(RatingFlowNode).filter_by(id=_id).first()

App/Router/RatingFlowRouter.py

@@ -582,3 +582,19 @@ def func(region: list, db: Session = Depends(get_db)):
 @router.get("/import/shareholder_support", summary="导入股东外部支持", tags=["评级节点"])
 def func(shareholder: str, db: Session = Depends(get_db)):
     return {"外部支持级别": "AA", "最终调整级别": None, "调整说明": None}
+
+
+@router.post("/change_status", summary="更改节点状态", tags=["评级流程"])
+def func(req_body: RatingFlowSchema.ChangeStatusReqBody, db: Session = Depends(get_db)):
+    # 获取流程; rf_item: 流程对象
+    rf_item = Crud.get_rating_flow(db=db, rating_flow_id=req_body.rating_flow_id)
+    if not rf_item:
+        raise HTTPException(status_code=404, detail="RatingFlow Not Found")
+    # 查询节点保存的数据
+    node = [node for node in rf_item.nodes if node.node_name == req_body.node_name]
+    if not node:
+        raise HTTPException(status_code=404, detail="Node Not Found")
+    result = Crud.change_rating_status(db, _id=node[0].id, status=req_body.status)
+    if not result:
+        raise HTTPException(status_code=400, detail="Node Not Found")
+    return result

App/Schemas/RatingFlowSchema.py

@@ -77,3 +77,9 @@ class GetRatingFlowNodeContentReqBody(BaseModel):
 class CalculationAdjustReqBody(BaseModel):
     rating_flow_id: str
     adjustment: list
+
+
+class ChangeStatusReqBody(BaseModel):
+    rating_flow_id: str
+    node_name: str
+    status: bool