73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
from django.http import HttpResponse, JsonResponse
|
|
from django.contrib.auth.models import User
|
|
from rest_framework import permissions, viewsets
|
|
from knox.auth import TokenAuthentication
|
|
from rest_framework.decorators import action
|
|
from rest_framework.exceptions import NotFound
|
|
|
|
from movie_manager.models import Movie, MovieList
|
|
from movie_manager.serializers import MovieListSerializer, MovieSerializer
|
|
|
|
|
|
# Create your views here.
|
|
class MovieViewset(viewsets.ModelViewSet):
|
|
fields = '__all__'
|
|
queryset = Movie.objects.all().order_by("title")
|
|
authentication_classes = [TokenAuthentication]
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
serializer_class = MovieSerializer
|
|
|
|
class MovieListViewset(viewsets.ModelViewSet):
|
|
fields = '__all__'
|
|
queryset = MovieList.objects.all().order_by("name")
|
|
authentication_classes = [TokenAuthentication]
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
serializer_class = MovieListSerializer
|
|
|
|
|
|
def retrieve(self, request, pk=None, *args, **kwargs):
|
|
movie_list = MovieList.objects.get(pk=pk)
|
|
return JsonResponse(MovieListSerializer(movie_list).data)
|
|
|
|
|
|
def update(self, request, pk=None, *args, **kwargs):
|
|
movie_list = MovieList.objects.get(pk=pk)
|
|
movie_list.name = request.data.get('name')
|
|
movie_list.owner = User.objects.get(pk=request.data.get("owner"))
|
|
|
|
if request.data.get('movies'):
|
|
movie_ids = request.data.get('movies')
|
|
for movie_id in movie_ids:
|
|
try:
|
|
movie = Movie.objects.get(pk=movie_id)
|
|
movie_list.movies.add(movie)
|
|
except Movie.DoesNotExist:
|
|
raise NotFound(f"Movie {movie_id} does not exist")
|
|
|
|
removed_movies = Movie.objects.exclude(id__in=movie_ids)
|
|
for removed_movie in removed_movies:
|
|
removed_movie.delete()
|
|
|
|
movie_list.save()
|
|
|
|
return JsonResponse(MovieListSerializer(movie_list).data)
|
|
|
|
@action(detail=True, methods=['put', 'delete'], url_path='movie/(?P<movie_id>[0-9]+)')
|
|
def add_movie(self, request, pk=None, movie_id=None, *args, **kwargs):
|
|
if request.method == 'DELETE':
|
|
return self.remove_movie(request, pk, movie_id)
|
|
|
|
movie_list = MovieList.objects.get(pk=pk)
|
|
movie = Movie.objects.get(pk=movie_id)
|
|
movie_list.movies.add(movie)
|
|
|
|
return JsonResponse(MovieListSerializer(movie_list).data)
|
|
|
|
def remove_movie(self, request, pk=None, movie_id=None, *args, **kwargs):
|
|
movie_list = MovieList.objects.get(pk=pk)
|
|
movie = Movie.objects.get(pk=movie_id)
|
|
movie_list.movies.remove(movie)
|
|
|
|
return JsonResponse(MovieListSerializer(movie_list).data)
|