Merge pull request #6 from HublastX/feat/multiple-updates

Feat/multiple updates

Author: VilasBoass

backend/app/api/controller/theme/delete_theme.py

@@ -0,0 +1,41 @@
+from fastapi import HTTPException, status
+from sqlalchemy.orm import Session
+
+from app.repository.theme_repository import ThemeRepository
+
+
+def delete_theme(theme_id: str, user_id: str, db: Session) -> dict:
+    try:
+        repository = ThemeRepository(db)
+
+        theme = repository.get_theme_by_id(theme_id)
+        if not theme:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="Theme not found"
+            )
+
+        if theme.user_id != user_id:
+            raise HTTPException(
+                status_code=status.HTTP_403_FORBIDDEN,
+                detail="Access denied"
+            )
+
+        deleted = repository.delete_theme(theme_id)
+
+        if not deleted:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="Theme not found"
+            )
+
+        return {"message": "Theme deleted successfully"}
+
+    except HTTPException:
+        raise
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"Error deleting theme: {str(e)}"
+        )

backend/app/api/controller/theme/update_theme.py

@@ -0,0 +1,64 @@
+from fastapi import HTTPException, status
+from sqlalchemy.orm import Session
+
+from app.repository.theme_repository import ThemeRepository
+from app.schemas.theme_schemas import ThemeUpdateSchema, ThemeResponseSchema
+
+
+def update_theme(
+    theme_id: str,
+    user_id: str,
+    data: ThemeUpdateSchema,
+    db: Session
+) -> ThemeResponseSchema:
+
+    try:
+        repository = ThemeRepository(db)
+        theme = repository.get_theme_by_id(theme_id)
+
+        if not theme:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="Theme not found"
+            )
+
+        if theme.user_id != user_id:
+            raise HTTPException(
+                status_code=status.HTTP_403_FORBIDDEN,
+                detail="Access denied"
+            )
+
+        update_data = data.model_dump(exclude_unset=True)
+
+        if not update_data:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="At least one field must be provided for update"
+            )
+
+        if "title" in update_data and update_data["title"] is not None:
+            if not update_data["title"].strip():
+                raise HTTPException(
+                    status_code=status.HTTP_400_BAD_REQUEST,
+                    detail="Title cannot be empty"
+                )
+
+        if "description" in update_data and update_data["description"] is not None:
+            if not update_data["description"].strip():
+                raise HTTPException(
+                    status_code=status.HTTP_400_BAD_REQUEST,
+                    detail="Description cannot be empty"
+                )
+
+        updated_theme = repository.update_theme(theme_id, update_data)
+
+        return ThemeResponseSchema.model_validate(updated_theme)
+
+    except HTTPException:
+        raise
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=f"Error updating theme: {str(e)}"
+        )

backend/app/api/routes/private/theme_routes/theme_routes.py

@@ -3,10 +3,13 @@
 
 from app.api.controller.theme.create_theme import create_theme
 from app.api.controller.theme.get_theme import get_theme_by_id, get_themes_by_user_id
+from app.api.controller.theme.update_theme import update_theme 
+from app.api.controller.theme.delete_theme import delete_theme
+
 from app.api.dependencies.auth import get_current_user
 from app.core.database.database import get_db
 from app.model.user_model import User
-from app.schemas.theme_schemas import ThemeCreateSchema, ThemeResponseSchema
+from app.schemas.theme_schemas import ThemeCreateSchema, ThemeResponseSchema, ThemeUpdateSchema
 
 router = APIRouter()
 
@@ -36,3 +39,21 @@ def get_themes_route(
 ):
     return get_themes_by_user_id(current_user.id, db)
 
+
+@router.patch("/{theme_id}", response_model=ThemeResponseSchema)
+def update_theme_route(
+    theme_id: str,
+    theme_update: ThemeUpdateSchema,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db),
+):
+    return update_theme(theme_id, theme_update, current_user.id, db)
+
+@router.delete("/{theme_id}")
+def delete_theme_route(
+    theme_id: str,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db),
+):
+    delete_theme(theme_id, current_user.id, db)
+    return {"message": "Theme deleted successfully"}

backend/app/repository/theme_repository.py

@@ -29,3 +29,34 @@ def get_theme_by_id(self, theme_id: str) -> Theme | None:
     def get_themes_by_user_id(self, user_id: str) -> list[Theme]:
         return self.db.query(Theme).filter(Theme.user_id == user_id).all()
 
+    def update_theme(self, theme_id: str, update_data: dict) -> Theme | None:
+        try:
+            theme = self.get_theme_by_id(theme_id)
+            if not theme:
+                return None
+
+            for key, value in update_data.items():
+                if value is not None:
+                    setattr(theme, key, value)
+
+            self.db.commit()
+            self.db.refresh(theme)
+            return theme
+
+        except Exception:
+            self.db.rollback()
+            raise
+
+    def delete_theme(self, theme_id: str) -> bool:
+        try:
+            theme = self.get_theme_by_id(theme_id)
+            if theme:
+                self.db.delete(theme)
+                self.db.commit()
+                return True
+
+            return False
+
+        except Exception:
+            self.db.rollback()
+            raise

backend/app/schemas/theme_schemas.py

@@ -7,6 +7,11 @@ class ThemeCreateSchema(BaseModel):
     description: str
 
 
+class ThemeUpdateSchema(BaseModel):
+    title: str | None = None
+    description: str | None = None
+
+
 class ThemeResponseSchema(BaseModel):
     id: str
     title: str