diff --git a/ruoyi-fastapi-backend/module_admin/controller/login_controller.py b/ruoyi-fastapi-backend/module_admin/controller/login_controller.py index 7cc4d26..9a255ef 100644 --- a/ruoyi-fastapi-backend/module_admin/controller/login_controller.py +++ b/ruoyi-fastapi-backend/module_admin/controller/login_controller.py @@ -81,25 +81,25 @@ async def login( return ResponseUtil.success(msg='登录成功', dict_content={'token': access_token}) -# @loginController.get('/getInfo', response_model=CurrentUserModel) -# async def get_login_user_info( -# request: Request, current_user: CurrentUserModel = Depends(LoginService.get_current_user) -# ): -# logger.info('获取成功') -# -# return ResponseUtil.success(model_content=current_user) +@loginController.get('/getInfo', response_model=CurrentUserModel) +async def get_login_user_info( + request: Request, current_user: CurrentUserModel = Depends(LoginService.get_current_user) +): + logger.info('获取成功') + + return ResponseUtil.success(model_content=current_user) -# @loginController.get('/getRouters') -# async def get_login_user_routers( -# request: Request, -# current_user: CurrentUserModel = Depends(LoginService.get_current_user), -# query_db: AsyncSession = Depends(get_db), -# ): -# logger.info('获取成功') -# user_routers = await LoginService.get_current_user_routers(current_user.user.user_id, query_db) -# -# return ResponseUtil.success(data=user_routers) +@loginController.get('/getRouters') +async def get_login_user_routers( + request: Request, + current_user: CurrentUserModel = Depends(LoginService.get_current_user), + query_db: AsyncSession = Depends(get_db), +): + logger.info('获取成功') + user_routers = await LoginService.get_current_user_routers(current_user.user.user_id, query_db) + + return ResponseUtil.success(data=user_routers) @loginController.post('/register', response_model=CrudResponseModel)