task done

cinema/models.py

@@ -84,24 +84,34 @@ class Ticket(models.Model):
  row = models.IntegerField()
  seat = models.IntegerField()
 
- def clean(self):
+ @staticmethod
+ def validate_ticket(row, seat, cinema_hall, error):
  for ticket_attr_value, ticket_attr_name, cinema_hall_attr_name in [
- (self.row, "row", "rows"),
- (self.seat, "seat", "seats_in_row"),
+ (row, "row", "rows"),
+ (seat, "seat", "seats_in_row"),
  ]:
  count_attrs = getattr(
- self.movie_session.cinema_hall, cinema_hall_attr_name
+ cinema_hall, cinema_hall_attr_name
  )
  if not (1 <= ticket_attr_value <= count_attrs):
- raise ValidationError(
+ raise error(
  {
  ticket_attr_name: f"{ticket_attr_name} "
- f"number must be in available range: "
- f"(1, {cinema_hall_attr_name}): "
- f"(1, {count_attrs})"
+ f"number must be "
+ f"in available range: "
+ f"(1, {cinema_hall_attr_name}): "
+ f"(1, {count_attrs})"
  }
  )
 
+ def clean(self):
+ Ticket.validate_ticket(
+ self.row,
+ self.seat,
+ self.movie_session.cinema_hall,
+ ValidationError
+ )
+
  def save(
  self,
  force_insert=False,

cinema/serializers.py

@@ -1,6 +1,16 @@
+from django.db import transaction
 from rest_framework import serializers
+from rest_framework.exceptions import ValidationError
 
-from cinema.models import Genre, Actor, CinemaHall, Movie, MovieSession
+from cinema.models import (
+ Genre,
+ Actor,
+ CinemaHall,
+ Movie,
+ MovieSession,
+ Order,
+ Ticket
+)
 
 
 class GenreSerializer(serializers.ModelSerializer):
@@ -59,6 +69,9 @@ class MovieSessionListSerializer(MovieSessionSerializer):
  cinema_hall_capacity = serializers.IntegerField(
  source="cinema_hall.capacity", read_only=True
  )
+ tickets_available = serializers.IntegerField(
+ read_only=True
+ )
 
  class Meta:
  model = MovieSession
@@ -68,13 +81,70 @@ class Meta:
  "movie_title",
  "cinema_hall_name",
  "cinema_hall_capacity",
+ "tickets_available"
  )
 
 
 class MovieSessionDetailSerializer(MovieSessionSerializer):
  movie = MovieListSerializer(many=False, read_only=True)
  cinema_hall = CinemaHallSerializer(many=False, read_only=True)
+ taken_places = serializers.SerializerMethodField()
 
  class Meta:
  model = MovieSession
- fields = ("id", "show_time", "movie", "cinema_hall")
+ fields = ("id", "show_time", "movie", "cinema_hall", "taken_places")
+
+ def get_taken_places(self, obj):
+ tickets = Ticket.objects.filter(movie_session=obj)
+ return [{"row": ticket.row, "seat": ticket.seat} for ticket in tickets]
+
+
+class TicketSerializer(serializers.ModelSerializer):
+ movie_session = MovieSessionListSerializer()
+
+ class Meta:
+ model = Ticket
+ fields = ("id", "row", "seat", "movie_session")
+
+ def validate(self, attrs):
+ Ticket.validate_ticket(
+ attrs["row"],
+ attrs["seat"],
+ attrs["movie_session"].cinema_hall,
+ ValidationError
+ )
+ return attrs
+
+
+class TicketCreateSerializer(TicketSerializer):
+ movie_session = serializers.PrimaryKeyRelatedField(
+ queryset=MovieSession.objects.all()
+ )
+
+
+class OrderSerializer(serializers.ModelSerializer):
+ tickets = TicketCreateSerializer(
+ many=True,
+ read_only=False,
+ allow_empty=False
+ )
+
+ class Meta:
+ model = Order
+ fields = ("id", "created_at", "tickets")
+
+ def create(self, validated_data):
+ with transaction.atomic():
+ tickets_data = validated_data.pop("tickets")
+ order = Order.objects.create(**validated_data)
+ for ticket_data in tickets_data:
+ Ticket.objects.create(order=order, **ticket_data)
+ return order
+
+
+class OrderRetrieveSerializer(OrderSerializer):
+ tickets = TicketSerializer(many=True, read_only=True)
+
+ class Meta:
+ model = Order
+ fields = ("id", "created_at", "tickets")

cinema/urls.py

@@ -7,6 +7,7 @@
  CinemaHallViewSet,
  MovieViewSet,
  MovieSessionViewSet,
+ OrderViewSet
 )
 
 router = routers.DefaultRouter()
