邹方成
cc352d3184
refactor: 优化API路由和响应模型 feat(admin): 添加App用户管理接口 feat(sms): 实现阿里云短信服务集成 feat(email): 添加SMTP邮件发送功能 feat(upload): 支持文件上传接口 feat(rate-limiter): 实现手机号限流器 fix: 修复计算步骤入库问题 docs: 更新API文档和测试计划 chore: 更新依赖和配置
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
import logging
|
|
|
|
from fastapi import APIRouter, Query
|
|
|
|
from app.controllers.menu import menu_controller
|
|
from app.schemas.base import Fail, Success, SuccessExtra, BasicResponse, PageResponse, MessageOut
|
|
from app.schemas.menus import BaseMenu
|
|
from app.schemas.menus import *
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/list", summary="查看菜单列表", response_model=PageResponse[BaseMenu])
|
|
async def list_menu(
|
|
page: int = Query(1, description="页码"),
|
|
page_size: int = Query(10, description="每页数量"),
|
|
):
|
|
async def get_menu_with_children(menu_id: int):
|
|
menu = await menu_controller.model.get(id=menu_id)
|
|
menu_dict = await menu.to_dict()
|
|
child_menus = await menu_controller.model.filter(parent_id=menu_id).order_by("order")
|
|
menu_dict["children"] = [await get_menu_with_children(child.id) for child in child_menus]
|
|
return menu_dict
|
|
|
|
parent_menus = await menu_controller.model.filter(parent_id=0).order_by("order")
|
|
res_menu = [await get_menu_with_children(menu.id) for menu in parent_menus]
|
|
return SuccessExtra(data=res_menu, total=len(res_menu), page=page, page_size=page_size)
|
|
|
|
|
|
@router.get("/get", summary="查看菜单", response_model=BasicResponse[BaseMenu])
|
|
async def get_menu(
|
|
menu_id: int = Query(..., description="菜单id"),
|
|
):
|
|
result = await menu_controller.get(id=menu_id)
|
|
return Success(data=result)
|
|
|
|
|
|
@router.post("/create", summary="创建菜单", response_model=BasicResponse[MessageOut])
|
|
async def create_menu(
|
|
menu_in: MenuCreate,
|
|
):
|
|
await menu_controller.create(obj_in=menu_in)
|
|
return Success(msg="Created Success")
|
|
|
|
|
|
@router.post("/update", summary="更新菜单", response_model=BasicResponse[MessageOut])
|
|
async def update_menu(
|
|
menu_in: MenuUpdate,
|
|
):
|
|
await menu_controller.update(id=menu_in.id, obj_in=menu_in)
|
|
return Success(msg="Updated Success")
|
|
|
|
|
|
@router.delete("/delete", summary="删除菜单", response_model=BasicResponse[MessageOut])
|
|
async def delete_menu(
|
|
id: int = Query(..., description="菜单id"),
|
|
):
|
|
child_menu_count = await menu_controller.model.filter(parent_id=id).count()
|
|
if child_menu_count > 0:
|
|
return Fail(msg="Cannot delete a menu with child menus")
|
|
await menu_controller.remove(id=id)
|
|
return Success(msg="Deleted Success")
|