antifroud_check

hotels/admin.py

@@ -5,7 +5,7 @@ from .models import (
     UserHotel,
     APIConfiguration,
     Reservation,
-    FraudLog
+    Room
 )
 from django.urls import path
 from django.shortcuts import redirect
@@ -27,8 +27,9 @@ class HotelForm(forms.ModelForm):
 
 @admin.register(Hotel)
 class HotelAdmin(admin.ModelAdmin):
-    list_display = ['name', 'hotel_id', 'pms', 'timezone', 'description']
+    list_display = ['name', 'hotel_id','room_count', 'pms', 'timezone', 'description']
     list_filter = ['name', 'pms', 'timezone']
+    list_sorting = ['name', 'pms', 'room_count', 'timezone']
     def sync_button(self, obj):
         return format_html(
             '<a class="button" href="{}">Синхронизировать</a>',
@@ -41,8 +42,36 @@ class HotelAdmin(admin.ModelAdmin):
             path('sync/<int:hotel_id>/', self.sync_hotel_data),
         ]
         return custom_urls + urls
+    def room_count(self, obj):
+        """
+        Подсчитывает количество комнат, связанных с отелем.
+        """
+        return Room.objects.filter(hotel=obj).count()
+    room_count.short_description = "Количество комнат"
+
+    def sync_button(self, obj):
+        """
+        Кнопка синхронизации данных.
+        """
+        return format_html(
+            '<a class="button" href="{}">Синхронизировать</a>',
+            f"/admin/hotels/sync/{obj.id}/"
+        )
+
+    def get_urls(self):
+        """
+        Добавление кастомного URL для синхронизации.
+        """
+        urls = super().get_urls()
+        custom_urls = [
+            path('sync/<int:hotel_id>/', self.sync_hotel_data),
+        ]
+        return custom_urls + urls
 
     def sync_hotel_data(self, request, hotel_id):
+        """
+        Метод синхронизации данных отеля.
+        """
         try:
             hotel = Hotel.objects.get(id=hotel_id)
             client = APIClient(hotel.pms)
@@ -66,4 +95,9 @@ class ReservationAdmin(admin.ModelAdmin):
     list_filter = ('hotel', 'room_number', 'room_type', 'check_in', 'check_out', 'status', 'price', 'discount')
     ordering = ('-check_in',)
 
-
+@admin.register(Room)
+class RoomAdmin(admin.ModelAdmin):
+    list_display = ('hotel', 'number', 'external_id', 'description', 'created_at', 'updated_at')
+    search_fields = ('hotel', 'number', 'external_id', 'description')
+    list_filter = ('hotel', 'number', 'external_id','description', 'created_at', 'updated_at')
+    ordering = ('-hotel', '-number')

hotels/migrations/0012_alter_room_number.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.4 on 2024-12-17 11:21
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('hotels', '0011_room_alter_fraudlog_check_in_date_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='room',
+            name='number',
+            field=models.CharField(max_length=50, unique=True, verbose_name='Номер комнаты'),
+        ),
+    ]

hotels/migrations/0013_alter_room_number.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.4 on 2024-12-17 11:21
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('hotels', '0012_alter_room_number'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='room',
+            name='number',
+            field=models.CharField(max_length=50, verbose_name='Номер комнаты'),
+        ),
+    ]

hotels/migrations/0014_alter_room_number.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.4 on 2024-12-17 11:24
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('hotels', '0013_alter_room_number'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='room',
+            name='number',
+            field=models.CharField(max_length=50, unique=True, verbose_name='Номер комнаты'),
+        ),
+    ]

hotels/models.py

@@ -1,5 +1,7 @@
 from django.db import models
-from django.core.validators import MinValueValidator, MaxValueValidator
+from django.core.validators import MinValueValidator, MaxValueValidator, RegexValidator
+from django.core.exceptions import ValidationError
+import pytz
 
 
 class APIConfiguration(models.Model):
@@ -17,25 +19,33 @@ class APIConfiguration(models.Model):
         verbose_name = "Конфигурация API"
         verbose_name_plural = "Конфигурации API"
 
-import pytz
 
 class Hotel(models.Model):
     name = models.CharField(max_length=255, verbose_name="Название отеля")
     hotel_id = models.CharField(max_length=255, unique=True, null=True, blank=True, verbose_name="ID отеля")
     created_at = models.DateTimeField(auto_now_add=True, verbose_name="Создан")
