From ba590d885d60f9cb50b2ae633b93377a6d746180 Mon Sep 17 00:00:00 2001 From: Edward Tirado Jr Date: Wed, 23 Apr 2025 15:37:03 -0500 Subject: [PATCH] added support for refreshing movie data --- movie_manager/views.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/movie_manager/views.py b/movie_manager/views.py index e7b961b..78d5412 100644 --- a/movie_manager/views.py +++ b/movie_manager/views.py @@ -33,6 +33,26 @@ class MovieViewset(viewsets.ModelViewSet): serializer_class = MovieSerializer + def update(self, request, pk=None, *args, **kwargs): + omdb = OMDb() + updated_movie = omdb.search(request.data.get("imdb_id"), {"type": "imdb_id"}) + + movie = Movie.objects.get(pk=pk) + + movie.title = updated_movie["title"] + movie.actors = updated_movie["actors"] + movie.year = updated_movie["year"] + movie.critic_scores = updated_movie["critic_scores"] + movie.mpaa_rating = updated_movie["mpaa_rating"] + movie.director = updated_movie["director"] + movie.poster = updated_movie["poster"] + movie.plot = updated_movie["plot"] + movie.genre = updated_movie["genre"] + + movie.save() + + return JsonResponse(MovieSerializer(movie).data) + class MovieListViewset(viewsets.ModelViewSet): queryset = MovieList.objects.all().order_by("name")