feat: 切换角色

backend/controller/menu.py

@@ -28,4 +28,3 @@ async def menu_put(pk: int, data: MenuIn) -> Response:
     if await put_menu(pk, data) == 0:
         return Response(code=400, msg="菜单不存在")
     return Response()
-

backend/controller/role.py

@@ -4,10 +4,15 @@ from fastapi import Query
 
 from core.utils import list_to_tree
 from dbhelper.relation import role_assigned_menu
-from dbhelper.role import (del_role, get_role, get_role_menus, get_roles,
-                           new_role, put_role)
-from schemas import (ListAll, Response, RoleIn, RoleInfo, RoleMenuIn,
-                     RoleQuery, RoleRead)
+from dbhelper.role import (
+    del_role,
+    get_role,
+    get_role_menus,
+    get_roles,
+    new_role,
+    put_role,
+)
+from schemas import ListAll, Response, RoleIn, RoleInfo, RoleMenuIn, RoleQuery, RoleRead
 
 
 async def role_add(data: RoleIn) -> Response[RoleInfo]:

backend/controller/user.py

@@ -1,8 +1,16 @@
-from fastapi import Query
+from fastapi import Depends, Query
+from starlette.requests import Request
 
-from core.security import get_password_hash
-from dbhelper.user import (del_user, get_user, get_user_info, get_users,
-                           insert_user, put_user)
+from core.security import check_token, get_password_hash
+from dbhelper.user import (
+    del_user,
+    get_user,
+    get_user_info,
+    get_users,
+    insert_user,
+    put_user,
+    select_role,
+)
 from schemas import Response, UserAdd, UserInfo, UserPut, UserQuery, UserRead
 from schemas.common import ListAll
 
@@ -63,3 +71,11 @@ async def user_put(pk: int, data: UserPut) -> Response:
     if isinstance(result, int):
         return Response(code=400, msg=f"角色不存在{result}")
     return Response()
+
+
+async def user_select_role(rid: int, user=Depends(check_token)):
+    """用户切换角色"""
+    res = await select_role(user.id, rid)
+    if res == 0:
+        return Response(code=400, msg=f"角色不存在{res}")
+    return Response()