import requests import hashlib import json from .base_plugin import BasePMSPlugin from datetime import datetime, timedelta from asgiref.sync import sync_to_async from touchh.utils.log import CustomLogger from hotels.models import Hotel, Reservation from app_settings.models import GlobalHotelSettings from django.utils import timezone class RealtyCalendarPlugin(BasePMSPlugin): 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") def get_default_parser_settings(self): """ Возвращает настройки по умолчанию для обработки данных. """ return { "date_format": "%Y-%m-%dT%H:%M:%S", "timezone": "UTC" } def _get_sorted_keys(self, obj): """ Возвращает отсортированный по имени список ключей. """ sorted_keys = sorted(obj.keys()) self.logger.debug(f"Отсортированные ключи: {sorted_keys}") return sorted_keys def _generate_data_string(self, obj): """ Формирует строку параметров для подписи. """ sorted_keys = self._get_sorted_keys(obj) string = "".join(f"{key}={obj[key]}" for key in sorted_keys) self.logger.debug(f"Сформированная строка данных: {string}") return string + self.private_key def _generate_md5(self, string): """ Генерирует MD5-хеш от строки. """ md5_hash = hashlib.md5(string.encode("utf-8")).hexdigest() self.logger.debug(f"Сформированный MD5-хеш: {md5_hash}") return md5_hash def _generate_sign(self, data): """ Генерирует подпись для данных запроса. """ data_string = self._generate_data_string(data) self.logger.debug(f"Строка для подписи: {data_string}") sign = self._generate_md5(data_string) self.logger.debug(f"Подпись: {sign}") return sign async def _fetch_data(self): """ Выполняет запрос к API RealtyCalendar для получения данных о бронированиях. """ 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", } now = datetime.now() data = { "begin_date": (now - timedelta(days=7)).strftime("%Y-%m-%d"), "end_date": now.strftime("%Y-%m-%d"), } data["sign"] = self._generate_sign(data) response = requests.post(url=base_url, headers=headers, json=data) self.logger.debug(f"Статус ответа: {response.status_code}") if response.status_code != 200: self.logger.error(f"Ошибка API: {response.status_code}, {response.text}") raise ValueError(f"Ошибка API RealtyCalendar: {response.status_code}") try: response_data = response.json() bookings = response_data.get("bookings", []) if not isinstance(bookings, list): raise ValueError(f"Ожидался список, но получен {type(bookings)}") except Exception as e: self.logger.error(f"Ошибка обработки ответа API: {e}") raise # Получаем глобальные настройки отеля hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config) hotel_tz = hotel.timezone try: hotel_settings = await sync_to_async(GlobalHotelSettings.objects.first)() check_in_time = hotel_settings.check_in_time.strftime("%H:%M:%S") check_out_time = hotel_settings.check_out_time.strftime("%H:%M:%S") except AttributeError: # Используем значения по умолчанию, если настроек нет check_in_time = "14:00:00" check_out_time = "12:00:00" filtered_data = [ { 'reservation_id': item.get('id'), 'checkin': timezone.make_aware( datetime.strptime( f"{item.get('begin_date')} {check_in_time}", "%Y-%m-%d %H:%M:%S" ) ), 'checkout': timezone.make_aware( datetime.strptime( f"{item.get('end_date')} {check_out_time}", "%Y-%m-%d %H:%M:%S" ) ), 'room_number': item.get('apartment_id'), 'room_type': item.get('notes', 'Описание отсутствует'), 'status': item.get('status') } for item in bookings if isinstance(item, dict) and item.get("status") in ["booked", "request"] ] await self._save_to_db(filtered_data) async def _save_to_db(self, data): """ Сохраняет данные в БД (например, информацию о номере). """ if not isinstance(data, list): self.logger.error(f"Ожидался список записей, но получен {type(data).__name__}") return for index, item in enumerate(data, start=1): try: hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config) reservation_id = item.get('reservation_id') if not reservation_id: self.logger.error(f"Пропущена запись {index}: отсутствует 'id'") continue existing_reservation = await sync_to_async(Reservation.objects.filter)(reservation_id=reservation_id) existing_reservation = await sync_to_async(existing_reservation.first)() defaults = { 'room_number': item['room_number'], 'room_type': item['room_type'], 'check_in': item['checkin'], 'check_out': item['checkout'], 'status': item['status'], 'hotel': hotel } if existing_reservation: await sync_to_async(Reservation.objects.update_or_create)( reservation_id=reservation_id, defaults=defaults ) self.logger.debug(f"Резервация {reservation_id} обновлена.") else: await sync_to_async(Reservation.objects.create)( reservation_id=reservation_id, **defaults ) self.logger.debug(f"Создана новая резервация {reservation_id}") except Exception as e: self.logger.error(f"Ошибка при обработке записи {index}: {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