Manually cast user to User schema in routes to avoid issue #1000

This commit is contained in:
François Voron
2022-05-25 08:27:46 +02:00
parent e8635d9132
commit 1b6b58c0e0
2 changed files with 7 additions and 5 deletions

View File

@@ -71,6 +71,6 @@ def get_register_router(
},
)
return created_user
return user_schema.from_orm(created_user)
return router

View File

@@ -48,7 +48,7 @@ def get_users_router(
async def me(
user: models.UP = Depends(get_current_active_user),
):
return user
return user_schema.from_orm(user)
@router.patch(
"/me",
@@ -93,9 +93,10 @@ def get_users_router(
user_manager: BaseUserManager[models.UP, models.ID] = Depends(get_user_manager),
):
try:
return await user_manager.update(
user = await user_manager.update(
user_update, user, safe=True, request=request
)
return user_schema.from_orm(user)
except exceptions.InvalidPasswordException as e:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
@@ -128,7 +129,7 @@ def get_users_router(
},
)
async def get_user(user=Depends(get_user_or_404)):
return user
return user_schema.from_orm(user)
@router.patch(
"/{id}",
@@ -179,9 +180,10 @@ def get_users_router(
user_manager: BaseUserManager[models.UP, models.ID] = Depends(get_user_manager),
):
try:
return await user_manager.update(
user = await user_manager.update(
user_update, user, safe=False, request=request
)
return user_schema.from_orm(user)
except exceptions.InvalidPasswordException as e:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,