antifroud_check

This commit is contained in:
2024-12-17 21:39:33 +09:00
parent 66750015e2
commit 0bf2bb8dff
17 changed files with 1298 additions and 524 deletions

View File

@@ -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 = "Журналы мошенничества"