updated movie fields and added movies to admin

This commit is contained in:
Edward Tirado Jr 2025-04-20 16:49:52 -05:00
parent a58198b5c9
commit 5daa017860
9 changed files with 184 additions and 18 deletions

View file

@ -3,6 +3,7 @@ import json
from django.http import JsonResponse
from django.contrib.auth.models import User
from django.utils.dateparse import parse_datetime
from rest_framework import permissions, viewsets
from knox.auth import TokenAuthentication
from rest_framework.decorators import action, api_view
@ -40,6 +41,14 @@ class MovieListViewset(viewsets.ModelViewSet):
serializer_class = MovieListSerializer
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 retrieve(self, request, pk=None, *args, **kwargs):
movie_list = MovieList.objects.get(pk=pk)
return JsonResponse(MovieListSerializer(movie_list).data)
@ -87,17 +96,18 @@ class MovieListViewset(viewsets.ModelViewSet):
poster=movie["poster"],
plot=movie["plot"],
genre=movie["genre"],
critic_score=movie["imdb_rating"],
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)
movie_list.movies.add(new_movie)
return JsonResponse(MovieListSerializer(movie_list).data)
def remove_movie(self, request, pk=None, movie_id=None, *args, **kwargs):
movie = Movie.objects.get(pk=movie_id)
def remove_movie(self, 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)
@ -158,12 +168,15 @@ class ShowingViewset(viewsets.ModelViewSet):
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=request.data.get("showtime"),
showtime=showtime,
public=request.data.get("public"),
owner=User.objects.get(pk=request.data.get("owner")),
owner=request.user,
)
schedule.showings.add(showing)