[FWA-3] completed update Profile

backend/routes/auth/auth.py

@@ -9,7 +9,7 @@ from backend.core.config import get_app_settings
 from backend.core.dependencies.dependencies import get_current_user
 from backend.core import MessageCode
 from backend.db.db_setup import generate_session
-from backend.schemas import ReturnValue, UserRequest, LoginResponse, UserCreate, PrivateUser
+from backend.schemas import ReturnValue, UserRequest, LoginResponse, PrivateUser
 from backend.services.user import UserService
 
 
@@ -29,14 +29,6 @@ async def get_token(form_data: Annotated[OAuth2PasswordRequestForm, Depends()],
   return {'access_token': token, 'token_type': 'bearer'}
 
 
-@auth_router.put('/register')
-def register_user(user: UserCreate, db: db_dependency) -> ReturnValue[Any]:
-  db_user = user_service.get_by_username(username=user.username)
-  if db_user:
-    raise HTTPException(status_code=400, detail=MessageCode.CREATED_USER)
-  user_service.create(db=db, user=user)
-  return ReturnValue(status=200, data="created")
-
 @auth_router.post('/login', response_model=ReturnValue[LoginResponse])
 def user_login(user: UserRequest, response: Response) -> ReturnValue[Any]:
   db_user = user_service.check_exist(user=user)

backend/routes/user/user.py

@@ -1,11 +1,12 @@
 from typing import Annotated, Any
-from fastapi import APIRouter, Depends
+from fastapi import APIRouter, Depends, HTTPException
 from sqlalchemy.orm import Session
 from backend.core.config import get_app_settings
 from backend.core.dependencies import is_logged_in
+from backend.core.message_code import MessageCode
 from backend.db.db_setup import generate_session
 from backend.schemas.common import ReturnValue
-from backend.schemas.user import ProfileResponse
+from backend.schemas.user import ProfileResponse, UserCreate, UserProfile
 from backend.services.user import UserService
 
 
@@ -16,6 +17,19 @@ settings = get_app_settings()
 db_dependency = Annotated[Session, Depends(generate_session)]
 current_user_token = Annotated[ProfileResponse, Depends(is_logged_in)]
 
+@public_router.put('/register')
+def register_user(user: UserCreate, db: db_dependency) -> ReturnValue[Any]:
+  db_user = user_service.get_by_username(username=user.username)
+  if db_user:
+    raise HTTPException(status_code=400, detail=MessageCode.CREATED_USER)
+  user_service.create(db=db, user=user)
+  return ReturnValue(status=200, data="created")
+
 @public_router.get("/me", response_model=ReturnValue[ProfileResponse])
 def get_user(current_user: current_user_token) -> ReturnValue[Any]:
   return ReturnValue(status=200, data=current_user)
+
+@public_router.put("/update-profile", response_model=ReturnValue[ProfileResponse])
+def update_user(user: UserProfile, current_user: current_user_token, db: db_dependency) -> ReturnValue[Any]:
+  db_user = user_service.update(db=db, user=user, user_id=current_user.id)
+  return ReturnValue(status=200, data=db_user)