-    phone= models.CharField(max_length=50, null=True, blank=True, verbose_name="Телефон")
+    phone = models.CharField(
+        max_length=50,
+        null=True,
+        blank=True,
+        verbose_name="Телефон",
+        validators=[
+            RegexValidator(regex=r'^\+?1?\d{9,15}$', message="Введите корректный номер телефона (до 15 цифр).")
+        ],
+    )
     email = models.EmailField(null=True, blank=True, verbose_name="Email")
     address = models.CharField(max_length=255, null=True, blank=True, verbose_name="Адрес")
     city = models.CharField(max_length=255, null=True, blank=True, verbose_name="Город")
     timezone = models.CharField(
         max_length=63,
-        choices=[(tz, tz) for tz in pytz.all_timezones],  # Список всех часовых поясов
-        default='UTC',  # Значение по умолчанию
+        choices=[(tz, tz) for tz in pytz.all_timezones],
+        default='UTC',
+        verbose_name="Часовой пояс",
     )
     description = models.TextField(null=True, blank=True, verbose_name="Описание")
-    
+
     pms = models.ForeignKey(
-        'pms_integration.PMSConfiguration',  
+        'pms_integration.PMSConfiguration',
         on_delete=models.SET_NULL,
         null=True,
         blank=True,
@@ -48,13 +58,11 @@ class Hotel(models.Model):
     class Meta:
         verbose_name = "Отель"
         verbose_name_plural = "Отели"
-        
+
+
 class Room(models.Model):
-    """
-    Модель номера отеля.
-    """
     hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, related_name="rooms", verbose_name="Отель")
-    number = models.CharField(max_length=50, verbose_name="Номер комнаты")
+    number = models.CharField(max_length=50, unique=True, verbose_name="Номер комнаты")
     external_id = models.CharField(max_length=255, unique=True, verbose_name="Внешний ID комнаты")
     description = models.TextField(blank=True, null=True, verbose_name="Описание")
     created_at = models.DateTimeField(auto_now_add=True, verbose_name="Дата создания")
@@ -66,17 +74,23 @@ class Room(models.Model):
     class Meta:
         verbose_name = "Номер"
         verbose_name_plural = "Номера"
-        unique_together = ("hotel", "number")  # Уникальность пары (отель, номер)
+        constraints = [
+            models.UniqueConstraint(fields=["hotel", "number"], name="unique_hotel_room")
+        ]
+        indexes = [
+            models.Index(fields=["hotel", "number"]),
+        ]
+
 
 class UserHotel(models.Model):
     user = models.ForeignKey(
-        'users.User',  # Используем строковую ссылку
+        'users.User',
         on_delete=models.CASCADE,
         related_name="user_hotels",
         verbose_name="Пользователь"
     )
     hotel = models.ForeignKey(
-        'hotels.Hotel',
+        Hotel,
         on_delete=models.CASCADE,
         related_name="hotel_users",
         verbose_name="Отель"
@@ -90,6 +104,76 @@ class UserHotel(models.Model):
         verbose_name_plural = "Пользователи отелей"
 
 
+class Reservation(models.Model):
+    id = models.BigAutoField(primary_key=True, auto_created=True, verbose_name="ID")
+    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, verbose_name="Отель")
+    reservation_id = models.BigIntegerField(unique=True, verbose_name="ID бронирования")
+    room_number = models.CharField(max_length=255, null=True, blank=True, verbose_name="Номер комнаты")
+    room_type = models.CharField(max_length=255, verbose_name="Тип комнаты")
+    check_in = models.DateTimeField(verbose_name="Дата заезда")
+    check_out = models.DateTimeField(verbose_name="Дата выезда")
+    status = models.CharField(max_length=50, verbose_name="Статус")
+    price = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True, verbose_name="Цена")
+    discount = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True, verbose_name="Скидка")
+
+    def clean(self):
+        if self.check_out and self.check_in and self.check_out <= self.check_in:
+            raise ValidationError("Дата выезда должна быть позже даты заезда.")
+
+    def __str__(self):
+        return f"Бронирование {self.reservation_id} - {self.hotel.name}"
+
+    class Meta:
+        verbose_name = "Бронирование"
+        verbose_name_plural = "Бронирования"
+        indexes = [
+            models.Index(fields=["hotel", "check_in", "check_out"]),
+        ]
+
+
+class Guest(models.Model):
+    reservation = models.ForeignKey(Reservation, on_delete=models.CASCADE, related_name="guests", verbose_name="Бронирование")
+    name = models.CharField(max_length=255, verbose_name="Имя гостя")
+    birthdate = models.DateField(null=True, blank=True, verbose_name="Дата рождения")
+    phone = models.CharField(
+        max_length=50,
+        null=True,
+        blank=True,
+        verbose_name="Телефон",
+        validators=[
+            RegexValidator(regex=r'^\+?1?\d{9,15}$', message="Введите корректный номер телефона (до 15 цифр).")
+        ],
+    )
+    email = models.EmailField(null=True, blank=True, verbose_name="Email")
+
+    def __str__(self):
+        return f"{self.name} ({self.birthdate})" if self.birthdate else self.name
+
+    class Meta:
+        verbose_name = "Гость"
+        verbose_name_plural = "Гости"
+
+
+class FraudLog(models.Model):
+    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, related_name="frauds")
+    reservation_id = models.BigIntegerField(verbose_name="ID бронирования")
+    guest_name = models.CharField(max_length=255, null=True, blank=True)
+    check_in_date = models.DateField(verbose_name="Дата заезда")
+    detected_at = models.DateTimeField(auto_now_add=True, verbose_name="Дата обнаружения")
+    message = models.TextField(verbose_name="Сообщение")
+
+    def __str__(self):
+        return f"FRAUD: {self.guest_name} ({self.check_in_date})"
+
+    class Meta:
+        verbose_name = "Журнал мошенничества"
+        verbose_name_plural = "Журналы мошенничества"
+        indexes = [
+            models.Index(fields=["reservation_id"]),
+            models.Index(fields=["detected_at"]),
+        ]
+
+
 class APIRequestLog(models.Model):
     api = models.ForeignKey(APIConfiguration, on_delete=models.CASCADE, verbose_name="API")
     request_time = models.DateTimeField(auto_now_add=True, verbose_name="Время запроса")
