From 20bbed188fd7833d14801dc202fa3cdbb085ca2b Mon Sep 17 00:00:00 2001
From: xuyonghao <898441624@qq.com>
Date: 星期五, 20 十二月 2024 11:51:34 +0800
Subject: [PATCH] 优化excel合并脚本

---
 app/api/resource.py |   14 ++++++++++++--
 1 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/api/resource.py b/app/api/resource.py
index c722bec..421b83b 100644
--- a/app/api/resource.py
+++ b/app/api/resource.py
@@ -1,10 +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 import MenuInfo, MenuStatus
 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, create_menu, \
-    edit_resource_data
+    edit_resource_data, edit_resource_status
 
 menu_router = APIRouter()
 
@@ -49,3 +49,13 @@
     if not is_create:
         return Response(code=500, msg="dept edit failure", data={})
     return Response(code=200, msg="dept edit successfully", data={})
+
+
+@menu_router.put("/status", response_model=Response)
+async def resource_edit_status(menu: MenuStatus, current_user: UserModel = Depends(get_current_user), db=Depends(get_db)):
+    if menu.status not in ["0", "1"]:
+        return Response(code=400, msg="The status cannot be {}!".format(menu.status))
+    is_create = await edit_resource_status(db, menu.menuId, menu.status)
+    if not is_create:
+        return Response(code=500, msg="dept status edit failure", data={})
+    return Response(code=200, msg="dept status edit success", data={})

--
Gitblit v1.8.0