mergse

2024-12-28 09:44:58 +09:00
parent 5e3ed91b3a
commit 7350989113
11 changed files with 14160 additions and 324 deletions
--- a/pms_integration/plugins/shelter_pms.py
+++ b/pms_integration/plugins/shelter_pms.py
@@ -1,3 +1,130 @@
+# import logging
+# import requests
+# from datetime import datetime, timedelta
+# from asgiref.sync import sync_to_async
+# from pms_integration.models import PMSConfiguration
+# from hotels.models import Hotel, Reservation
+# from .base_plugin import BasePMSPlugin
+
+
+# class ShelterPMSPlugin(BasePMSPlugin):
+#     """
+#     Плагин для интеграции с PMS Shelter.
+#     """
+
+#     def __init__(self, pms_config):
+#         super().__init__(pms_config)
+
+#         # Настройка логирования
+#         self.logger = logging.getLogger(self.__class__.__name__)
+#         handler = logging.StreamHandler()
+#         formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
+#         handler.setFormatter(formatter)
+#         self.logger.addHandler(handler)
+#         self.logger.setLevel(logging.DEBUG)
+
+#         # Инициализация параметров API
+#         self.api_url = pms_config.url
+#         self.token = pms_config.token
+#         self.username = pms_config.username
+#         self.password = pms_config.password
+
+#     def get_default_parser_settings(self):
+#         """
+#         Возвращает настройки парсера по умолчанию.
+#         """
+#         return {
+#             "field_mapping": {
+#                 "check_in": "checkin",
+#                 "check_out": "checkout",
+#                 "room_number": "room_number",
+#                 "room_type": "room_type",
+#                 "status": "status",
+#             },
+#             "date_format": "%Y-%m-%dT%H:%M:%S"
+#         }
+
+#     async def _fetch_data(self):
+#         """
+#         Получает данные из Shelter PMS API и сохраняет их в базу данных.
+#         """
+#         now = datetime.now()
+#         start_date = (now - timedelta(days=1)).strftime('%Y-%m-%d')
+#         end_date = now.strftime('%Y-%m-%d')
+
+#         headers = {
+#             "Authorization": f"Bearer {self.token}",
+#             "Content-Type": "application/json",
+#         }
+#         params = {
+#             "start_date": start_date,
+#             "end_date": end_date,
+#         }
+
+#         try:
+#             response = await sync_to_async(requests.get)(f"{self.api_url}/reservations", headers=headers, params=params)
+#             response.raise_for_status()
+#             data = response.json()
+#             self.logger.debug(f"Получены данные с API: {data}")
+#         except requests.exceptions.RequestException as e:
+#             self.logger.error(f"Ошибка запроса к API Shelter: {e}")
+#             return []
+
+#         # Обработка и сохранение данных
+#         processed_data = []
+#         for item in data.get("reservations", []):
+#             processed_item = {
+#                 "room_number": item.get("room_number"),
+#                 "check_in": datetime.strptime(item.get("check_in"), '%Y-%m-%dT%H:%M:%S'),
+#                 "check_out": datetime.strptime(item.get("check_out"), '%Y-%m-%dT%H:%M:%S'),
+#                 "status": item.get("status"),
+#                 "room_type": item.get("room_type"),
+#             }
+#             processed_data.append(processed_item)
+#             await self._save_to_db(processed_item)
+
+#         self.logger.debug("Все данные успешно сохранены в базу данных.")
+#         return processed_data
+
+#     async def _save_to_db(self, item):
+#         """
+#         Сохраняет данные в базу данных.
+#         """
+#         try:
+#             hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config)
+
+#             reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
+#                 room_number=item["room_number"],
+#                 check_in=item["check_in"],
+#                 defaults={
+#                     "check_out": item["check_out"],
+#                     "status": item["status"],
+#                     "hotel": hotel,
+#                     "room_type": item["room_type"],
+#                 },
+#             )
+
+#             if created:
+#                 self.logger.debug(f"Создана новая запись бронирования: {reservation}")
+#             else:
+#                 self.logger.debug(f"Обновлено существующее бронирование: {reservation}")
+
+#         except Exception as e:
+#             self.logger.error(f"Ошибка при сохранении данных в БД: {e}")
+
+#     async def fetch_and_process_data(self):
+#         """
+#         Загружает данные с API Shelter и сохраняет их в базу данных.
+#         """
+#         self.logger.info("Начало процесса загрузки данных из Shelter PMS.")
+#         try:
+#             data = await self._fetch_data()
+#             self.logger.info(f"Загрузка и обработка данных завершены. Обработано записей: {len(data)}")
+#             return data
+#         except Exception as e:
+#             self.logger.error(f"Ошибка в процессе загрузки данных: {e}")
+#             raise
+
 import logging
 import requests
 from datetime import datetime, timedelta
