RealtyCalendar plugin

pms_integration/plugins/ecvi_pms.py

@@ -74,7 +74,7 @@ class EcviPMSPlugin(BasePMSPlugin):
         except requests.exceptions.RequestException as e:
             self.logger.error(f"Ошибка запроса: {e}")
             return []
-        self.logger.debug(f"\n\n\n\n\ndata: {data}\n\n\n\n\n")
+
         # Фильтрация данных
         filtered_data = [
             {
@@ -87,7 +87,6 @@ class EcviPMSPlugin(BasePMSPlugin):
             } for item in data if isinstance(item, dict) and item.get('occupancy') in ['проживание', 'под выезд', 'под заезд']
         ]
         
-        
         self.logger.debug(f"filtered_data: {filtered_data}")
         
         # Сохранение данных в базу данных
@@ -108,7 +107,6 @@ class EcviPMSPlugin(BasePMSPlugin):
 
             # Проверяем, существует ли уже резервация с таким внешним ID
             reservation_id = item.get('reservation_id')
-            self.logger.debug(f"-----\n\n\nITEM : {item}\n\n\n---")
             if not reservation_id:
                 self.logger.error("Ошибка: 'reservation_id' отсутствует в данных.")
                 return
@@ -123,6 +121,7 @@ class EcviPMSPlugin(BasePMSPlugin):
                 await sync_to_async(Reservation.objects.update_or_create)(
                     reservation_id=reservation_id,
                     defaults={
+                        ''
                         'room_number': item.get('room_number'),
                         'room_type': item.get('room_type'),
                         'check_in': item.get('checkin'),

pms_integration/plugins/realtycalendar_pms.py

@@ -4,17 +4,16 @@ import json
 from .base_plugin import BasePMSPlugin
 from datetime import datetime, timedelta
 from asgiref.sync import sync_to_async
-from math import ceil
-
+from touchh.utils.log import CustomLogger
+from hotels.models import Hotel, Reservation
 class RealtyCalendarPlugin(BasePMSPlugin):
-    """Плагин для импорта данных из системы RealtyCalendar
-    """
+    """Плагин для импорта данных из системы RealtyCalendar."""
     def __init__(self, config):
         super().__init__(config)
         self.public_key = config.public_key
         self.private_key = config.private_key
         self.api_url = config.url.rstrip("/")
-
+        self.logger = CustomLogger(name="RealtyCalendarPlugin", log_level="DEBUG").get_logger()
         if not self.public_key or not self.private_key:
             raise ValueError("Публичный или приватный ключ отсутствует для RealtyCalendar")
 
@@ -31,8 +30,8 @@ class RealtyCalendarPlugin(BasePMSPlugin):
         """
         Возвращает отсортированный по имени список ключей.
         """
-        sorted_keys = sorted(list(obj.keys()))
-        print(f"[DEBUG] Отсортированные ключи: {sorted_keys}")
+        sorted_keys = sorted(obj.keys())
+        self.logger.debug(f"Отсортированные ключи: {sorted_keys}")
         return sorted_keys
 
     def _generate_data_string(self, obj):
@@ -41,7 +40,7 @@ class RealtyCalendarPlugin(BasePMSPlugin):
         """
         sorted_keys = self._get_sorted_keys(obj)
         string = "".join(f"{key}={obj[key]}" for key in sorted_keys)
-        print(f"[DEBUG] Сформированная строка данных: {string}")
+        self.logger.debug(f"Сформированная строка данных: {string}")
         return string + self.private_key
 
     def _generate_md5(self, string):
@@ -49,7 +48,7 @@ class RealtyCalendarPlugin(BasePMSPlugin):
         Генерирует MD5-хеш от строки.
         """
         md5_hash = hashlib.md5(string.encode("utf-8")).hexdigest()
-        print(f"[DEBUG] Сформированный MD5-хеш: {md5_hash}")
+        self.logger.debug(f"Сформированный MD5-хеш: {md5_hash}")
         return md5_hash
 
     def _generate_sign(self, data):
@@ -57,16 +56,17 @@ class RealtyCalendarPlugin(BasePMSPlugin):
         Генерирует подпись для данных запроса.
         """
         data_string = self._generate_data_string(data)
-        print(f"[DEBUG] Строка для подписи: {data_string}")
+        self.logger.debug(f"Строка для подписи: {data_string}")
         sign = self._generate_md5(data_string)
-        print(f"[DEBUG] Подпись: {sign}")
+        self.logger.debug(f"Подпись: {sign}")
         return sign
 
-    def _fetch_data(self):
+    async def _fetch_data(self):
         """
         Выполняет запрос к API RealtyCalendar для получения данных о бронированиях.
         """
-        base_url = f"https://realtycalendar.ru/api/v1/bookings/{self.public_key}/"
+        self.logger.debug("Начало выполнения функции _fetch_data")
+        base_url = f"{self.api_url}/api/v1/bookings/{self.public_key}/"
         headers = {
             "Accept": "application/json",
             "Content-Type": "application/json",
@@ -77,81 +77,150 @@ class RealtyCalendarPlugin(BasePMSPlugin):
         data = {
             "begin_date": (now - timedelta(days=7)).strftime("%Y-%m-%d"),
             "end_date": now.strftime("%Y-%m-%d"),
-            "room_number": ""
         }
 
-        print(f"[DEBUG] Даты выборки: {data}")
+        self.logger.debug(f"Даты выборки: {data}")
 
         # Генерация подписи
         data["sign"] = self._generate_sign(data)
 
         # Отправляем запрос
-        print(f"[DEBUG] URL запроса: {base_url}")
-        print(f"[DEBUG] Заголовки: {headers}")
-        print(f"[DEBUG] Данные запроса: {data}")
+        self.logger.debug(f"URL запроса: {base_url}")
+        self.logger.debug(f"Заголовки: {headers}")
+        self.logger.debug(f"Данные запроса: {data}")
 
         response = requests.post(url=base_url, headers=headers, json=data)
+        self.logger.debug(f"Запрос: {response}")
+        self.logger.debug(f"Статус ответа: {response.status_code}")
+        # self.logger.debug(f"Ответ: {response.text}")
 
-        # Логируем результат
-        print(f"[DEBUG] Статус ответа: {response.status_code}")
-        print(f"[DEBUG] Ответ: {response.text}")
-
-        # Проверяем успешность запроса
         if response.status_code == 200:
-            bookings = response.json().get("bookings", [])
-            print(f"[DEBUG] Полученные данные бронирований: {bookings}")
-            return bookings
-        else:
-            raise ValueError(f"Ошибка API RealtyCalendar: {response.status_code}, {response.text}")
+            try:
+                response_data = response.json()
+                self.logger.debug(f"Тип данных ответа: {type(response_data)}")
+                if not isinstance(response_data, dict) or "bookings" not in response_data:
+                    raise ValueError(f"Неожиданная структура ответа: {response_data}")
+                bookings = response_data.get("bookings", [])
+                # self.logger.debug(f"Полученные данные бронирований: {bookings}")
+            except json.JSONDecodeError as e:
+                self.logger.error(f"Ошибка декодирования JSON: {e}")
+                raise ValueError("Ошибка декодирования JSON ответа.")
 
+            # Фильтрация данных
+            filtered_data = [
+                {
+                    "id": item.get("id"),
+                    "begin_date": item.get("begin_date"),
+                    "end_date": item.get("end_date"),
+                    "amount": item.get("amount"),
+                    "status": item.get("status"),
+                    "is_delete": item.get("is_delete"),
+                    "apartment_id": item.get("apartment_id"),
+                    "prepayment": item.get("prepayment"),
+                    "deposit": item.get("deposit"),
+                    "source": item.get("source"),
+                    "notes": item.get("notes"),
+                }
+                for item in bookings
+                if isinstance(item, dict) and item.get("status") in ["booked", "request"]
+            ]
+            self.logger.debug(f"Отфильтрованные данные: {type(filtered_data)}")
 
-    async def _save_to_db(self, data, hotel_id, batch_size=50):
+            for item in filtered_data:
+                self.logger.debug(f"Данные бронирования: {item}")
+                await self._save_to_db(item)
+                
+        from django.utils import timezone
+
+    async def _save_to_db(self, data):
+        from django.utils import timezone
         """
-        Сохраняет данные о бронированиях в базу данных партиями.
+        Сохраняет данные в БД (например, информацию о номере).
         """
-        from hotels.models import Reservation, Hotel
-
         try:
-            hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
-            self.logger.info(f"Загружен отель: {hotel.name}")
+            # Проверяем общее количество записей для обработки
+            if not isinstance(data, list):
+                self.logger.error(f"Ожидался список записей, но получен {type(data).__name__}")
+                return
 
-            # Разделение данных на батчи
             total_records = len(data)
-            batches = [data[i:i + batch_size] for i in range(0, total_records, batch_size)]
-            self.logger.info(f"Обработка {total_records} записей в {len(batches)} партиях...")
+            self.logger.info(f"Общее количество записей для обработки: {total_records}")
 
-            for batch_index, batch in enumerate(batches):
-                self.logger.info(f"Обработка партии {batch_index + 1}/{len(batches)}")
+            for index, item in enumerate(data, start=1):
+                try:
+                    self.logger.info(f"Обработка записи {index}/{total_records}")
 
-                for item in batch:
+                    # Проверка типа данных
+                    if not isinstance(item, dict):
+                        self.logger.error(f"Пропущена запись {index}/{total_records}: ожидался dict, но получен {type(item).__name__}")
+                        continue
+
+                    # Получаем отель по настройкам PMS
+                    hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config)
+                    self.logger.debug(f"Отель найден: {hotel.name}")
+
+                    # Проверяем, существует ли уже резервация с таким внешним ID
+                    reservation_id = item.get('id')
+                    if not reservation_id:
+                        self.logger.error(f"Пропущена запись {index}/{total_records}: отсутствует 'id' в данных.")
+                        continue
+
+                    # Преобразуем даты в "aware" объекты
                     try:
-                        if item.get("is_delete", False):
-                            self.logger.info(f"Пропущена запись с ID {item.get('id')} (удалена).")
-                            continue
-
-                        client_data = item.get("client", {})
-                        if not item.get("id") or not item.get("begin_date") or not item.get("end_date"):
-                            self.logger.warning(f"Пропущена запись с неполными данными: {item}")
-                            continue
-
-                        reservation_defaults = {
-                            "room_number": item.get("apartment_id", ""),
-                            "check_in": datetime.strptime(item["begin_date"], "%Y-%m-%d"),
-                            "check_out": datetime.strptime(item["end_date"], "%Y-%m-%d"),
-                            "status": item.get("status", ""),
-                            "price": item.get("amount", 0),
-                            "client_name": client_data.get("fio", ""),
-                            "client_email": client_data.get("email", ""),
-                            "client_phone": client_data.get("phone", ""),
-                        }
-
-                        await sync_to_async(Reservation.objects.update_or_create)(
-                            reservation_id=item["id"],
-                            hotel=hotel,
-                            defaults=reservation_defaults
-                        )
-                        self.logger.info(f"Сохранена запись для бронирования ID {item['id']}.")
+                        check_in = timezone.make_aware(datetime.strptime(item.get('begin_date'), "%Y-%m-%d"))
+                        check_out = timezone.make_aware(datetime.strptime(item.get('end_date'), "%Y-%m-%d"))
                     except Exception as e:
-                        self.logger.error(f"Ошибка при обработке бронирования ID {item.get('id', 'неизвестно')}: {e}")
+                        self.logger.error(f"Ошибка преобразования дат для записи {index}/{total_records}: {e}")
+                        continue
+
+                    existing_reservation = await sync_to_async(Reservation.objects.filter)(reservation_id=reservation_id)
+
+                    # Теперь вызываем .first() после асинхронного вызова
+                    existing_reservation = await sync_to_async(existing_reservation.first)()
+
+                    if existing_reservation:
+                        self.logger.debug(f"Резервация {reservation_id} уже существует. Обновляем...")
+                        await sync_to_async(Reservation.objects.update_or_create)(
+                            reservation_id=reservation_id,
+                            defaults={
+                                'room_number': item.get('apartment_id'),
+                                'room_type': 'Описание отсутствует',
+                                'check_in': check_in,
+                                'check_out': check_out,
+                                'status': item.get('status'),
+                                'hotel': hotel
+                            }
+                        )
+                        self.logger.debug(f"Резервация обновлена.")
+                    else:
+                        self.logger.debug(f"Резервация не найдена, создаем новую...")
+                        reservation = await sync_to_async(Reservation.objects.create)(
+                            reservation_id=reservation_id,
+                            room_number=item.get('apartment_id'),
+                            room_type='Описание отсутствует',
+                            check_in=check_in,
+                            check_out=check_out,
+                            status=item.get('status'),
+                            hotel=hotel
+                        )
+                        self.logger.debug(f"Новая резервация создана с ID: {reservation.reservation_id}")
+
+                except Exception as e:
+                    self.logger.error(f"Ошибка при обработке записи {index}/{total_records}: {e}")
+
         except Exception as e:
-            self.logger.error(f"Ошибка при обработке данных: {e}")
+            self.logger.error(f"Ошибка обработки данных в _save_to_db: {e}")
+
+
+    def validate_plugin(self):
+        """
+        Проверка на соответствие требованиям.
+        Можно проверить наличие методов или полей.
+        """
+        # Проверяем наличие обязательных методов
+        required_methods = ["fetch_data", "get_default_parser_settings", "_fetch_data"]
+        for m in required_methods:
+            if not hasattr(self, m):
+                raise ValueError(f"Плагин {type(self).__name__} не реализует метод {m}.")
+        self.logger.debug(f"Плагин {self.__class__.__name__} прошел валидацию.")
+        return True