reorganized viewsets into their own files
This commit is contained in:
parent
83982877c3
commit
2e7dbe4ddc
13 changed files with 330 additions and 277 deletions
11
movie_manager/viewsets/__init__.py
Normal file
11
movie_manager/viewsets/__init__.py
Normal file
|
@ -0,0 +1,11 @@
|
|||
from .movie import MovieViewset
|
||||
from .movie_list import MovieListViewset
|
||||
from .schedule import ScheduleViewset
|
||||
from .showing import ShowingViewset
|
||||
|
||||
__all__ = [
|
||||
"MovieViewset",
|
||||
"MovieListViewset",
|
||||
"ScheduleViewset",
|
||||
"ShowingViewset",
|
||||
]
|
37
movie_manager/viewsets/movie.py
Normal file
37
movie_manager/viewsets/movie.py
Normal file
|
@ -0,0 +1,37 @@
|
|||
from django.http import JsonResponse
|
||||
from rest_framework import permissions, viewsets
|
||||
|
||||
from movie_db.db_providers.omdb import OMDb
|
||||
from movie_manager.models import Movie
|
||||
|
||||
from knox.auth import TokenAuthentication
|
||||
|
||||
from movie_manager.serializers import MovieSerializer
|
||||
|
||||
|
||||
class MovieViewset(viewsets.ModelViewSet):
|
||||
queryset = Movie.objects.all().order_by("title")
|
||||
authentication_classes = [TokenAuthentication]
|
||||
permission_classes = [permissions.IsAuthenticated]
|
||||
|
||||
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)
|
120
movie_manager/viewsets/movie_list.py
Normal file
120
movie_manager/viewsets/movie_list.py
Normal file
|
@ -0,0 +1,120 @@
|
|||
from django.http import JsonResponse
|
||||
from django.db import models
|
||||
from django.contrib.auth.models import User
|
||||
from rest_framework import permissions, viewsets
|
||||
from rest_framework.decorators import action
|
||||
from rest_framework.exceptions import NotFound
|
||||
|
||||
from movie_db.db_providers.omdb import OMDb
|
||||
from movie_manager.models import MovieList, Movie
|
||||
|
||||
from knox.auth import TokenAuthentication
|
||||
|
||||
from movie_manager.permissions import ReadOnly
|
||||
from movie_manager.serializers import MovieListSerializer, MovieListListSerializer
|
||||
|
||||
|
||||
class MovieListViewset(viewsets.ModelViewSet):
|
||||
queryset = MovieList.objects.all()
|
||||
authentication_classes = [TokenAuthentication]
|
||||
permission_classes = [permissions.IsAuthenticated | ReadOnly]
|
||||
|
||||
def get_serializer_class(self):
|
||||
if self.action == "list":
|
||||
return MovieListListSerializer
|
||||
else:
|
||||
return MovieListSerializer
|
||||
|
||||
def get_queryset(self):
|
||||
base_qs = MovieList.objects.all()
|
||||
|
||||
if self.action == "list":
|
||||
if self.request.user.is_authenticated:
|
||||
return base_qs.filter(
|
||||
models.Q(public=True) | models.Q(owner=self.request.user)
|
||||
).order_by("name")
|
||||
|
||||
return base_qs.filter(public=True).order_by("name")
|
||||
else:
|
||||
return MovieList.objects.prefetch_related(
|
||||
"movies", "movies__showing_set"
|
||||
).order_by("name")
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(owner=self.request.user)
|
||||
|
||||
def get_permissions(self):
|
||||
if self.action in ["update", "partial_update", "destroy"]:
|
||||
self.permission_classes = [permissions.IsAuthenticated]
|
||||
return super().get_permissions()
|
||||
|
||||
def create(self, request, *args, **kwargs):
|
||||
movie_list = MovieList.objects.create(
|
||||
name=request.data.get("name"),
|
||||
owner=request.user,
|
||||
)
|
||||
|
||||
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<imdb_id>tt[0-9]+)"
|
||||
)
|
||||
def add_movie(self, request, pk=None, imdb_id=None, *args, **kwargs):
|
||||
if request.method == "DELETE":
|
||||
return self.remove_movie(request, pk, imdb_id)
|
||||
|
||||
movie_list = MovieList.objects.get(pk=pk)
|
||||
try:
|
||||
new_movie = Movie.objects.get(imdb_id=imdb_id)
|
||||
except Movie.DoesNotExist:
|
||||
omdb = OMDb()
|
||||
movie = omdb.search(imdb_id, {"type": "imdb_id"})
|
||||
|
||||
new_movie = Movie.objects.create(
|
||||
title=movie["title"],
|
||||
actors=movie["actors"],
|
||||
year=movie["year"],
|
||||
imdb_id=movie["imdb_id"],
|
||||
poster=movie["poster"],
|
||||
plot=movie["plot"],
|
||||
genre=movie["genre"],
|
||||
critic_scores=movie["critic_scores"],
|
||||
mpaa_rating=movie["mpaa_rating"],
|
||||
director=movie["director"],
|
||||
added_by_id=request.user.id,
|
||||
)
|
||||
|
||||
movie_list.movies.add(new_movie)
|
||||
|
||||
return JsonResponse(MovieListSerializer(movie_list).data)
|
||||
|
||||
@staticmethod
|
||||
def remove_movie(request, pk=None, imdb_id=None, *args, **kwargs):
|
||||
movie = Movie.objects.filter(imdb_id=imdb_id).first()
|
||||
|
||||
movie_list = MovieList.objects.get(pk=pk)
|
||||
movie_list.movies.remove(movie)
|
||||
|
||||
return JsonResponse(MovieListSerializer(movie_list).data)
|
58
movie_manager/viewsets/schedule.py
Normal file
58
movie_manager/viewsets/schedule.py
Normal file
|
@ -0,0 +1,58 @@
|
|||
import datetime
|
||||
|
||||
from django.http import JsonResponse
|
||||
from knox.auth import TokenAuthentication
|
||||
from rest_framework import viewsets, permissions
|
||||
|
||||
from movie_manager.models import Schedule, Showing
|
||||
from movie_manager.permissions import ReadOnly
|
||||
from movie_manager.serializers import (
|
||||
ScheduleSerializer,
|
||||
ShowingSerializer,
|
||||
MovieSerializer,
|
||||
)
|
||||
|
||||
|
||||
class ScheduleViewset(viewsets.ModelViewSet):
|
||||
queryset = Schedule.objects.all().order_by("name")
|
||||
authentication_classes = [TokenAuthentication]
|
||||
permission_classes = [permissions.IsAuthenticated | ReadOnly]
|
||||
|
||||
serializer_class = ScheduleSerializer
|
||||
|
||||
def retrieve(self, request, pk=None, *args, **kwargs):
|
||||
# Get the schedule instance
|
||||
instance = self.get_object()
|
||||
now = datetime.datetime.now()
|
||||
# get time from start of day
|
||||
today = datetime.datetime(now.year, now.month, now.day)
|
||||
|
||||
upcoming_showings = Showing.objects.filter(
|
||||
showtime__gte=today, schedule=instance
|
||||
)
|
||||
|
||||
# Create a serialized response
|
||||
serializer = self.get_serializer(instance)
|
||||
data = serializer.data
|
||||
|
||||
# Replace all showings with only future showings
|
||||
data["showings"] = ShowingSerializer(upcoming_showings, many=True).data
|
||||
|
||||
if request.GET.get("past_showings") == "true":
|
||||
past_showings = Showing.objects.filter(
|
||||
showtime__lt=today, schedule=instance
|
||||
)
|
||||
|
||||
# Add both to the response
|
||||
data["past_showings"] = [
|
||||
{
|
||||
"id": showing.id,
|
||||
"showtime": showing.showtime.isoformat(),
|
||||
"movie": MovieSerializer(showing.movie).data,
|
||||
}
|
||||
for showing in past_showings
|
||||
]
|
||||
else:
|
||||
data["past_showings"] = []
|
||||
|
||||
return JsonResponse(data)
|
36
movie_manager/viewsets/showing.py
Normal file
36
movie_manager/viewsets/showing.py
Normal file
|
@ -0,0 +1,36 @@
|
|||
from django.http import JsonResponse
|
||||
from django.utils.dateparse import parse_datetime
|
||||
from knox.auth import TokenAuthentication
|
||||
from rest_framework import viewsets, permissions
|
||||
|
||||
from movie_manager.models import Showing, Movie, Schedule
|
||||
from movie_manager.permissions import ReadOnly
|
||||
from movie_manager.serializers import ShowingSerializer
|
||||
|
||||
|
||||
class ShowingViewset(viewsets.ModelViewSet):
|
||||
queryset = Showing.objects.all().order_by("showtime")
|
||||
authentication_classes = [TokenAuthentication]
|
||||
permission_classes = [permissions.IsAuthenticated | ReadOnly]
|
||||
|
||||
serializer_class = ShowingSerializer
|
||||
|
||||
def create(self, request, *args, **kwargs):
|
||||
movie_id = request.data.get("movie")
|
||||
movie = Movie.objects.get(pk=movie_id)
|
||||
|
||||
schedule_id = request.data.get("schedule")
|
||||
schedule = Schedule.objects.get(pk=schedule_id)
|
||||
|
||||
showtime_str = request.data.get("showtime")
|
||||
showtime = parse_datetime(showtime_str)
|
||||
|
||||
showing = Showing.objects.create(
|
||||
movie=movie,
|
||||
schedule=schedule,
|
||||
showtime=showtime,
|
||||
public=request.data.get("public"),
|
||||
owner=request.user,
|
||||
)
|
||||
|
||||
return JsonResponse(ShowingSerializer(showing).data)
|
Loading…
Add table
Add a link
Reference in a new issue