ECVI fully functional

2024-12-27 14:47:04 +09:00
parent 9bff204fb4
commit 6af0886a64
10 changed files with 162 additions and 246 deletions
--- a/pms_integration/manager.py
+++ b/pms_integration/manager.py
@@ -33,37 +33,31 @@ class PluginLoader:
        return plugins

 class PMSIntegrationManager:
-    def __init__(self, hotel_id):
-        self.hotel_id = hotel_id
-        self.hotel = None
-        self.pms_config = None
+    def __init__(self, hotel):
+        """
+        Инициализирует PMSIntegrationManager с объектом отеля.
+        :param hotel: Объект отеля, связанный с PMS.
+        """
+        self.hotel = hotel
        self.plugin = None

-    async def load_hotel(self):
+    def load_hotel(self):
        """
-        Загружает данные отеля и PMS конфигурацию.
+        Проверяет, что у отеля есть связанная PMS конфигурация.
        """
-        from hotels.models import Hotel
-        self.hotel = await sync_to_async(Hotel.objects.select_related("pms").get)(id=self.hotel_id)
-        self.pms_config = self.hotel.pms
-        if not self.pms_config:
+        if not self.hotel.pms:
            raise ValueError(f"Отель {self.hotel.name} не имеет связанной PMS конфигурации.")

    def load_plugin(self):
        """
-        Загружает плагин для PMS на основе конфигурации отеля.
+        Загружает плагин, соответствующий PMS конфигурации отеля.
        """
-        plugins = PluginLoader.load_plugins()
-        if not self.hotel:
-            raise ValueError("Отель не загружен. Пожалуйста, вызовите load_hotel перед загрузкой плагина.")
-        if not self.hotel.pms:
-            raise ValueError(f"Отель {self.hotel.name} не связан с PMS конфигурацией.")
-        if self.hotel.pms.plugin_name not in plugins:
-            raise ValueError(f"Плагин для PMS {self.hotel.pms.plugin_name} не найден.")
-
-        # Передача объекта Hotel в плагин
-        self.plugin = plugins[self.hotel.pms.plugin_name](self.hotel)
-
+        pms_name = self.hotel.pms.plugin_name.lower()  # Приводим название плагина к нижнему регистру
+        if pms_name == "ecvi_intermark" or pms_name == "ecvi":
+            from pms_integration.plugins.ecvi_pms import EcviPMSPlugin
+            self.plugin = EcviPMSPlugin(self.hotel)
+        else:
+            raise ValueError(f"Неизвестный PMS: {pms_name}")
    def fetch_data(self):
        """
        Получает данные из PMS с использованием загруженного плагина.
--- a/pms_integration/plugins/base_plugin.py
+++ b/pms_integration/plugins/base_plugin.py
@@ -8,26 +8,25 @@ class BasePMSPlugin(ABC):
    - Предоставлять дефолтные parser_settings
    - Проходить базовую валидацию (validate_plugin)
    """
-    
+
    def __init__(self, pms_config):
        """
        pms_config: объект PMSConfiguration
        """
        self.pms_config = pms_config

-
    @abstractmethod
-    def _fetch_data(self):
+    async def _fetch_data(self):
        """
        Абстрактный метод для получения данных.
        """
        pass

-    def fetch_data(self):
+    async def fetch_data(self):
        """
        Обертка для выполнения _fetch_data с возможной дополнительной обработкой.
        """
-        return self._fetch_data()
+        return await self._fetch_data()

    @abstractmethod
    def get_default_parser_settings(self):
@@ -45,7 +44,6 @@ class BasePMSPlugin(ABC):
            }
        }
        """
-        print("get_default_parser_settings. pms_config:", self.pms_config)
        return {}

    def validate_plugin(self):
@@ -53,10 +51,8 @@ class BasePMSPlugin(ABC):
        Проверка на соответствие требованиям.
        Можно проверить наличие методов или полей.
        """
-        # Например, проверить наличие fetch_data и get_default_parser_settings
-        required_methods = ["fetch_data", "get_default_parser_settings"]
-        for m in required_methods:
-            if not hasattr(self, m):
-                raise ValueError(f"Плагин {type(self).__name__} не реализует метод {m}.")
-        # Можно добавить дополнительные проверки
+        required_methods = ["fetch_data", "get_default_parser_settings", "_fetch_data"]
+        for method in required_methods:
+            if not hasattr(self, method):
+                raise ValueError(f"Плагин {type(self).__name__} не реализует метод {method}.")
        return True
--- a/pms_integration/plugins/ecvi_pms.py
+++ b/pms_integration/plugins/ecvi_pms.py
@@ -7,179 +7,136 @@ from .base_plugin import BasePMSPlugin

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

-    def __init__(self, pms_config):
-        super().__init__(pms_config)
-        
-        # Инициализация логгера
-        self.logger = logging.getLogger(self.__class__.__name__)  # Логгер с именем класса
-        handler_console = logging.StreamHandler()  # Потоковый обработчик для вывода в консоль
-        handler_file = logging.FileHandler('ecvi_pms_plugin.log')  # Обработчик для записи в файл
+    def __init__(self, hotel):
+        super().__init__(hotel.pms)  # Передаем PMS-конфигурацию в базовый класс
+        self.hotel = hotel  # Сохраняем объект отеля

+        # Проверка PMS-конфигурации
+        if not self.hotel.pms:
+            raise ValueError(f"Отель {self.hotel.name} не имеет связанной PMS конфигурации.")
+
+        # Инициализация параметров API
+        self.api_url = self.hotel.pms.url
+        self.token = self.hotel.pms.token
+        self.username = self.hotel.pms.username
+        self.password = self.hotel.pms.password
+
+        # Настройка логгера
+        self.logger = logging.getLogger(self.__class__.__name__)
+        handler_console = logging.StreamHandler()
+        handler_file = logging.FileHandler('ecvi_pms_plugin.log')
        formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
        handler_console.setFormatter(formatter)
        handler_file.setFormatter(formatter)
-
-        # Добавляем оба обработчика к логгеру
        self.logger.addHandler(handler_console)
        self.logger.addHandler(handler_file)
-        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.logger.setLevel(logging.DEBUG)

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

    async def _fetch_data(self):
        """
-        Получает данные из PMS API, фильтрует и сохраняет в базу данных.
+        Получает данные из PMS API и сохраняет их в базу.
        """
-        now = datetime.now()
-        current_date = now.strftime('%Y-%m-%d')
-        yesterday_date = (now - timedelta(days=1)).strftime('%Y-%m-%d')
-        processed_items = 0
-        errors = []
-        headers = {
-            "Content-Type": "application/json",
-        }
-        data = {
-            "token": self.token,
-        }
+        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}")
+            response = await sync_to_async(requests.post)(
+                self.api_url, headers=headers, json=data, auth=(self.username, self.password)
+            )
+            response.raise_for_status()
+            response_data = response.json()
+            self.logger.debug(f"Полученные данные с API: {response_data}")
+            return await self._process_data(response_data)
        except requests.exceptions.RequestException as e:
-            self.logger.error(f"Ошибка запроса: {e}")
-            return []
+            self.logger.error(f"Ошибка API: {e}")
+            return {
+                "processed_intervals": 0,
+                "processed_items": 0,
+                "errors": [str(e)]
+            }
+
+    async def _process_data(self, data):
+        """
+        Обрабатывает данные и сохраняет их в базу.
+        """
+        processed_items = 0
+        errors = []
+
+        date_formats = ["%Y-%m-%d %H:%M:%S", "%Y-%m-%dT%H:%M:%S"]  # Поддержка нескольких форматов даты

-        # Фильтрация данных
-        filtered_data = []
        for item in data:
            try:
