Solution

db/migrations/0002_user_order_ticket_alter_movie_actors_and_more.py

@@ -0,0 +1,111 @@
+# Generated by Django 4.0.2 on 2024-09-02 12:35
+
+from django.conf import settings
+import django.contrib.auth.models
+import django.contrib.auth.validators
+from django.db import migrations, models
+import django.db.models.deletion
+import django.utils.timezone
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('auth', '0012_alter_user_first_name_max_length'),
+ ('db', '0001_initial'),
+ ]
+
+ operations = [
+ migrations.CreateModel(
+ name='User',
+ fields=[
+ ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+ ('password', models.CharField(max_length=128, verbose_name='password')),
+ ('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
+ ('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
+ ('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
+ ('first_name', models.CharField(blank=True, max_length=150, verbose_name='first name')),
+ ('last_name', models.CharField(blank=True, max_length=150, verbose_name='last name')),
+ ('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
+ ('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
+ ('is_active', models.BooleanField(default=True, help_text='Designates whether this user should be treated as active. Unselect this instead of deleting accounts.', verbose_name='active')),
+ ('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
+ ],
+ options={
+ 'verbose_name': 'user',
+ 'verbose_name_plural': 'users',
+ 'abstract': False,
+ },
+ managers=[
+ ('objects', django.contrib.auth.models.UserManager()),
+ ],
+ ),
+ migrations.CreateModel(
+ name='Order',
+ fields=[
+ ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+ ('created_at', models.DateTimeField(auto_now_add=True)),
+ ],
+ ),
+ migrations.CreateModel(
+ name='Ticket',
+ fields=[
+ ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+ ('row', models.IntegerField()),
+ ('seat', models.IntegerField()),
+ ],
+ ),
+ migrations.AlterField(
+ model_name='movie',
+ name='actors',
+ field=models.ManyToManyField(related_name='movies', to='db.Actor'),
+ ),
+ migrations.AlterField(
+ model_name='movie',
+ name='genres',
+ field=models.ManyToManyField(related_name='movies', to='db.Genre'),
+ ),
+ migrations.AlterField(
+ model_name='moviesession',
+ name='cinema_hall',
+ field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='movie_sessions', to='db.cinemahall'),
+ ),
+ migrations.AlterField(
+ model_name='moviesession',
+ name='movie',
+ field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='movie_sessions', to='db.movie'),
+ ),
+ migrations.AddIndex(
+ model_name='movie',
+ index=models.Index(fields=['title'], name='db_movie_title_5d0841_idx'),
+ ),
+ migrations.AddField(
+ model_name='ticket',
+ name='movie_session',
+ field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='tickets', to='db.moviesession'),
+ ),
+ migrations.AddField(
+ model_name='ticket',
+ name='order',
+ field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='tickets', to='db.order'),
+ ),
+ migrations.AddField(
+ model_name='order',
+ name='user',
+ field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='orders', to=settings.AUTH_USER_MODEL),
+ ),
+ migrations.AddField(
+ model_name='user',
+ name='groups',
+ field=models.ManyToManyField(blank=True, help_text='The groups this user belongs to. A user will get all permissions granted to each of their groups.', related_name='user_set', related_query_name='user', to='auth.Group', verbose_name='groups'),
+ ),
+ migrations.AddField(
+ model_name='user',
+ name='user_permissions',
+ field=models.ManyToManyField(blank=True, help_text='Specific permissions for this user.', related_name='user_set', related_query_name='user', to='auth.Permission', verbose_name='user permissions'),
+ ),
+ migrations.AddConstraint(
+ model_name='ticket',
+ constraint=models.UniqueConstraint(fields=('movie_session', 'row', 'seat'), name='unique_movie_session_row_seat'),
+ ),
+ ]

db/migrations/0003_alter_order_options.py

@@ -0,0 +1,17 @@
+# Generated by Django 4.0.2 on 2024-09-02 13:32
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+ dependencies = [
+ ('db', '0002_user_order_ticket_alter_movie_actors_and_more'),
+ ]
+
+ operations = [
+ migrations.AlterModelOptions(
+ name='order',
+ options={'ordering': ('-created_at',)},
+ ),
+ ]

db/models.py

@@ -1,8 +1,15 @@
+from django.contrib.auth.models import AbstractUser
+from django.core.exceptions import ValidationError
 from django.db import models
 
+import settings
+
 
 class Genre(models.Model):
- name = models.CharField(max_length=255, unique=True)
+ name = models.CharField(
+ max_length=255,
+ unique=True
+ )
 
  def __str__(self) -> str:
  return self.name
@@ -19,8 +26,17 @@ def __str__(self) -> str:
 class Movie(models.Model):
  title = models.CharField(max_length=255)
  description = models.TextField()
- actors = models.ManyToManyField(to=Actor, related_name="movies")
- genres = models.ManyToManyField(to=Genre, related_name="movies")
+ 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
@@ -42,11 +58,86 @@ def __str__(self) -> str:
 class MovieSession(models.Model):
  show_time = models.DateTimeField()
  cinema_hall = models.ForeignKey(
- to=CinemaHall, on_delete=models.CASCADE, related_name="movie_sessions"
+ to=CinemaHall,
+ on_delete=models.CASCADE,
+ related_name="movie_sessions",
  )
  movie = models.ForeignKey(
- to=Movie, on_delete=models.CASCADE, related_name="movie_sessions"
+ to=Movie,
+ on_delete=models.CASCADE,
+ related_name="movie_sessions",
  )
 
  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(
+ to=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,
+ related_name="tickets",
+ )
+ order = models.ForeignKey(
+ to=Order,
+ on_delete=models.CASCADE,
+ related_name="tickets",
+ )
+ row = models.IntegerField()
+ seat = models.IntegerField()
+
+ class Meta:
+ constraints = [
+ models.UniqueConstraint(
+ fields=["movie_session", "row", "seat"],
+ name="unique_movie_session_row_seat",
+ ),
+ ]
+
+ def __str__(self) -> str:
+ return (
+ f"{self.movie_session.movie.title} "
+ f"{self.movie_session.show_time} "
+ f"(row: {self.row}, seat: {self.seat})"
+ )
+
+ def clean(self) -> None:
+ if not 1 <= self.seat <= self.movie_session.cinema_hall.seats_in_row:
+ raise ValidationError(
+ {
+ "seat":
+ "seat number must be in available range: "
+ "(1, seats_in_row): "
+ f"(1, {self.movie_session.cinema_hall.seats_in_row})"
+ }
+ )
+ if not 1 <= self.row <= self.movie_session.cinema_hall.rows:
+ raise ValidationError(
+ {
+ "row": "row number must be in available range: (1, rows): "
+ f"(1, {self.movie_session.cinema_hall.rows})"
+ }
+ )
+
+ def save(self, *args, **kwargs) -> None:
+ self.full_clean()
+ return super().save(*args, **kwargs)

services/cinema_hall.py

@@ -8,8 +8,12 @@ def get_cinema_halls() -> QuerySet:
 
 
 def create_cinema_hall(
- hall_name: str, hall_rows: int, hall_seats_in_row: int
+ hall_name: str,
+ hall_rows: int,
+ hall_seats_in_row: int
 ) -> CinemaHall:
  return CinemaHall.objects.create(
- name=hall_name, rows=hall_rows, seats_in_row=hall_seats_in_row
+ name=hall_name,
+ rows=hall_rows,
+ seats_in_row=hall_seats_in_row
  )

services/movie.py

@@ -1,11 +1,15 @@
+from typing import Optional
+
+from django.db import transaction
 from django.db.models import QuerySet
 
 from db.models import Movie
 
 
 def get_movies(
- genres_ids: list[int] = None,
- actors_ids: list[int] = None,
+ genres_ids: Optional[list[int]] = None,
+ actors_ids: Optional[list[int]] = None,
+ title: Optional[str] = None,
 ) -> QuerySet:
  queryset = Movie.objects.all()
 
@@ -15,18 +19,22 @@ def get_movies(
  if actors_ids:
  queryset = queryset.filter(actors__id__in=actors_ids)
 
+ if title:
+ queryset = queryset.filter(title__icontains=title)
+
  return queryset
 
 
 def get_movie_by_id(movie_id: int) -> Movie:
  return Movie.objects.get(id=movie_id)
 
 
+@transaction.atomic
 def create_movie(
  movie_title: str,
  movie_description: str,
- genres_ids: list = None,
- actors_ids: list = None,
+ genres_ids: Optional[list] = None,
+ actors_ids: Optional[list] = None,
 ) -> Movie:
  movie = Movie.objects.create(
  title=movie_title,

services/movie_session.py

@@ -1,10 +1,14 @@
+from typing import Optional
+
 from django.db.models import QuerySet
 
-from db.models import MovieSession
+from db.models import MovieSession, Ticket
 
 
 def create_movie_session(
- movie_show_time: str, movie_id: int, cinema_hall_id: int
+ movie_show_time: str,
+ movie_id: int,
+ cinema_hall_id: int
 ) -> MovieSession:
  return MovieSession.objects.create(
  show_time=movie_show_time,
@@ -13,7 +17,7 @@ def create_movie_session(
  )
 
 
-def get_movies_sessions(session_date: str = None) -> QuerySet:
+def get_movies_sessions(session_date: Optional[str] = None) -> QuerySet:
  queryset = MovieSession.objects.all()
  if session_date:
  queryset = queryset.filter(show_time__date=session_date)
@@ -26,9 +30,9 @@ def get_movie_session_by_id(movie_session_id: int) -> MovieSession:
 
 def update_movie_session(
  session_id: int,
- show_time: str = None,
- movie_id: int = None,
- cinema_hall_id: int = None,
+ show_time: Optional[str] = None,
+ movie_id: Optional[int] = None,
+ cinema_hall_id: Optional[int] = None,
 ) -> None:
  movie_session = MovieSession.objects.get(id=session_id)
  if show_time:
@@ -42,3 +46,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[dict]:
+ return list(
+ Ticket.objects
+ .filter(movie_session__id=movie_session_id)
+ .values("row", "seat")
+ )