From 0094b18ddc3e9c2738987eebe735edb7e858700f Mon Sep 17 00:00:00 2001
From: Dmitry <ksenon.104@gmail.com>
Date: Sat, 17 Aug 2024 22:13:49 +0300
Subject: [PATCH] Endpoint cancelling a booking added

---
 back/api.py              | 30 +++++++++++++++++++++++++++---
 back/pydantic_schemas.py |  3 +++
 2 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/back/api.py b/back/api.py
index 0c571b6..7a3a47e 100644
--- a/back/api.py
+++ b/back/api.py
@@ -124,11 +124,11 @@ async def delete_from_db(announcement: pydantic_schemas.DelAnnouncement, db: Ann
 
         return {"Success": True}
     except:
-        raise HTTPException(status_code=500, detail="Problem with adding to database")
+        raise HTTPException(status_code=500, detail="Problem with deleteng the announcement from the database")
     
 
 # Забронировать объявление
-@app.post("/api/book")
+@app.put("/api/book")
 async def change_book_status(data: pydantic_schemas.Book, current_user: Annotated[pydantic_schemas.User, Depends(auth_utils.get_current_user)], 
         db: Annotated[Session, Depends(auth_utils.get_session)]):
     # Находим объявление по данному id
@@ -163,7 +163,31 @@ async def change_book_status(data: pydantic_schemas.Book, current_user: Annotate
         raise HTTPException(status_code=403, detail="The announcement is already booked by this user")
 
 
-# reginstration
+# Отмена брони
+@app.delete("/api/book")
+async def cancel_booking(data: pydantic_schemas.CancelBooking, current_user: Annotated[pydantic_schemas.User, Depends(auth_utils.get_current_user)], 
+        db: Annotated[Session, Depends(auth_utils.get_session)]):
+    
+    # ищем пару с заданными id объявления и пользователя
+    query = await db.execute(select(orm_models.AnnouncementUser).where(
+        orm_models.AnnouncementUser.announcement_id == data.announcement_id).where(
+            orm_models.AnnouncementUser.booking_user_id == current_user.id))
+    pair_found = query.scalars().first()
+    # если не найдена
+    if not pair_found:
+        raise HTTPException(status_code=404, detail="Item not found")
+    # если найдена пытаемся удалить из бд
+    else:
+        try:
+            await db.delete(pair_found)   # удаление из БД
+            await db.commit()     # сохраняем изменения
+
+            return {"Success": True}
+        except:
+            raise HTTPException(status_code=500, detail="Problem with deleteng pair 'booked_announcement_id:user_id' from the database")
+
+
+# Регистрация
 @app.post("/api/signup")
 async def create_user(nickname: Annotated[str, Form()], password: Annotated[str, Form()], db: Annotated[Session, Depends(auth_utils.get_session)],
         name: Annotated[str, Form()]=None, surname: Annotated[str, Form()]=None, avatar: Annotated[UploadFile, Form()]=None):
diff --git a/back/pydantic_schemas.py b/back/pydantic_schemas.py
index 0bc2b33..0562cf1 100644
--- a/back/pydantic_schemas.py
+++ b/back/pydantic_schemas.py
@@ -8,6 +8,9 @@ class Book(BaseModel):
     id: int
     
 
+class CancelBooking(BaseModel):
+    announcement_id: int   
+
 class DelAnnouncement(BaseModel):
     id: int