-                if not isinstance(item, dict):
-                    raise ValueError(f"Некорректный формат элемента: {item}")
+                # Парсинг даты с поддержкой нескольких форматов
+                checkin = self._parse_date(item['checkin'], date_formats)
+                checkout = self._parse_date(item['checkout'], date_formats)

-                reservation_id = item.get('task_id')
-                if not reservation_id:
-                    raise ValueError("Отсутствует task_id в записи")
+                reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
+                    reservation_id=item['task_id'],
+                    defaults={
+                        'room_number': item['room_name'],
+                        'room_type': item['room_type'],
+                        'check_in': checkin,
+                        'check_out': checkout,
+                        'status': item['occupancy'],
+                        'hotel': self.hotel,
+                    }
+                )

-                checkin = datetime.strptime(item.get('checkin'), '%Y-%m-%d %H:%M:%S')
-                checkout = datetime.strptime(item.get('checkout'), '%Y-%m-%d %H:%M:%S')
+                if created:
+                    self.logger.debug(f"Создана новая резервация: {reservation.reservation_id}")
+                else:
+                    self.logger.debug(f"Обновлена существующая резервация: {reservation.reservation_id}")

-                filtered_data.append({
-                    'reservation_id': reservation_id,
-                    'room_number': item.get('room_name'),
-                    'room_type': item.get('room_type'),
-                    'checkin': checkin,
-                    'checkout': checkout,
-                    'status': item.get('occupancy')
-                })
                processed_items += 1
+
            except Exception as e:
-                self.logger.error(f"Ошибка обработки элемента: {e}")
+                self.logger.error(f"Ошибка обработки записи: {e}")
                errors.append(str(e))

-        # Сохранение данных в базу данных
-        try:
-            for item in filtered_data:
-                await self._save_to_db(item)
-        except Exception as e:
-            self.logger.error(f"Ошибка сохранения данных в БД: {e}")
-            errors.append(f"Ошибка сохранения данных в БД: {str(e)}")
-
-        # Формирование отчета
-        report = {
+        return {
            "processed_intervals": 1,
            "processed_items": processed_items,
            "errors": errors
        }
-        self.logger.debug(f"Сформированный отчет: {report}")
-        return report

-    async def _save_to_db(self, item):
+    @staticmethod
+    def _parse_date(date_str, formats):
        """
-        Сохраняет данные в БД (например, информацию о номере).
+        Парсит дату, пытаясь использовать несколько форматов.
        """
-        try:
-            # Получаем отель по настройкам PMS
-            hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config)
-            self.logger.debug(f"Отель найден: {hotel.name}")
-
-            # Проверяем, существует ли уже резервация с таким внешним ID
-            reservation_id = item.get('reservation_id')
-            if not reservation_id:
-                self.logger.error("Ошибка: 'reservation_id' отсутствует в данных.")
-                return
-            
-            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('room_number'),
-                        'room_type': item.get('room_type'),
-                        'check_in': item.get('checkin'),
-                        'check_out': item.get('checkout'),
-                        '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('room_number'),
-                    room_type=item.get('room_type'),
-                    check_in=item.get('checkin'),
-                    check_out=item.get('checkout'),
-                    status=item.get('status'),
-                    hotel=hotel
-                )
-                self.logger.debug(f"Новая резервация создана с ID: {reservation.reservation_id}")
-
-        except Exception as e:
-            self.logger.error(f"Ошибка сохранения данных: {e}")
+        for fmt in formats:
+            try:
+                return datetime.strptime(date_str, fmt)
+            except ValueError:
+                continue
+        raise ValueError(f"Дата '{date_str}' не соответствует ожидаемым форматам: {formats}")

    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}.")
+        for method in required_methods:
+            if not hasattr(self, method):
+                raise ValueError(f"Плагин {type(self).__name__} не реализует метод {method}.")
        self.logger.debug(f"Плагин {self.__class__.__name__} прошел валидацию.")
        return True