From e5e2be29cfc033c976558a4d0b5fa73426818b90 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 26 二月 2025 11:30:48 +0800
Subject: [PATCH] 增加系统信息修改

---
 app/service/resource.py |   64 +++++++++++++++++++++++++------
 1 files changed, 51 insertions(+), 13 deletions(-)

diff --git a/app/service/resource.py b/app/service/resource.py
index 7880c01..869910b 100644
--- a/app/service/resource.py
+++ b/app/service/resource.py
@@ -1,6 +1,7 @@
 import uuid
 
 from Log import logger
+from app.config.const import RESOURCE_STATUS_ON, RESOURCE_STATUS_OFF, RESOURCE_STATUS_DELETE
 from app.models import ResourceModel
 
 
@@ -17,14 +18,28 @@
 
 
 async def delete_resource_info(db, menu_id):
+    delete_list = []
+    dept_list = []
+    next_dept_list = []
+    base_dept = db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).first()
+    dept_list.append(base_dept)
+    while dept_list:
+        for dept in dept_list:
+            delete_list.append(dept.id)
+            for child_dept in dept.children:
+                next_dept_list.append(child_dept)
+
+        dept_list = next_dept_list
+        next_dept_list = []
+
     try:
-        db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).delete()
+        db.query(ResourceModel).filter(ResourceModel.id.in_(delete_list)).update({"status": RESOURCE_STATUS_DELETE})
         db.commit()
     except Exception as e:
         logger.error(e)
         db.rollback()
-        return False
-    return True
+        return '鏈嶅姟寮傚父 锛�'
+    return ""
 
 
 async def create_menu(db, menu_name, component, path, order_num, perms, menu_type, description, parent_id, status, icon):
@@ -66,13 +81,36 @@
 
 
 async def edit_resource_status(db, menu_id, status):
-    try:
-        dept_model = db.query(ResourceModel).filter(ResourceModel.id == menu_id).first()
-        dept_model.status = status
-        db.commit()
-        db.refresh(dept_model)
-    except Exception as e:
-        logger.error(e)
-        db.rollback()
-        return False
-    return True
\ No newline at end of file
+    delete_list = []
+    dept_list = []
+    next_dept_list = []
+    base_dept = db.query(ResourceModel).filter(ResourceModel.id.__eq__(menu_id)).first()
+    if status == RESOURCE_STATUS_ON:
+        if base_dept.parent and base_dept.parent.status != RESOURCE_STATUS_ON:
+            return '涓婄骇鑺傜偣鐘舵�佸紓甯革紒'
+        try:
+            base_dept.status = status
+            db.commit()
+        except Exception as e:
+            logger.error(e)
+            db.rollback()
+            return '鏈嶅姟寮傚父 锛�'
+    else:
+        dept_list.append(base_dept)
+        while dept_list:
+            for dept in dept_list:
+                delete_list.append(dept.id)
+                for child_dept in dept.children:
+                    next_dept_list.append(child_dept)
+
+            dept_list = next_dept_list
+            next_dept_list = []
+        try:
+            db.query(ResourceModel).filter(ResourceModel.id.in_(delete_list)).update(
+                {"status": RESOURCE_STATUS_OFF})
+            db.commit()
+        except Exception as e:
+            logger.error(e)
+            db.rollback()
+            return '鏈嶅姟寮傚父 锛�'
+    return ""
\ No newline at end of file

--
Gitblit v1.8.0