solving

db/models.py

@@ -1,4 +1,9 @@
+from django.contrib.auth.models import AbstractUser
+from django.core.exceptions import ValidationError
 from django.db import models
+from django.db.models import UniqueConstraint
+
+import settings
 
 
 class Genre(models.Model):
@@ -22,6 +27,12 @@ class Movie(models.Model):
  actors = models.ManyToManyField(to=Actor, related_name="movies")
  genres = models.ManyToManyField(to=Genre, related_name="movies")
 
+
+ class Meta:
+ indexes = [
+ models.Index(fields=["title"]),
+ ]
+
  def __str__(self) -> str:
  return self.title
 
@@ -50,3 +61,65 @@ class MovieSession(models.Model):
 
  def __str__(self) -> str:
  return f"{self.movie.title} {str(self.show_time)}"
+
+
+class User(AbstractUser):
+ pass
+
+
+class Order(models.Model):
+ created_at = models.DateTimeField(auto_now_add=True)
+ user = models.ForeignKey(
+ settings.AUTH_USER_MODEL,
+ on_delete=models.CASCADE,
+ related_name="orders"
+ )
+
+ class Meta:
+ ordering = ["-created_at"]
+
+ def __str__(self) -> str:
+ return f"{self.created_at.strftime("%Y-%m-%d %H:%M:%S")}"
+
+
+class Ticket(models.Model):
+ movie_session = models.ForeignKey(to=MovieSession, on_delete=models.CASCADE)
+ order = models.ForeignKey(Order, on_delete=models.CASCADE)
+ row = models.IntegerField()
+ seat = models.IntegerField()
+
+
+ class Meta:
+ constraints = [
+ UniqueConstraint(
+ fields=["movie_session", "row", "seat"],
+ name="unique_movie_session"
+ )
+ ]
+
+ def clean(self):
+ cinema_hall = self.movie_session.cinema_hall
+
+ if not 0 < self.row <= cinema_hall.rows:
+ raise ValidationError({
+ "row": f"row number must be in available range: "
+ f"(1, rows): (1, {cinema_hall.rows})"
+ })
+
+ if not 0 < self.seat <= cinema_hall.seats_in_row:
+ raise ValidationError({
+ "seat": f"seat number must be in available range: "
+ f"(1, seats_in_row): (1, {cinema_hall.seats_in_row})"
+ })
+
+ def save(self, *args, **kwargs):
+ self.full_clean()
+ return super().save(*args, **kwargs)
+
+
+ def __str__(self) -> str:
+ return (
+ self.movie_session.movie.title + " "
+ + self.movie_session.show_time.strftime("%Y-%m-%d %H:%M:%S")
+ + f" (row: {self.row}, seat: {self.seat})"
+ )

services/movie.py

@@ -1,3 +1,4 @@
+from django.db import transaction
 from django.db.models import QuerySet
 
 from db.models import Movie
@@ -6,6 +7,7 @@
 def get_movies(
  genres_ids: list[int] = None,
  actors_ids: list[int] = None,
+ title: str | None = None
 ) -> QuerySet:
  queryset = Movie.objects.all()
 
@@ -15,6 +17,9 @@ def get_movies(
  if actors_ids:
  queryset = queryset.filter(actors__id__in=actors_ids)
 
+ if title:
+ queryset = queryset.filter(title__icontains=title)
+
  return queryset
 
 
@@ -28,13 +33,14 @@ def create_movie(
  genres_ids: list = None,
  actors_ids: list = None,
 ) -> Movie:
- movie = Movie.objects.create(
- title=movie_title,
- description=movie_description,
- )
- if genres_ids:
- movie.genres.set(genres_ids)
- if actors_ids:
- movie.actors.set(actors_ids)
-
- return movie
+ with transaction.atomic():
+ movie = Movie.objects.create(
+ title=movie_title,
+ description=movie_description,
+ )
+ if genres_ids:
+ movie.genres.set(genres_ids)
+ if actors_ids:
+ movie.actors.set(actors_ids)
+
+ return movie

services/movie_session.py

@@ -1,6 +1,6 @@
 from django.db.models import QuerySet
 
-from db.models import MovieSession
+from db.models import MovieSession, Ticket
 
 
 def create_movie_session(
@@ -42,3 +42,11 @@ def update_movie_session(
 
 def delete_movie_session_by_id(session_id: int) -> None:
  MovieSession.objects.get(id=session_id).delete()
+
+
+def get_taken_seats(movie_session_id: int) -> list:
+ if MovieSession.objects.filter(id=movie_session_id).exists():
+ return list(Ticket.objects.filter(
+ movie_session__id=movie_session_id
+ ).values("row", "seat"))
+ print("Movie session does not exist")

services/order.py

@@ -0,0 +1,31 @@
+from django.contrib.auth import get_user_model
+from django.db import transaction
+from django.db.models import QuerySet
+
+from db.models import Order, Ticket
+
+
+def create_order(
+ tickets: list[dict],
+ username: str,
+ date: str | None = None
+) -> None:
+ user = get_user_model().objects.get(username=username)
+ new_data = {}
+ if date:
+ new_data["created_at"] = date
+
+ with transaction.atomic():
+ order = Order.objects.create(user=user)
+ Order.objects.filter(id=order.id).update(**new_data)
+ for ticket in tickets:
+ ticket["movie_session_id"] = ticket["movie_session"]
+ del ticket["movie_session"]
+ Ticket.objects.create(order=order, **ticket)
+
+
+def get_orders(username: str | None = None) -> QuerySet:
+ kwargs = {}
+ if username:
+ kwargs["user"] = get_user_model().objects.get(username=username)
+ return Order.objects.filter(**kwargs)

services/user.py

@@ -0,0 +1,68 @@
+from django.contrib.auth import get_user_model
+
+from db.models import User
+
+
+def create_user(
+ username: str,
+ password: str,
+ email: str | None = None,
+ first_name: str | None = None,
+ last_name: str | None = None
+) -> None:
+ if get_user_model().objects.filter(username=username).exists():
+ print("Username already exists")
+ return
+
+ users_data = {}
+ if email:
+ users_data["email"] = email
+ if first_name:
+ users_data["first_name"] = first_name
+ if last_name:
+ users_data["last_name"] = last_name
+
+ get_user_model().objects.create_user(
+ username=username,
+ password=password,
+ **users_data
+ )
+
+
+def get_user(user_id: int) -> User:
+ return get_user_model().objects.filter(pk=user_id).first()
+
+
+def update_user(
+ user_id: int,
+ username: str | None = None,
+ password: str | None = None,
+ email: str | None = None,
+ first_name: str | None = None,
+ last_name: str | None = None
+) -> None:
+ if not get_user_model().objects.filter(pk=user_id).exists():
+ print("User not found")
+ return
+
+ user = get_user_model().objects.get(pk=user_id)
+
+ if username:
+ if (
+ user.username != username
+ and get_user_model().objects.filter(username=username).exists()
+ ):
+ print("Username already exists")
+ return
+ user.username = username
+
+ if password:
+ user.set_password(password)
+ if email:
+ user.email = email
+ if first_name:
+ user.first_name = first_name
+ if last_name:
+ user.last_name = last_name
+
+ user.save()

settings.py

@@ -25,4 +25,8 @@
 
 INSTALLED_APPS = [
  "db",
+ "django.contrib.auth",
+ "django.contrib.contenttypes",
 ]
+
+AUTH_USER_MODEL = "db.User"