From de45d8dc19cd16b85344a3c3bbb243a12e8811a0 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 13 十一月 2024 15:16:59 +0800
Subject: [PATCH] 增加功能
---
app/service/resource.py | 44 ++++++++++++++++++++++++++++++++++++++++----
1 files changed, 40 insertions(+), 4 deletions(-)
diff --git a/app/service/resource.py b/app/service/resource.py
index 572c55e..9d93cef 100644
--- a/app/service/resource.py
+++ b/app/service/resource.py
@@ -1,9 +1,7 @@
import uuid
-from app.Log import logger
-from app.models import OrganizationModel, GroupModel, ResourceModel
-from app.models.role_model import RoleModel
-
+from Log import logger
+from app.models import ResourceModel
async def get_resource_tree(db):
@@ -26,4 +24,42 @@
logger.error(e)
db.rollback()
return False
+ return True
+
+
+async def create_menu(db, menu_name, component, path, order_num, perms, menu_type, description, parent_id, status, icon):
+ try:
+ dept_model = ResourceModel(id=str(uuid.uuid4()),name=menu_name, url=component,path=path,perms=perms,description=description,status=status, icon=icon,seq=order_num, resource_type_id=menu_type)
+ if parent_id:
+ dept_model.parent = db.get(ResourceModel, parent_id)
+ db.add(dept_model)
+ db.commit()
+ db.refresh(dept_model)
+ except Exception as e:
+ logger.error(e)
+ db.rollback()
+ return False
+ return True
+
+
+async def edit_resource_data(db, menu_id, menu_name, component, path, order_num, perms, menu_type, description, parent_id, status, icon):
+ try:
+ dept_model = db.query(ResourceModel).filter(ResourceModel.id == menu_id).first()
+ dept_model.name = menu_name
+ dept_model.url = component
+ dept_model.path = path
+ dept_model.perms = perms
+ dept_model.status = status
+ dept_model.icon = icon
+ dept_model.seq = order_num
+ dept_model.description = description
+ dept_model.resource_type_id = menu_type
+ if parent_id:
+ dept_model.parent = db.get(ResourceModel, parent_id)
+ 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
--
Gitblit v1.8.0