Froud

bot/utils/froud_check.py

@@ -0,0 +1,54 @@
+from asgiref.sync import sync_to_async
+from django.db import connections
+from datetime import datetime, timedelta
+from hotels.models import Reservation, Hotel, FraudLog
+
+
+async def detect_fraud(hotel_id, period="day"):
+    """
+    Сравнивает данные из PMS и QR базы для обнаружения FRAUD-действий.
+    :param hotel_id: ID отеля для анализа.
+    :param period: Период времени для анализа (day, week, month).
+    """
+    now = datetime.now()
+    if period == "day":
+        start_date = now - timedelta(days=1)
+    elif period == "week":
+        start_date = now - timedelta(weeks=1)
+    elif period == "month":
+        start_date = now - timedelta(days=30)
+    else:
+        start_date = None
+
+    end_date = now
+
+    # Данные из PMS
+    reservations = await sync_to_async(list)(
+        Reservation.objects.filter(
+            hotel_id=hotel_id,
+            check_in__date__range=(start_date, end_date)
+        )
+    )
+
+    # Данные из QR-системы
+    qr_checkins = []
+    with connections['wordpress'].cursor() as cursor:
+        cursor.execute(
+            "SELECT reservation_id, guest_name, check_in_date FROM qr_checkins WHERE check_in_date BETWEEN %s AND %s",
+            [start_date, end_date]
+        )
+        qr_checkins = cursor.fetchall()
+
+    qr_checkins_set = {row[0] for row in qr_checkins}  # Множество reservation_id
+
+    # Сравнение данных
+    for res in reservations:
+        if res.id not in qr_checkins_set:
+            # Записываем FRAUD
+            await sync_to_async(FraudLog.objects.create)(
+                hotel_id=hotel_id,
+                reservation_id=res.id,
+                guest_name=res.guest_name,
+                check_in_date=res.check_in.date(),
+                message="Проверка на QR-систему провалилась."
+            )