update avg_rating

app/schema/movie_schema.py

@@ -17,7 +17,7 @@ class MovieCreateSchema(BaseModel):
 class MovieResponseSchema(MovieCreateSchema):
     id: int
     cover_image_url: str
-    average_rating: float | None
+    average_rating: float
     reviews: list[ReviewResponseSchema]
 
 

app/services/movie_services.py

@@ -3,6 +3,7 @@
 from app.database import session
 from app.forms import AddMovieForm
 from app.utils import delete_image
+from app.services.reviews_service import get_all_reviews_by_movie
 
 """ START MOVIE OPERATIONS """
 
@@ -14,7 +15,18 @@ def get_all_movies(db: session):
     :param db:
     :return: All movies in the database
     """
-    return db.query(Movie).all()
+    movies = db.query(Movie).all()
+    for movie in movies:
+        reviews = get_all_reviews_by_movie(movie.id, db)
+
+        if reviews:
+            average_rating = sum(review.rating for review in reviews) / len(reviews)
+            movie.average_rating = average_rating
+        else:
+            movie.average_rating = 0.0
+
+    return movies
+
 
 
 def add_movie(movie: AddMovieForm, db: session):

app/services/reviews_service.py

@@ -45,13 +45,8 @@ def add_new_review(
 
 def get_all_reviews_by_movie(movie_id, db: session):
     try:
-        reviews = db.query(Review).filter(Review.movie_id == movie_id).all()
-        average_reviews = (
-            sum([review.review for review in reviews]) / len(reviews)
-            if len(reviews) > 0
-            else sum([review.review for review in reviews])
-        )
-        return {"reviews": reviews, "average_reviews": average_reviews}
+        return db.query(Review).filter(Review.movie_id == movie_id).all()
+
     except Exception as e:
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e)