create endpoint to update user data

Author: violetaperezandrade

app/controller/Users.py

@@ -27,3 +27,10 @@ def handle_login(self, auth_code: str):
             "message": user,
             "status": status.HTTP_200_OK,
         }
+
+    def handle_update_user(self, user_id: int, update_data: dict):
+        self.users_service.update_user(user_id, update_data)
+        return {
+            "message": "User updated successfully",
+            "status": status.HTTP_200_OK,
+        }

app/main.py

@@ -2,7 +2,7 @@
 from controller.Users import UsersController
 from service.Users import UsersService
 from repository.Users import UsersRepository
-from schemas.Schemas import CreateUserSchema
+from schemas.Schemas import CreateUserSchema, UpdateUserSchema
 from schemas.Schemas import LoginRequest
 
 
@@ -35,3 +35,8 @@ async def create_user(user_data: CreateUserSchema):
 @app.post("/login")
 def login_with_google(request: LoginRequest):
     return users_controller.handle_login(request.auth_code)
+
+
+@app.patch("/users/{user_id}")
+async def update_user(user_id: int, update_data: UpdateUserSchema):
+    return users_controller.handle_update_user(user_id, update_data.dict())

app/repository/Users.py

@@ -58,6 +58,14 @@ def create_user(
         self.session.commit()
         return new_user
 
+    def edit_user(self, user_id: int, data_to_edit: dict):
+        user = self.session.query(User).filter_by(id=user_id).first()
+        for field, value in data_to_edit.items():
+            setattr(user, field, value)
+
+        self.session.commit()
+        return user
+
     def __parse_result(self, result):
         if not result:
             return []

app/schemas/Schemas.py

@@ -1,4 +1,5 @@
-from pydantic import BaseModel
+from pydantic import BaseModel, HttpUrl
+from typing import Optional
 
 
 class UserSchema(BaseModel):
@@ -15,3 +16,9 @@ class CreateUserSchema(BaseModel):
 
 class LoginRequest(BaseModel):
     auth_code: str
+
+
+class UpdateUserSchema(BaseModel):
+    name: Optional[str] = None
+    gender: Optional[str] = None
+    photo: Optional[HttpUrl] = None

app/service/Users.py

@@ -67,3 +67,7 @@ def _get_user_info(self, access_token):
         if response.json().get("picture") is not None:
             user_data['photo'] = response.json().get("picture")
         return user_data
+
+    def update_user(self, user_id: int, update_data: dict):
+        self.get_user(user_id)
+        self.user_repository.edit_user(user_id, update_data)