eeettt5

db/migrations/0002_user_order_alter_movie_actors_alter_movie_genres_and_more.py

@@ -0,0 +1,95 @@
+# Generated by Django 4.0.2 on 2023-11-14 21:06
+
+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')),
+ ('groups', 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')),
+ ('user_permissions', 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')),
+ ],
+ 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)),
+ ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='orders', to=settings.AUTH_USER_MODEL)),
+ ],
+ options={
+ 'ordering': ['-created_at'],
+ },
+ ),
+ 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='movie',
+ name='title',
+ field=models.CharField(db_index=True, max_length=255),
+ ),
+ 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.CreateModel(
+ name='Ticket',
+ fields=[
+ ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+ ('row', models.IntegerField()),
+ ('seat', models.IntegerField()),
+ ('movie_session', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='db.moviesession')),
+ ('order', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='db.order')),
+ ],
+ ),
+ migrations.AddConstraint(
+ model_name='ticket',
+ constraint=models.UniqueConstraint(fields=('row', 'seat', 'movie_session'), name='unique_order_row_seat_order_session'),
+ ),
+ ]

db/models.py

@@ -1,3 +1,6 @@
+from django.conf import settings
+from django.contrib.auth.models import AbstractUser
+from django.core.exceptions import ValidationError
 from django.db import models
 
 
@@ -17,7 +20,7 @@ def __str__(self) -> str:
 
 
 class Movie(models.Model):
- title = models.CharField(max_length=255)
+ title = models.CharField(max_length=255, db_index=True)
  description = models.TextField()
  actors = models.ManyToManyField(to=Actor, related_name="movies")
  genres = models.ManyToManyField(to=Genre, related_name="movies")
@@ -49,4 +52,75 @@ class MovieSession(models.Model):
  )
 
  def __str__(self) -> str:
- return f"{self.movie.title} {str(self.show_time)}"
+ return (
+ f"{self.movie.title} "
+ f"{self.show_time} "
+ f"{self.cinema_hall.rows} "
+ f"{self.cinema_hall.seats_in_row}"
+ )
+
+
+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 str(self.created_at)
+
+
+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()
+
+ def clean(self) -> None:
+ max_row = self.movie_session.cinema_hall.rows
+ max_seat = self.movie_session.cinema_hall.seats_in_row
+
+ if self.row > max_row:
+ raise ValidationError(
+ {"row":
+ f"row number must be in available range: "
+ f"(1, rows): (1, {max_row})"}
+ )
+
+ if self.seat > max_seat:
+ raise ValidationError(
+ {"seat": f"seat number must "
+ f"be in available range: (1, seats_in_row): "
+ f"(1, {max_seat})"}
+ )
+
+ def save(self, *args, **kwargs) -> None:
+ self.full_clean()
+ super().save(*args, **kwargs)
+
+ class Meta:
+ constraints = [
+ models.UniqueConstraint(
+ fields=["row", "seat", "movie_session"],
+ name="unique_order_row_seat_order_session"
+ )
+ ]
+
+ def __str__(self) -> str :
+ return (
+ f"{self.movie_session.movie.title} {self.movie_session.show_time} "
+ f"(row: {self.row}, "
+ f"seat: {self.seat})"
+ )
+
+
+class User(AbstractUser):
+ REQUIRED_FIELDS = []

services/movie.py

@@ -1,11 +1,12 @@
 from django.db.models import QuerySet
-
+from django.db import transaction
 from db.models import Movie
 
 
 def get_movies(
  genres_ids: list[int] = None,
  actors_ids: list[int] = None,
+ title: str = None,
 ) -> QuerySet:
  queryset = Movie.objects.all()
 
@@ -15,13 +16,17 @@ 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,

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:
+ taken_seats = []
+ tickets = Ticket.objects.filter(movie_session_id=movie_session_id)
+ for ticket in tickets:
+ taken_seats.append({"row": ticket.row, "seat": ticket.seat})
+ return taken_seats

services/order.py

@@ -0,0 +1,40 @@
+import datetime
+
+from django.db.models import QuerySet
+
+from db.models import Order
+from django.db import transaction
+from typing import List
+from db.models import Ticket
+from db.models import User
+
+
+@transaction.atomic
+def create_order(
+ tickets: List[dict],
+ username: str,
+ date: datetime = None
+) -> None:
+ user = User.objects.get(username=username)
+ order = Order.objects.create(user=user)
+ if date is not None:
+ order.created_at = date
+ order.save()
+ for ticket_data in tickets:
+ row = ticket_data.get("row")
+ seat = ticket_data.get("seat")
+ ticket, created = Ticket.objects.get_or_create(
+ row=row,
+ seat=seat,
+ movie_session_id=ticket_data["movie_session"],
+ order=order
+ )
+ ticket.order = order
+ ticket.save()
+
+
+def get_orders(username: str = None) -> QuerySet[Order]:
+ orders = Order.objects.all()
+ if username is not None:
+ orders = orders.filter(user__username=username)
+ return orders.order_by("-created_at")

services/user.py

@@ -0,0 +1,56 @@
+from django.contrib.auth.hashers import make_password
+from db.models import User
+from django.shortcuts import get_object_or_404
+
+
+def create_user(
+ username: str,
+ password: str,
+ user_id: int = None,
+ email: str = None,
+ first_name: str = None,
+ last_name: str = None
+) -> None:
+ user, created = User.objects.update_or_create(
+ id=user_id,
+ defaults=(
+ {"username": username,
+ "password": make_password(password)}
+ )
+ )
+ if email is not None:
+ user.email = email
+ if first_name is not None:
+ user.first_name = first_name
+ if last_name is not None:
+ user.last_name = last_name
+ user.save()
+ return user
+
+
+def get_user(user_id: int) -> int:
+ return User.objects.get(id=user_id)
+
+
+def update_user(
+ user_id: int,
+ username: str = None,
+ password: str = None,
+ email: str = None,
+ first_name: str = None,
+ last_name: str = None
+) -> None:
+ user = get_object_or_404(User, id=user_id)
+ if username is not None:
+ user.username = username
+ if password is not None:
+ user.set_password(password)
+ if email is not None:
+ user.email = email
+ if first_name is not None:
+ user.first_name = first_name
+ if last_name is not None:
+ user.last_name = last_name
+
+ user.save()
+ return user

settings.py

@@ -24,5 +24,9 @@
 USE_TZ = False
 
 INSTALLED_APPS = [
+ "django.contrib.auth",
+ "django.contrib.contenttypes",
  "db",
 ]
+
+AUTH_USER_MODEL = "db.User"