updated movie fields and added movies to admin
This commit is contained in:
parent
a58198b5c9
commit
5daa017860
9 changed files with 184 additions and 18 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue