fix: 添加角色增加菜单id”
This commit is contained in:
@@ -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:
|
||||
|
@@ -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):
|
||||
|
@@ -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)
|
||||
|
Reference in New Issue
Block a user