From de45d8dc19cd16b85344a3c3bbb243a12e8811a0 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 13 十一月 2024 15:16:59 +0800
Subject: [PATCH] 增加功能
---
app/api/resource.py | 32 +++++++++++++++++++++++---------
1 files changed, 23 insertions(+), 9 deletions(-)
diff --git a/app/api/resource.py b/app/api/resource.py
index c3abb75..c722bec 100644
--- a/app/api/resource.py
+++ b/app/api/resource.py
@@ -1,8 +1,10 @@
from fastapi import APIRouter, Depends
from app.api import Response, pwd_context, get_current_user
+from app.models import MenuInfo
from app.models.base_model import get_db
from app.models.user_model import UserModel
-from app.service.resource import get_resource_tree, get_resource_info, delete_resource_info
+from app.service.resource import get_resource_tree, get_resource_info, delete_resource_info, create_menu, \
+ edit_resource_data
menu_router = APIRouter()
@@ -26,12 +28,24 @@
@menu_router.post("/add_menu", response_model=Response)
-async def add_dept(dept: DeptInfo, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
- if not dept.deptName:
- return Response(code=400, msg="The deptName cannot be empty!")
- if dept.status not in ["0", "1"]:
- return Response(code=400, msg="The status cannot be {}!".format(dept.status))
- is_create = await create_dept(db, dept.deptName, dept.leader, dept.phone, dept.address, dept.status, dept.orderNum, dept.roles, dept.groups, dept.parentId)
+async def resource_add_menu(menu: MenuInfo, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
+ if not menu.menuName:
+ return Response(code=400, msg="The menuName cannot be empty!")
+ if menu.status not in ["0", "1"]:
+ return Response(code=400, msg="The status cannot be {}!".format(menu.status))
+ is_create = await create_menu(db, menu.menuName, menu.component, menu.path, menu.orderNum, menu.perms, menu.menuType, menu.description, menu.parentId, menu.status,menu.icon)
if not is_create:
- return Response(code=500, msg="dept create failure", data={})
- return Response(code=200, msg="dept create successfully", data={})
+ return Response(code=500, msg="menu create failure", data={})
+ return Response(code=200, msg="menu create successfully", data={})
+
+
+@menu_router.put("/edit_menu", response_model=Response)
+async def resource_edit_menu(menu: MenuInfo, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
+ if not menu.menuName:
+ return Response(code=400, msg="The menuName cannot be empty!")
+ if menu.status not in ["0", "1"]:
+ return Response(code=400, msg="The status cannot be {}!".format(menu.status))
+ is_create = await edit_resource_data(db, menu.menuId, menu.menuName, menu.component, menu.path, menu.orderNum, menu.perms, menu.menuType, menu.description, menu.parentId, menu.status,menu.icon)
+ if not is_create:
+ return Response(code=500, msg="dept edit failure", data={})
+ return Response(code=200, msg="dept edit successfully", data={})
--
Gitblit v1.8.0