mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 13:12:24 +00:00
100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Any
|
|
|
|
from fastapi import Request
|
|
|
|
from backend.app.admin.crud.crud_menu import menu_dao
|
|
from backend.app.admin.crud.crud_role import role_dao
|
|
from backend.app.admin.model import Menu
|
|
from backend.app.admin.schema.menu import CreateMenuParam, UpdateMenuParam
|
|
from backend.common.exception import errors
|
|
from backend.core.conf import settings
|
|
from backend.database.db_mysql import async_db_session
|
|
from backend.database.db_redis import redis_client
|
|
from backend.utils.build_tree import get_tree_data
|
|
|
|
|
|
class MenuService:
|
|
@staticmethod
|
|
async def get(*, pk: int) -> Menu:
|
|
async with async_db_session() as db:
|
|
menu = await menu_dao.get(db, menu_id=pk)
|
|
if not menu:
|
|
raise errors.NotFoundError(msg='菜单不存在')
|
|
return menu
|
|
|
|
@staticmethod
|
|
async def get_menu_tree(*, title: str | None = None, status: int | None = None) -> list[dict[str, Any]]:
|
|
async with async_db_session() as db:
|
|
menu_select = await menu_dao.get_all(db, title=title, status=status)
|
|
menu_tree = get_tree_data(menu_select)
|
|
return menu_tree
|
|
|
|
@staticmethod
|
|
async def get_role_menu_tree(*, pk: int) -> list[dict[str, Any]]:
|
|
async with async_db_session() as db:
|
|
role = await role_dao.get_with_relation(db, pk)
|
|
if not role:
|
|
raise errors.NotFoundError(msg='角色不存在')
|
|
menu_ids = [menu.id for menu in role.menus]
|
|
menu_select = await menu_dao.get_role_menus(db, False, menu_ids)
|
|
menu_tree = get_tree_data(menu_select)
|
|
return menu_tree
|
|
|
|
@staticmethod
|
|
async def get_user_menu_tree(*, request: Request) -> list[dict[str, Any]]:
|
|
async with async_db_session() as db:
|
|
roles = request.user.roles
|
|
menu_ids = []
|
|
menu_tree = []
|
|
if roles:
|
|
for role in roles:
|
|
menu_ids.extend([menu.id for menu in role.menus])
|
|
menu_select = await menu_dao.get_role_menus(db, request.user.is_superuser, menu_ids)
|
|
menu_tree = get_tree_data(menu_select)
|
|
return menu_tree
|
|
|
|
@staticmethod
|
|
async def create(*, obj: CreateMenuParam) -> None:
|
|
async with async_db_session.begin() as db:
|
|
title = await menu_dao.get_by_title(db, obj.title)
|
|
if title:
|
|
raise errors.ForbiddenError(msg='菜单标题已存在')
|
|
if obj.parent_id:
|
|
parent_menu = await menu_dao.get(db, obj.parent_id)
|
|
if not parent_menu:
|
|
raise errors.NotFoundError(msg='父级菜单不存在')
|
|
await menu_dao.create(db, obj)
|
|
|
|
@staticmethod
|
|
async def update(*, pk: int, obj: UpdateMenuParam) -> int:
|
|
async with async_db_session.begin() as db:
|
|
menu = await menu_dao.get(db, pk)
|
|
if not menu:
|
|
raise errors.NotFoundError(msg='菜单不存在')
|
|
if menu.title != obj.title:
|
|
if await menu_dao.get_by_title(db, obj.title):
|
|
raise errors.ForbiddenError(msg='菜单标题已存在')
|
|
if obj.parent_id:
|
|
parent_menu = await menu_dao.get(db, obj.parent_id)
|
|
if not parent_menu:
|
|
raise errors.NotFoundError(msg='父级菜单不存在')
|
|
if obj.parent_id == menu.id:
|
|
raise errors.ForbiddenError(msg='禁止关联自身为父级')
|
|
count = await menu_dao.update(db, pk, obj)
|
|
await redis_client.delete_prefix(settings.PERMISSION_REDIS_PREFIX)
|
|
return count
|
|
|
|
@staticmethod
|
|
async def delete(*, pk: int) -> int:
|
|
async with async_db_session.begin() as db:
|
|
children = await menu_dao.get_children(db, pk)
|
|
if children:
|
|
raise errors.ForbiddenError(msg='菜单下存在子菜单，无法删除')
|
|
count = await menu_dao.delete(db, pk)
|
|
return count
|
|
|
|
|
|
menu_service: MenuService = MenuService()
|