From 565c908bdc378e6ba1b83062962c9ff0c727d715 Mon Sep 17 00:00:00 2001 From: Backend IM Bot Date: Wed, 30 Apr 2025 19:45:44 +0000 Subject: [PATCH] feat: add endpoint to delete fruits by id --- endpoints/fruits.delete.py | 5 ++--- helpers/fruit_helpers.py | 4 +++- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/endpoints/fruits.delete.py b/endpoints/fruits.delete.py index 1486cf6..5529840 100644 --- a/endpoints/fruits.delete.py +++ b/endpoints/fruits.delete.py @@ -1,15 +1,14 @@ from fastapi import APIRouter, HTTPException, status from schemas.fruit import FruitSchema -from helpers.fruit_helpers import delete_fruit, get_fruit_by_id +from helpers.fruit_helpers import get_fruit_by_id, delete_fruit from sqlalchemy.orm import Session from fastapi import Depends from core.database import get_db -import uuid router = APIRouter() @router.delete("/fruits/{fruit_id}", status_code=status.HTTP_200_OK, response_model=FruitSchema) -async def delete_fruit_endpoint(fruit_id: uuid.UUID, db: Session = Depends(get_db)): +async def delete_fruit_endpoint(fruit_id: str, db: Session = Depends(get_db)): fruit = get_fruit_by_id(db, fruit_id) if not fruit: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Fruit not found") diff --git a/helpers/fruit_helpers.py b/helpers/fruit_helpers.py index cf1b382..4d31c14 100644 --- a/helpers/fruit_helpers.py +++ b/helpers/fruit_helpers.py @@ -68,4 +68,6 @@ def delete_fruit(db: Session, fruit_id: UUID) -> None: fruit = db.query(Fruit).filter(Fruit.id == fruit_id).first() if fruit: db.delete(fruit) - db.commit() \ No newline at end of file + db.commit() + else: + raise ValueError(f"No fruit found with ID {fruit_id}") \ No newline at end of file