changes 删除评级

App/Crud/RatingFlowCrud.py

@@ -100,3 +100,14 @@ def change_rating_status(db, _id: str, status: bool):
     db.commit()
     return db.query(RatingFlowNode).filter_by(id=_id).first()
 
+
+def delete_rating_flow(db, _id: str):
+    db.query(RatingFlow).filter_by(id=_id).delete()
+    db.commit()
+    return True
+
+
+def delete_rating_flow_node(db, _id: str):
+    db.query(RatingFlowNode).filter_by(id=_id).delete()
+    db.commit()
+    return True

App/Router/RatingFlowRouter.py

@@ -130,6 +130,21 @@ def func(rating_flow_id: str, operation: str, db: Session = Depends(get_db)):
         return {"info": "评级已确认无法执行操作"}
 
 
+@router.get("/delete", summary="删除评级", tags=["评级流程"])
+def func(rating_flow_id: str, db: Session = Depends(get_db)):
+    # 获取流程; rf_item: 流程对象
+    rf_item = Crud.get_rating_flow(db=db, rating_flow_id=rating_flow_id)
+    if not rf_item:
+        raise HTTPException(status_code=404, detail="RatingFlow Not Found")
+    nodes = rf_item.nodes
+    for node in nodes:
+        Crud.delete_rating_flow_node(db, node.id)
+    result = Crud.delete_rating_flow(db, rating_flow_id)
+    if not result:
+        raise HTTPException(status_code=401, detail="Delete Failed")
+    return {"info": "Success"}
+
+
 @router.get("/import/input_data", summary="导入填报数据", tags=["评级节点"])
 def func(rating_flow_id: str, year: int, db: Session = Depends(get_db), mongodb: MongoHelper = Depends(get_mongodb)):
     # 获取流程; rf_item: 流程对象