From f441f503057cf2a19e997d6c973de6acb0dfa090 Mon Sep 17 00:00:00 2001
From: P3ngSaM <61768364+P3ngSaM@users.noreply.github.com>
Date: Tue, 29 Nov 2022 10:50:04 +0800
Subject: [PATCH] =?UTF-8?q?changes=20=E5=88=A0=E9=99=A4=E8=AF=84=E7=BA=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 App/Crud/RatingFlowCrud.py     | 11 +++++++++++
 App/Router/RatingFlowRouter.py | 15 +++++++++++++++
 2 files changed, 26 insertions(+)

diff --git a/App/Crud/RatingFlowCrud.py b/App/Crud/RatingFlowCrud.py
index 7eb6bfb..ad4d2ec 100644
--- a/App/Crud/RatingFlowCrud.py
+++ b/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
diff --git a/App/Router/RatingFlowRouter.py b/App/Router/RatingFlowRouter.py
index 6c18e5b..6e310bb 100644
--- a/App/Router/RatingFlowRouter.py
+++ b/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: 流程对象