geoip intgration

antifroud/check_fraud.py

@@ -33,31 +33,33 @@ class ReservationChecker:
         logger.error(message)
 
     def fetch_user_logs(self):
-        """ Извлекает записи из UserActivityLog за период. """
-        user_logs = UserActivityLog.objects.filter(created__range=(self.start_time, self.end_time))
-        self.log_info(f"Найдено {user_logs.count()} логов активности для анализа.")
-        return user_logs
+        try:
+            self.log_info("Начинается извлечение логов активности пользователей.")
+            user_logs = UserActivityLog.objects.filter(created__range=(self.start_time, self.end_time))
+            self.log_info(f"Найдено {user_logs.count()} логов активности для анализа.")
+            return user_logs
+        except Exception as e:
+            self.log_error(f"Ошибка при извлечении логов активности: {e}")
+            return UserActivityLog.objects.none()
 
     def fetch_reservations(self):
-        """ Извлекает бронирования за период. """
-        reservations = Reservation.objects.filter(
-            Q(check_in__lte=self.end_time) & Q(check_out__gte=self.start_time)
-        )
-        self.log_info(f"Найдено {reservations.count()} бронирований для анализа.")
-        return reservations
+        try:
+            self.log_info("Начинается извлечение бронирований.")
+            reservations = Reservation.objects.filter(
+                Q(check_in__lte=self.end_time) & Q(check_out__gte=self.start_time)
+            )
+            self.log_info(f"Найдено {reservations.count()} бронирований для анализа.")
+            return reservations
+        except Exception as e:
+            self.log_error(f"Ошибка при извлечении бронирований: {e}")
+            return Reservation.objects.none()
 
     def find_violations(self):
-        """
-        Сравнивает записи бронирований и логи активности для выявления нарушений:
-        - Сканирование QR без бронирования.
-        - Бронь со статусом "заселен" без сканирования QR.
-        - Раннее заселение.
-        """
+        self.log_info("Начинается анализ несоответствий.")
         user_logs = self.fetch_user_logs()
         reservations = self.fetch_reservations()
 
-        # Сопоставляем записи
-        log_lookup = {}  # Словарь: (hotel_id, room_number) -> список логов
+        log_lookup = {}
         for log in user_logs:
             params = parse_qs(log.url_parameters or "")
             hotel_id = params.get("utm_content", [None])[0]
@@ -70,7 +72,6 @@ class ReservationChecker:
             key = (reservation.hotel.hotel_id, reservation.room_number)
             logs = log_lookup.get(key, [])
 
-            # Бронь со статусом "заселен" без сканирования QR
             if reservation.status == "заселен" and not logs:
                 self.record_violation(
                     hotel=reservation.hotel,
@@ -80,7 +81,6 @@ class ReservationChecker:
                             f"не имеет записи сканирования QR."
                 )
 
-            # Раннее заселение
             for log in logs:
                 if log.created < reservation.check_in:
                     self.record_violation(
@@ -91,7 +91,6 @@ class ReservationChecker:
                                 f"{reservation.check_in} для номера {reservation.room_number}."
                     )
 
-        # Сканирование QR без бронирования
         for (hotel_id, room_number), logs in log_lookup.items():
             matching_reservations = reservations.filter(
                 hotel__hotel_id=hotel_id,
@@ -108,9 +107,6 @@ class ReservationChecker:
                     )
 
     def record_violation(self, hotel, room_number, violation_type, details):
-        """
-        Записывает нарушение в список для последующего сохранения.
-        """
         if hotel:
             self.violations.append(ViolationLog(
                 hotel=hotel,
@@ -121,7 +117,6 @@ class ReservationChecker:
             self.log_warning(f"Зафиксировано нарушение: {details}")
 
     def save_violations(self):
-        """ Сохраняет найденные нарушения в базу данных. """
         if self.violations:
             ViolationLog.objects.bulk_create(self.violations)
             self.log_info(f"Создано {len(self.violations)} записей в ViolationLog.")
@@ -129,7 +124,6 @@ class ReservationChecker:
             self.log_info("Нарушений не обнаружено.")
 
     def run_check(self):
-        """ Основной метод для запуска проверки. """
         self.log_info(f"Запуск проверки с {self.start_time} по {self.end_time}.")
         try:
             self.find_violations()
@@ -140,8 +134,10 @@ class ReservationChecker:
 
 # Функция для запуска из планировщика
 def run_reservation_check():
-    checker = ReservationChecker()
-    checker.run_check()
-
-
-
+    logger.info("Планировщик вызывает run_reservation_check.")
+    try:
+        checker = ReservationChecker()
+        checker.run_check()
+    except Exception as e:
+        logger.error(f"Ошибка при запуске проверки: {e}")
+    logger.info("run_reservation_check завершена.")