@@ -103,57 +187,6 @@ class APIRequestLog(models.Model):
         verbose_name = "Журнал запросов API"
         verbose_name_plural = "Журналы запросов API"
         indexes = [
-            models.Index(fields=['api']),
-            models.Index(fields=['request_time']),
+            models.Index(fields=["api"]),
+            models.Index(fields=["request_time"]),
         ]
-
-
-class Reservation(models.Model):
-    id = models.BigAutoField(primary_key=True, auto_created=True, verbose_name="ID")
-    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, verbose_name="Отель")
-    reservation_id = models.BigIntegerField(unique=True, verbose_name="ID бронирования")
-    room_number = models.CharField(max_length=255, null=True, blank=True)
-    room_type = models.CharField(max_length=255, verbose_name="Тип комнаты")
-    check_in = models.DateTimeField(verbose_name="Дата заезда")
-    check_out = models.DateTimeField(verbose_name="Дата выезда")
-    status = models.CharField(max_length=50, verbose_name="Статус")
-    price = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True, verbose_name="Цена")
-    discount = models.DecimalField(max_digits=10, decimal_places=2, null=True, blank=True, verbose_name="Скидка")
-
-    def __str__(self):
-        return f"Бронирование {self.reservation_id} - {self.hotel.name}"
-
-    class Meta:
-        verbose_name = "Бронирование"
-        verbose_name_plural = "Бронирования"
-
-
-class Guest(models.Model):
-    reservation = models.ForeignKey(Reservation, on_delete=models.CASCADE, related_name="guests", verbose_name="Бронирование")
-    name = models.CharField(max_length=255, verbose_name="Имя гостя")
-    birthdate = models.DateField(null=True, blank=True, verbose_name="Дата рождения")
-    phone = models.CharField(max_length=50, null=True, blank=True, verbose_name="Телефон")
-    email = models.EmailField(null=True, blank=True, verbose_name="Email")
-
-    def __str__(self):
-        return self.name
-
-    class Meta:
-        verbose_name = "Гость"
-        verbose_name_plural = "Гости"
-
-
-class FraudLog(models.Model):
-    hotel = models.ForeignKey(Hotel, on_delete=models.CASCADE, related_name="frauds")
-    reservation_id = models.BigIntegerField(unique=True, verbose_name="ID бронирования")
-    guest_name = models.CharField(max_length=255, null=True, blank=True)
-    check_in_date = models.DateField()
-    detected_at = models.DateTimeField(auto_now_add=True)
-    message = models.TextField()
-
-    def __str__(self):
-        return f"FRAUD: {self.guest_name} ({self.check_in_date})"
-
-    class Meta:
-        verbose_name = "Журнал мошенничества"
-        verbose_name_plural = "Журналы мошенничества"