@@ -6,144 +133,132 @@ from pms_integration.models import PMSConfiguration
 from hotels.models import Hotel, Reservation
 from .base_plugin import BasePMSPlugin

-
-class EcviPMSPlugin(BasePMSPlugin):
+class ShelterPMSPlugin(BasePMSPlugin):
    """
-    Плагин для интеграции с PMS Ecvi (интерфейс для получения данных об отеле).
+    Плагин для интеграции с PMS Shelter (интерфейс для получения данных об отеле).
    """

    def __init__(self, pms_config):
        super().__init__(pms_config)
-        
-        # Инициализация логгера
-        self.logger = logging.getLogger(self.__class__.__name__)  # Логгер с именем класса
-        handler = logging.StreamHandler()  # Потоковый обработчик для вывода в консоль
+        self.logger = logging.getLogger(self.__class__.__name__)
+        handler = logging.StreamHandler()
        formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
        handler.setFormatter(formatter)
        self.logger.addHandler(handler)
-        self.logger.setLevel(logging.DEBUG)  # Уровень логирования
+        self.logger.setLevel(logging.DEBUG)

-        # Инициализация параметров API
-        self.api_url = pms_config.url
-        self.token = pms_config.token
-        self.username = pms_config.username
-        self.password = pms_config.password
-        self.pagination_count = 50  # Максимальное количество записей на страницу (если используется пагинация)
+        self.api_url = "https://pms.frontdesk24.ru/sheltercloudapi/Reservations/"
+        self.access_token = "679CA9C5-9847-4151-883E-5F61181AA37E"

    def get_default_parser_settings(self):
        """
-        Возвращает настройки парсера по умолчанию.
+        Возвращает настройки по умолчанию для обработки данных.
        """
        return {
            "field_mapping": {
-                "check_in": "checkin",
-                "check_out": "checkout",
-                "room_number": "room_name",  
-                "room_type_name": "room_type",
-                "status": "occupancy",
+                "check_in": "check_in",
+                "check_out": "check_out",
+                "room_number": "room_number",
+                "status": "status",
            },
-            "date_format": "%Y-%m-%dT%H:%M:%S"
+            "date_format": "%Y-%m-%dT%H:%M:%S",
        }

    async def _fetch_data(self):
        """
-        Получает данные из PMS API, фильтрует и сохраняет в базу данных.
+        Получает данные бронирований с API PMS Shelter.
        """
-        now = datetime.now()
-        current_date = now.strftime('%Y-%m-%d')
-        yesterday_date = (now - timedelta(days=1)).strftime('%Y-%m-%d')
-
-        headers = {
-            "Content-Type": "application/json",
-        }
-        data = {
-            "token": self.token,
-        }
-
        try:
-            # Запрос данных из PMS API
-            response = await sync_to_async(requests.post)(self.api_url, headers=headers, json=data, auth=(self.username, self.password))
-            response.raise_for_status()  # Если ошибка, выбросит исключение
-            data = response.json()  # Преобразуем ответ в JSON
-            self.logger.debug(f"Получены данные с API: {data}")
-        except requests.exceptions.RequestException as e:
-            self.logger.error(f"Ошибка запроса: {e}")
-            return []
+            # Формируем параметры запроса
+            now = datetime.now()
+            create_from = (now - timedelta(days=7)).strftime("%Y-%m-%d")
+            create_to = now.strftime("%Y-%m-%d")

