From 77d51a55f7241e1eae25d89b4c76865c51565319 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期五, 06 十二月 2024 16:04:13 +0800 Subject: [PATCH] Merge branch 'master' of http://192.168.5.5:10010/r/rag-gateway --- app/api/user.py | 20 ++++++++++++++++++-- 1 files changed, 18 insertions(+), 2 deletions(-) diff --git a/app/api/user.py b/app/api/user.py index 6b6c866..90a23cd 100644 --- a/app/api/user.py +++ b/app/api/user.py @@ -1,11 +1,11 @@ from fastapi import APIRouter, Depends from app.api import Response, pwd_context, get_current_user -from app.models.app_model import AppRegisterModel +from app.models.public_api_model import AppRegisterModel from app.models.base_model import get_db from app.models.user import PageParameter, UserStatus, UserInfo, LoginData from app.models.user_model import UserModel from app.service.user import get_user_list, edit_user_status, delete_user_data, create_user, edit_user_data, \ - edit_user_pwd + edit_user_pwd, get_user_info, get_user_routers user_router = APIRouter() @@ -80,3 +80,19 @@ if not is_edit: return Response(code=500, msg="user pwd reset failure", data={}) return Response(code=200, msg="user pwd reset successfully", data={}) + + +@user_router.get("/user_info", response_model=Response) +async def user_info(current_user: UserModel = Depends(get_current_user),db=Depends(get_db)): + user_info = await get_user_info(db, current_user.id) + if not user_info: + return Response(code=500, msg="user get failure", data={}) + return Response(code=200, msg="successfully", data=user_info) + + +@user_router.get("/user_routers", response_model=Response) +async def user_routers(current_user: UserModel = Depends(get_current_user),db=Depends(get_db)): + routers = await get_user_routers(db, current_user.id) + if not routers: + return Response(code=500, msg="user get failure", data={}) + return Response(code=200, msg="successfully", data=routers) \ No newline at end of file -- Gitblit v1.8.0