diff --git a/endpoints/kella.delete.py b/endpoints/kella.delete.py index 5c3b10e..41deeee 100644 --- a/endpoints/kella.delete.py +++ b/endpoints/kella.delete.py @@ -5,15 +5,16 @@ from sqlalchemy.orm import Session from core.database import get_db from models.book import Book from schemas.book import BookSchema -from helpers.book_helpers import delete_book_by_id +from helpers.book_helpers import delete_book router = APIRouter() @router.delete("/books/{book_id}", status_code=status.HTTP_204_NO_CONTENT) -async def delete_book( +async def delete_book_by_id( book_id: int, db: Session = Depends(get_db) ): - if not delete_book_by_id(db, book_id): + book = delete_book(db, book_id) + if not book: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Book not found") return \ No newline at end of file