-        # Фильтрация данных
-        filtered_data = []
-        for item in data:
-            if item.get('occupancy') in ['проживание', 'под выезд', 'под заезд']:
-                filtered_item = {
-                    'checkin': datetime.strptime(item.get('checkin'), '%Y-%m-%d %H:%M:%S'),
-                    'checkout': datetime.strptime(item.get('checkout'), '%Y-%m-%d %H:%M:%S'),
-                    'room_number': item.get('room_name'),
-                    'room_type': item.get('room_type'),
-                    'status': item.get('occupancy')
+            data = {
+                "from": create_from,
+                "until": create_to,
+                "pagination": {
+                    "from": 0,
+                    "count": 100
                }
-                filtered_data.append(filtered_item)
+            }

-        # Логируем результат фильтрации
-        self.logger.debug(f"Отфильтрованные данные: {filtered_data}")
+            headers = {
+                "Content-Type": "application/json",
+                "Accept": "text/plain",
+                "Authorization": f"Bearer {self.access_token}"
+            }

-        # Сохранение данных в базу данных
-        for item in filtered_data:
-            await self._save_to_db(item)
+            # Логирование запроса
+            self.logger.debug(f"Отправка запроса к API: {self.api_url}")
+            self.logger.debug(f"Тело запроса: {data}")
+            self.logger.debug(f"Заголовки: {headers}")

-        self.logger.debug(f"Данные успешно сохранены.")
-        return filtered_data
+            # Выполняем запрос
+            response = requests.post(self.api_url, json=data, headers=headers)
+            response.raise_for_status()

-    async def _save_to_db(self, item):
+            # Обрабатываем ответ
+            bookings = response.json()
+            self.logger.info(f"Получено бронирований: {len(bookings)}")
+            return bookings
+
+        except requests.HTTPError as http_err:
+            self.logger.error(f"HTTP ошибка: {http_err}")
+            self.logger.error(f"Текст ответа: {response.text if 'response' in locals() else 'Нет данных'}")
+            raise
+        except Exception as e:
+            self.logger.error(f"Ошибка получения данных PMS Shelter: {e}")
+            raise
+
+    async def fetch_and_process_data(self):
        """
-        Сохраняет данные в БД (например, информацию о номере).
+        Получение данных с API, обработка и сохранение в базу.
        """
+        self.logger.info("Начало загрузки данных с API Shelter")
        try:
-            # Проверяем, что item — это словарь
-            if not isinstance(item, dict):
-                self.logger.error(f"Ожидался словарь, но получен: {type(item)}. Данные: {item}")
-                return
+            bookings = await self._fetch_data()
+            report = await self._process_and_save_bookings(bookings)
+            self.logger.info(f"Данные успешно обработаны. Отчет: {report}")
+            return report
+        except Exception as e:
+            self.logger.error(f"Ошибка загрузки и обработки данных: {e}")
+            raise

-            # Получаем отель по настройкам PMS
-            hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config)
-            self.logger.debug(f"Отель найден: {hotel.name}")
+    async def _process_and_save_bookings(self, bookings):
+        """
+        Обрабатывает и сохраняет бронирования в базу данных.
+        """
+        self.logger.info("Начало обработки данных о бронированиях для сохранения в базу.")
+        processed_items = 0
+        errors = []

-            # Сохраняем данные бронирования
-            room_number = item.get("room_number")
-            check_in = item.get("checkin")
-            check_out = item.get("checkout")
-            room_type = item.get("room_type")
+        for record in bookings:
+            try:
+                # Пример обработки данных бронирования
+                booking_id = record.get("id")
+                room_number = record.get("room_number")
+                check_in = record.get("check_in")
+                check_out = record.get("check_out")
+                status = record.get("status")

-            # Логируем полученные данные
-            self.logger.debug(f"Полученные данные для сохранения: room_number={room_number}, check_in={check_in}, check_out={check_out}, room_type={room_type}")
-
-            # Проверяем, существует ли запись с таким номером и датой заезда
-            existing_reservation = await sync_to_async(
-                Reservation.objects.filter(room_number=room_number, check_in=check_in).first
-            )()
-
-            if existing_reservation:
-                self.logger.debug(f"Резервация с таким номером и датой заезда уже существует. Обновляем...")
-                await sync_to_async(Reservation.objects.update_or_create)(
-                    room_number=room_number,
-                    check_in=check_in,
+                # Сохраняем или обновляем запись в базе данных
+                reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
+                    external_id=booking_id,
                    defaults={
+                        "hotel": self.pms_config.hotel,
+                        "room_number": room_number,
+                        "check_in": check_in,
                        "check_out": check_out,
-                        "hotel": hotel,
-                        "room_type": room_type,
+                        "status": status,
                    },
                )
-                self.logger.debug(f"Обновлена существующая резервация.")
-            else:
-                self.logger.debug(f"Резервация не найдена, создаем новую...")
-                reservation = await sync_to_async(Reservation.objects.create)(
-                    room_number=room_number,
-                    check_in=check_in,
-                    check_out=check_out,
-                    hotel=hotel,
-                    room_type=room_type,
-                )
-                self.logger.debug(f"Создана новая резервация. ID: {reservation.reservation_id}")

-        except Exception as e:
-            self.logger.error(f"Ошибка сохранения данных: {e}")
+                processed_items += 1

+            except Exception as e:
+                self.logger.error(f"Ошибка обработки бронирования {record.get('id')}: {e}")
+                errors.append(str(e))
+
+        self.logger.info(f"Обработано бронирований: {processed_items}, ошибок: {len(errors)}")
+        return {"processed_items": processed_items, "errors": errors}