@@ -15,6 +16,7 @@
 router.register("cinema_halls", CinemaHallViewSet)
 router.register("movies", MovieViewSet)
 router.register("movie_sessions", MovieSessionViewSet)
+router.register("orders", OrderViewSet)
 
 urlpatterns = [path("", include(router.urls))]
 

cinema/views.py

@@ -1,7 +1,8 @@
+from django.db.models import Count, F
 from rest_framework import viewsets
+from rest_framework.pagination import PageNumberPagination
 
-from cinema.models import Genre, Actor, CinemaHall, Movie, MovieSession
-
+from cinema.models import Genre, Actor, CinemaHall, Movie, MovieSession, Order
 from cinema.serializers import (
  GenreSerializer,
  ActorSerializer,
@@ -12,6 +13,8 @@
  MovieDetailSerializer,
  MovieSessionDetailSerializer,
  MovieListSerializer,
+ OrderSerializer,
+ OrderRetrieveSerializer,
 )
 
 
@@ -34,6 +37,33 @@ class MovieViewSet(viewsets.ModelViewSet):
  queryset = Movie.objects.all()
  serializer_class = MovieSerializer
 
+ @staticmethod
+ def _params_to_ints(query_string):
+ return [int(str_id) for str_id in query_string.split(",")]
+
+ def get_queryset(self):
+ queryset = self.queryset
+
+ actors = self.request.query_params.get("actors")
+ genres = self.request.query_params.get("genres")
+ titles = self.request.query_params.get("title")
+
+ if actors:
+ actors = self._params_to_ints(actors)
+ queryset = queryset.filter(actors__id__in=actors)
+ elif genres:
+ genres = self._params_to_ints(genres)
+ queryset = queryset.filter(genres__id__in=genres)
+ elif titles:
+ queryset = queryset.filter(title__icontains=titles)
+
+ if self.action == "list":
+ queryset = queryset.prefetch_related(
+ "genres",
+ "actors"
+ )
+ return queryset
+
  def get_serializer_class(self):
  if self.action == "list":
  return MovieListSerializer
@@ -48,6 +78,29 @@ class MovieSessionViewSet(viewsets.ModelViewSet):
  queryset = MovieSession.objects.all()
  serializer_class = MovieSessionSerializer
 
+ def get_queryset(self):
+ queryset = self.queryset
+
+ date = self.request.query_params.get("date")
+ movie = self.request.query_params.get("movie")
+
+ if date:
+ queryset = queryset.filter(show_time__date=date)
+ if movie:
+ queryset = queryset.filter(movie_id=movie)
+
+ if self.action in "list":
+ queryset = (
+ queryset
+ .prefetch_related("movie", "cinema_hall", "tickets")
+ .annotate(
+ capacity=(F("cinema_hall__rows")
+ * F("cinema_hall__seats_in_row")),
+ tickets_available=F("capacity") - Count("tickets"))
+ )
+
+ return queryset
+
  def get_serializer_class(self):
  if self.action == "list":
  return MovieSessionListSerializer
@@ -56,3 +109,32 @@ def get_serializer_class(self):
  return MovieSessionDetailSerializer
 
  return MovieSessionSerializer
+
+
+class OrderSetPagination(PageNumberPagination):
+ page_size = 1
+ page_query_param = "page_size"
+ max_page_size = 20
+
+
+class OrderViewSet(viewsets.ModelViewSet):
+ queryset = Order.objects.all()
+ pagination_class = OrderSetPagination
+
+ def get_queryset(self):
+ queryset = Order.objects.filter(user=self.request.user)
+
+ if self.action == "list":
+ queryset = queryset.prefetch_related(
+ "tickets__movie_session__movie",
+ "tickets__movie_session__cinema_hall"
+ )
+ return queryset
+
+ def get_serializer_class(self):
+ if self.action in ("list", "retrieve"):
+ return OrderRetrieveSerializer
+ return OrderSerializer
+
+ def perform_create(self, serializer):
+ serializer.save(user=self.request.user)

cinema_service/settings.py

@@ -124,7 +124,7 @@
 
 USE_I18N = True
 
-USE_TZ = False
+USE_TZ = True
 
 
 # Static files (CSS, JavaScript, Images)