zhaoqingang
2024-12-18 ae88193c5153aa04226a8307e1294a9c75d312bb
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={})