fix: 添加角色增加菜单id”

This commit is contained in:
zy7y
2022-09-16 11:52:08 +08:00
parent 65f1d7d2bc
commit 7ba3f23684
20 changed files with 106 additions and 49 deletions

View File

@@ -1,6 +1,7 @@
from fastapi import Query
from dbhelper.menu import del_menu, get_menus, insert_menu, put_menu
from core.utils import list_to_tree
from dbhelper.menu import del_menu, get_menus, get_tree_menu, insert_menu, put_menu
from schemas import ListAll, MenuIn, MenuRead, Response
@@ -11,10 +12,9 @@ async def menu_add(data: MenuIn) -> Response[MenuRead]:
async def menu_arr(
offset: int = Query(default=1, description="偏移量"),
limit: int = Query(default=10, description="数量"),
) -> Response[ListAll[list[MenuRead]]]:
skip = (offset - 1) * limit
menus, count = await get_menus(skip, limit)
return Response(data=ListAll(total=count, items=menus))
) -> Response:
menus = await get_tree_menu()
return Response(data=list_to_tree(menus))
async def menu_del(pk: int) -> Response:

View File

@@ -16,7 +16,9 @@ from schemas import ListAll, Response, RoleIn, RoleInfo, RoleMenuIn, RoleQuery,
async def role_add(data: RoleIn) -> Response[RoleInfo]:
return Response(data=await new_role(data))
if result := await new_role(data):
return Response(data=result)
return Response(code=400, msg="菜单不存在")
async def role_has_menu(rid: int):

View File

@@ -19,10 +19,10 @@ async def user_add(data: UserAdd) -> Response[UserRead]:
"""新增用户并分配角色 一步到位"""
if await get_user({"username": data.username}) is not None:
return Response(code=400, msg="用户名已存在")
roles = data.rids
del data.rids
rids = data.roles
del data.roles
data.password = get_password_hash(data.password)
result = await insert_user(data, roles)
result = await insert_user(data, rids)
if isinstance(result, int):
return Response(code=400, msg=f"角色{result}不存在")
return Response(data=result)