python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2025-11-02 03:31:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix #13: add /me update route

Browse Source
This commit is contained in:
François Voron
2019-10-20 10:11:48 +02:00
parent 1996ed69ea
commit e146e4bca4
4 changed files with 119 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
fastapi_users/models.py
View File

@ -18,6 +18,11 @@ class BaseUser(BaseModel):
def default_id(cls, v):
return v or str(uuid.uuid4())
def create_update_dict(self):
return self.dict(
skip_defaults=True, exclude={"id", "is_superuser", "is_active"}
)
class BaseUserCreate(BaseUser):
email: EmailStr
@ -25,7 +30,7 @@ class BaseUserCreate(BaseUser):
class BaseUserUpdate(BaseUser):
pass
password: Optional[str]
class BaseUserDB(BaseUser):

18
fastapi_users/router.py
View File

@ -45,8 +45,7 @@ def get_user_router(
hashed_password = get_password_hash(user.password)
db_user = models.UserDB(
**user.dict(exclude={"id", "is_superuser", "is_active"}),
hashed_password=hashed_password
**user.create_update_dict(), hashed_password=hashed_password
)
created_user = await user_db.create(db_user)
return created_user
@ -110,4 +109,19 @@ def get_user_router(
):
return user
@router.patch("/me", response_model=models.User)
async def update_me(
updated_user: models.UserUpdate, # type: ignore
user: models.UserDB = Depends(get_current_active_user), # type: ignore
):
updated_user_data = updated_user.create_update_dict()
for field in updated_user_data:
if field == "password":
hashed_password = get_password_hash(updated_user_data[field])
user.hashed_password = hashed_password
else:
setattr(user, field, updated_user_data[field])
return await user_db.update(user)
return router