| | |
| | | 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() |
| | | |
| | |
| | | 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={}) |