Ratings table added

back/api.py

@@ -43,33 +43,66 @@ if not os.path.exists("./uploads"):
 # создаем эндпоинт для хранения файлов пользователя
 app.mount("/uploads", StaticFiles(directory = "./uploads"))
 
+
 # эндпоинт для возвращения согласия в pdf
 @app.get("/privacy_policy.pdf")
 async def privacy_policy():
     return FileResponse("./privacy_policy.pdf")
 
+
 # получение списка объявлений
-@app.get("/api/announcements", response_model=List[pydantic_schemas.Announcement])#адрес объявлений 
-async def announcements_list(db: Annotated[Session, Depends(auth_utils.get_session)], obsolete: Union[bool, None] = False, user_id: Union[int, None] = None,
-         metro: Union[str, None] = None,category: Union[str, None] = None):
+@app.get("/api/announcements", response_model=List[pydantic_schemas.Announcement])
+async def announcements_list(db: Annotated[Session, Depends(auth_utils.get_session)], 
+        current_user: Annotated[pydantic_schemas.User, Depends(auth_utils.get_current_active_user)], 
+        obsolete: Union[bool, None] = False, user_id: Union[int, None] = None, metro: Union[str, None] = None, 
+        category: Union[str, None] = None):
     # параметры для сортировки (схема pydantic schemas.SortAnnouncements)
     params_to_sort = pydantic_schemas.SortAnnouncements(obsolete=obsolete, user_id=user_id, metro=metro, category=category)
-    # получаем результат
+    # получаем результат (значения с определенными полями obsolete, user_id, metro, category).
+    # user_id - id пользователя, которому принадлежат объявления
     result = await add_poems_and_filters.filter_ann(db=db, schema=params_to_sort)
 
-    return result
+    # для каждого отфильтрованного объявления проверяем, забронировано ли оно текущим пользователем и в зависимости 
+    # от этого флаг booked_by_current_user устанавливаем в 0 или в 1
+    check_if_booked = [pydantic_schemas.Announcement.from_orm(elem) for elem in result]
+    for an in check_if_booked:
+        # ищем пару с заданными id объявления и id текущего пользователя
+        query = await db.execute(select(orm_models.AnnouncementUser).where(
+            orm_models.AnnouncementUser.announcement_id == an.id).where(
+            orm_models.AnnouncementUser.booking_user_id == current_user.id))
+        pair_found = query.scalars().first()
+        
+        if pair_found:
+            an.booked_by_current_user = True
+        else:
+            an.booked_by_current_user = False
+    
+    return check_if_booked
 
 
 # получаем данные одного объявления
 @app.get("/api/announcement", response_model=pydantic_schemas.AnnResponce) 
-async def single_announcement(ann_id:int, db: Annotated[Session, Depends(auth_utils.get_session)]): # передаем индекс обявления
+async def single_announcement(ann_id:int, db: Annotated[Session, Depends(auth_utils.get_session)], 
+    current_user: Annotated[pydantic_schemas.User, Depends(auth_utils.get_current_active_user)]):
 # Считываем данные из Body и отображаем их на странице.
 # В последствии будем вставлять данные в html-форму
     announcement = await db.get(orm_models.Announcement, ann_id)
     #announcement = await db.execute(select(orm_models.Announcement)).scalars().all()
     if not announcement:
         raise HTTPException(status_code=404, detail="Item not found")
-    return announcement
+
+    # создаем форму pydantic_schemas.AnnResponce из ORM-объекта announcement
+    announcement_model = pydantic_schemas.AnnResponce.from_orm(announcement)
+
+    # ищем пару с заданными id объявления и id текущего пользователя
+    query = await db.execute(select(orm_models.AnnouncementUser).where(
+        orm_models.AnnouncementUser.announcement_id == announcement.id).where(
+            orm_models.AnnouncementUser.booking_user_id == current_user.id))
+    pair_found = query.scalars().first()
+    # если такая пара найдена, записываем в поле booked_by_current_user True
+    if pair_found:
+        announcement_model.booked_by_current_user = True
+    return announcement_model
 
 
 # Занести объявление в базу данных
@@ -153,7 +186,7 @@ async def change_book_status(data: pydantic_schemas.Book, current_user: Annotate
             new_pair = orm_models.AnnouncementUser(announcement_to_change.id, current_user.id)
             # Инкрементируем поле booked_counter на 1
             announcement_to_change.booked_counter += 1
-            # вставляем индекс забронировавшего пользователя в поле booked_by
+            # добавляем запись в таблицу announcementuser
             db.add(new_pair)
 
             # фиксируем изменения в бд