112 lines
4.5 KiB
Python
112 lines
4.5 KiB
Python
import logging
|
|
import requests
|
|
import json
|
|
import os
|
|
from datetime import datetime, timedelta
|
|
from asgiref.sync import sync_to_async
|
|
from hotels.models import Hotel, Reservation
|
|
from .base_plugin import BasePMSPlugin
|
|
|
|
class RealtyCalendarPMSPlugin(BasePMSPlugin):
|
|
def __init__(self, hotel):
|
|
super().__init__(hotel.pms)
|
|
self.hotel = hotel
|
|
|
|
if not self.hotel.pms:
|
|
raise ValueError(f"Отель {self.hotel.name} не имеет связанной PMS конфигурации.")
|
|
|
|
self.api_url = self.hotel.pms.url.rstrip("/")
|
|
self.public_key = self.hotel.pms.public_key
|
|
self.private_key = self.hotel.pms.private_key
|
|
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
handler_console = logging.StreamHandler()
|
|
handler_file = logging.FileHandler('var/log/realtycalendar_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.WARNING)
|
|
|
|
def get_default_parser_settings(self):
|
|
return {
|
|
"field_mapping": {
|
|
"check_in": "begin_date",
|
|
"check_out": "end_date",
|
|
"room_number": "apartment_id",
|
|
"room_type_name": "notes",
|
|
"status": "status",
|
|
},
|
|
"date_format": "%Y-%m-%d %H:%M:%S"
|
|
}
|
|
|
|
async def _fetch_data(self):
|
|
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"),
|
|
"sign": self._generate_sign()
|
|
}
|
|
|
|
try:
|
|
response = await sync_to_async(requests.post)(
|
|
f"{self.api_url}/api/v1/bookings/{self.public_key}/",
|
|
headers=headers, json=data
|
|
)
|
|
response.raise_for_status()
|
|
response_data = response.json()
|
|
return await self._process_data(response_data)
|
|
except requests.exceptions.RequestException as e:
|
|
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"]
|
|
|
|
for item in data.get("bookings", []):
|
|
try:
|
|
checkin = self._parse_date(item['begin_date'], date_formats)
|
|
checkout = self._parse_date(item['end_date'], date_formats)
|
|
|
|
reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
|
|
reservation_id=item['id'],
|
|
defaults={
|
|
'room_number': item['apartment_id'],
|
|
'room_type': item.get('notes', 'Описание отсутствует'),
|
|
'check_in': checkin,
|
|
'check_out': checkout,
|
|
'status': item['status'],
|
|
'hotel': self.hotel,
|
|
}
|
|
)
|
|
processed_items += 1
|
|
except Exception as e:
|
|
self.logger.error(f"Ошибка обработки записи: {e}")
|
|
errors.append(str(e))
|
|
|
|
return {"processed_intervals": 1, "processed_items": processed_items, "errors": errors}
|
|
|
|
def _generate_sign(self):
|
|
return hashlib.md5((self.public_key + self.private_key).encode("utf-8")).hexdigest()
|
|
|
|
@staticmethod
|
|
def _parse_date(date_str, formats):
|
|
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 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
|