-
Notifications
You must be signed in to change notification settings - Fork 722
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
4 changed files
with
170 additions
and
48 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,10 +1,36 @@ | ||
from django.db import models | ||
|
||
|
||
class Actor(models.Model): | ||
first_name = models.CharField(max_length=50) | ||
last_name = models.CharField(max_length=50) | ||
|
||
def __str__(self): | ||
return f"{self.first_name} {self.last_name}" | ||
|
||
|
||
class Genre(models.Model): | ||
name = models.CharField(max_length=100, unique=True) | ||
|
||
def __str__(self): | ||
return self.name | ||
|
||
|
||
class CinemaHall(models.Model): | ||
name = models.CharField(max_length=100) | ||
rows = models.PositiveIntegerField() | ||
seats_in_row = models.PositiveIntegerField() | ||
|
||
def __str__(self): | ||
return self.name | ||
|
||
|
||
class Movie(models.Model): | ||
title = models.CharField(max_length=255) | ||
description = models.TextField() | ||
duration = models.IntegerField() | ||
actors = models.ManyToManyField(Actor, related_name="movies") | ||
genres = models.ManyToManyField(Genre, related_name="movies") | ||
|
||
def __str__(self): | ||
return self.title |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,24 +1,56 @@ | ||
from rest_framework import serializers | ||
from cinema.models import Movie, Actor, Genre, CinemaHall | ||
|
||
from cinema.models import Movie | ||
|
||
class ActorSerializer(serializers.ModelSerializer): | ||
class Meta: | ||
model = Actor | ||
fields = ["id", "first_name", "last_name"] | ||
|
||
class MovieSerializer(serializers.Serializer): | ||
id = serializers.IntegerField(read_only=True) | ||
title = serializers.CharField(max_length=255) | ||
description = serializers.CharField() | ||
duration = serializers.IntegerField() | ||
|
||
class GenreSerializer(serializers.ModelSerializer): | ||
class Meta: | ||
model = Genre | ||
fields = ["id", "name"] | ||
|
||
|
||
class CinemaHallSerializer(serializers.ModelSerializer): | ||
class Meta: | ||
model = CinemaHall | ||
fields = ["id", "name", "rows", "seats_in_row"] | ||
|
||
|
||
class MovieSerializer(serializers.ModelSerializer): | ||
actors = serializers.ListField( | ||
child=serializers.IntegerField(), write_only=True, required=False | ||
) | ||
genres = serializers.ListField( | ||
child=serializers.IntegerField(), write_only=True, required=False | ||
) | ||
|
||
class Meta: | ||
model = Movie | ||
fields = ["id", "title", "description", "duration", "actors", "genres"] | ||
|
||
def create(self, validated_data): | ||
return Movie.objects.create(**validated_data) | ||
actors = validated_data.pop("actors", []) | ||
genres = validated_data.pop("genres", []) | ||
movie = Movie.objects.create(**validated_data) | ||
movie.actors.set(actors) | ||
movie.genres.set(genres) | ||
return movie | ||
|
||
def update(self, instance, validated_data): | ||
instance.title = validated_data.get("title", instance.title) | ||
instance.description = validated_data.get( | ||
"description", instance.description | ||
) | ||
instance.duration = validated_data.get("duration", instance.duration) | ||
actors = validated_data.pop("actors", None) | ||
genres = validated_data.pop("genres", None) | ||
|
||
instance.save() | ||
for attr, value in validated_data.items(): | ||
setattr(instance, attr, value) | ||
|
||
if actors is not None: | ||
instance.actors.set(actors) | ||
if genres is not None: | ||
instance.genres.set(genres) | ||
|
||
instance.save() | ||
return instance |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,10 +1,19 @@ | ||
from django.urls import path | ||
from rest_framework.routers import DefaultRouter | ||
from cinema.views import ( | ||
MovieViewSet, GenreList, GenreDetail, | ||
CinemaHallViewSet, ActorList, ActorDetail | ||
) | ||
|
||
from cinema.views import movie_list, movie_detail | ||
|
||
urlpatterns = [ | ||
path("movies/", movie_list, name="movie-list"), | ||
path("movies/<int:pk>/", movie_detail, name="movie-detail"), | ||
] | ||
router = DefaultRouter() | ||
router.register("movies", MovieViewSet, basename="movie") | ||
router.register("cinema_halls", CinemaHallViewSet, basename="cinema_hall") | ||
|
||
app_name = "cinema" | ||
|
||
urlpatterns = [ | ||
path("genres/", GenreList.as_view(), name="genre-list"), | ||
path("genres/<int:pk>/", GenreDetail.as_view(), name="genre-detail"), | ||
path("actors/", ActorList.as_view(), name="actor-list"), | ||
path("actors/<int:pk>/", ActorDetail.as_view(), name="actor-detail"), | ||
] + router.urls |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,45 +1,100 @@ | ||
from rest_framework.decorators import api_view | ||
from rest_framework.viewsets import GenericViewSet | ||
from rest_framework.mixins import ( | ||
ListModelMixin, CreateModelMixin, RetrieveModelMixin, | ||
UpdateModelMixin, DestroyModelMixin | ||
) | ||
from rest_framework.generics import ( | ||
ListCreateAPIView, RetrieveUpdateDestroyAPIView | ||
) | ||
from rest_framework.views import APIView | ||
from rest_framework.response import Response | ||
from rest_framework import status | ||
|
||
from django.shortcuts import get_object_or_404 | ||
|
||
from cinema.models import Movie | ||
from cinema.serializers import MovieSerializer | ||
from cinema.models import Actor, Genre, CinemaHall, Movie | ||
from cinema.serializers import ( | ||
ActorSerializer, GenreSerializer, | ||
CinemaHallSerializer, MovieSerializer | ||
) | ||
|
||
from rest_framework.viewsets import ModelViewSet | ||
|
||
|
||
@api_view(["GET", "POST"]) | ||
def movie_list(request): | ||
if request.method == "GET": | ||
movies = Movie.objects.all() | ||
serializer = MovieSerializer(movies, many=True) | ||
return Response(serializer.data, status=status.HTTP_200_OK) | ||
class GenreList(APIView): | ||
def get(self, request): | ||
genres = Genre.objects.all() | ||
serializer = GenreSerializer(genres, many=True) | ||
return Response(serializer.data) | ||
|
||
if request.method == "POST": | ||
serializer = MovieSerializer(data=request.data) | ||
def post(self, request): | ||
serializer = GenreSerializer(data=request.data) | ||
if serializer.is_valid(): | ||
serializer.save() | ||
return Response(serializer.data, status=status.HTTP_201_CREATED) | ||
return Response(serializer.data, status=201) | ||
return Response(serializer.errors, status=400) | ||
|
||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) | ||
|
||
class GenreDetail(APIView): | ||
def get(self, request, pk): | ||
genre = get_object_or_404(Genre, pk=pk) | ||
serializer = GenreSerializer(genre) | ||
return Response(serializer.data) | ||
|
||
@api_view(["GET", "PUT", "DELETE"]) | ||
def movie_detail(request, pk): | ||
movie = get_object_or_404(Movie, pk=pk) | ||
|
||
if request.method == "GET": | ||
serializer = MovieSerializer(movie) | ||
return Response(serializer.data, status=status.HTTP_200_OK) | ||
def put(self, request, pk): | ||
genre = get_object_or_404(Genre, pk=pk) | ||
serializer = GenreSerializer(genre, data=request.data) | ||
if serializer.is_valid(): | ||
serializer.save() | ||
return Response(serializer.data) | ||
return Response(serializer.errors, status=400) | ||
|
||
if request.method == "PUT": | ||
serializer = MovieSerializer(movie, data=request.data) | ||
def patch(self, request, pk): | ||
genre = get_object_or_404(Genre, pk=pk) | ||
serializer = GenreSerializer(genre, data=request.data, partial=True) | ||
if serializer.is_valid(): | ||
serializer.save() | ||
return Response(serializer.data, status=status.HTTP_200_OK) | ||
return Response(serializer.data) | ||
return Response(serializer.errors, status=400) | ||
|
||
def delete(self, request, pk): | ||
genre = get_object_or_404(Genre, pk=pk) | ||
genre.delete() | ||
return Response(status=204) | ||
|
||
|
||
class CinemaHallViewSet( | ||
GenericViewSet, ListModelMixin, CreateModelMixin, | ||
RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin | ||
): | ||
queryset = CinemaHall.objects.all() | ||
serializer_class = CinemaHallSerializer | ||
|
||
|
||
class ActorList(ListCreateAPIView): | ||
queryset = Actor.objects.all() | ||
serializer_class = ActorSerializer | ||
|
||
|
||
class ActorDetail(RetrieveUpdateDestroyAPIView): | ||
queryset = Actor.objects.all() | ||
serializer_class = ActorSerializer | ||
|
||
|
||
class MovieViewSet(ModelViewSet): | ||
queryset = Movie.objects.all() | ||
serializer_class = MovieSerializer | ||
|
||
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) | ||
def perform_create(self, serializer): | ||
actors = self.request.data.get("actors", []) | ||
genres = self.request.data.get("genres", []) | ||
movie = serializer.save() | ||
movie.actors.set(actors) | ||
movie.genres.set(genres) | ||
|
||
if request.method == "DELETE": | ||
movie.delete() | ||
return Response(status=status.HTTP_204_NO_CONTENT) | ||
def perform_update(self, serializer): | ||
actors = self.request.data.get("actors", None) | ||
genres = self.request.data.get("genres", None) | ||
movie = serializer.save() | ||
if actors is not None: | ||
movie.actors.set(actors) | ||
if genres is not None: | ||
movie.genres.set(genres) |