183 lines
8.0 KiB
Python
183 lines
8.0 KiB
Python
import requests
|
||
import json
|
||
from datetime import datetime, timedelta, timezone
|
||
|
||
from asgiref.sync import sync_to_async
|
||
from hotels.models import Reservation, Hotel
|
||
from .base_plugin import BasePMSPlugin
|
||
from pms_integration.models import PMSConfiguration
|
||
|
||
class Shelter(BasePMSPlugin):
|
||
"""
|
||
Плагин для интеграции с Shelter PMS.
|
||
"""
|
||
def __init__(self, pms_config):
|
||
super().__init__(pms_config)
|
||
self.api_url = pms_config.url
|
||
self.token = pms_config.token
|
||
self.pagination_count = 50
|
||
|
||
def get_default_parser_settings(self):
|
||
"""
|
||
Возвращает настройки по умолчанию для разбора данных PMS Shelter.
|
||
"""
|
||
return {
|
||
"fields_mapping": {
|
||
"reservation_id": "id",
|
||
"hotel_id": "hotelId",
|
||
"hotel_name": "hotelName",
|
||
"check_in": "from",
|
||
"check_out": "until",
|
||
"reservation_date": "date",
|
||
"room_type_id": "roomTypeId",
|
||
"room_id": "roomId",
|
||
"room_number": "roomNumber",
|
||
"room_type_name": "roomTypeName",
|
||
"check_in_status": "checkInStatus",
|
||
"is_annul": "isAnnul",
|
||
"tariff_id": "tariffId",
|
||
"reservation_price": "reservationPrice",
|
||
"discount": "discount",
|
||
"guests": "guests",
|
||
},
|
||
"date_format": "%Y-%m-%dT%H:%M:%S",
|
||
"timezone": "UTC",
|
||
}
|
||
|
||
async def _get_last_saved_date(self):
|
||
"""
|
||
Получает дату последнего сохраненного бронирования для отеля.
|
||
"""
|
||
try:
|
||
last_reservation = await sync_to_async(
|
||
Reservation.objects.filter(hotel__pms=self.pms_config).order_by('-check_in').first
|
||
)()
|
||
return last_reservation.check_in if last_reservation else None
|
||
except Exception as e:
|
||
print(f"[ERROR] Ошибка получения последнего сохраненного бронирования: {e}")
|
||
return None
|
||
|
||
async def _fetch_data(self):
|
||
"""
|
||
Получает данные о бронированиях из PMS Shelter.
|
||
"""
|
||
try:
|
||
now = datetime.utcnow()
|
||
start_date = await self._get_last_saved_date() or (now - timedelta(days=60))
|
||
end_date = now + timedelta(days=60)
|
||
total_count = None
|
||
from_index = 0
|
||
|
||
headers = {
|
||
'accept': 'text/plain',
|
||
'Authorization': f'Bearer {self.token}',
|
||
'Content-Type': 'application/json',
|
||
}
|
||
|
||
print(f"[DEBUG] Start date: {start_date}, End date: {end_date}")
|
||
|
||
while total_count is None or from_index < total_count:
|
||
payload = {
|
||
"from": start_date.strftime('%Y-%m-%dT%H:%M:%SZ'),
|
||
"until": end_date.strftime('%Y-%m-%dT%H:%M:%SZ'),
|
||
"pagination": {
|
||
"from": from_index,
|
||
"count": self.pagination_count,
|
||
},
|
||
}
|
||
print(f"[DEBUG] Payload: {json.dumps(payload)}")
|
||
|
||
try:
|
||
response = await sync_to_async(requests.post)(self.api_url, headers=headers, data=json.dumps(payload))
|
||
except requests.RequestException as e:
|
||
print(f"[ERROR] Ошибка HTTP-запроса: {e}")
|
||
raise ValueError(f"Ошибка HTTP-запроса: {e}")
|
||
|
||
print(f"[DEBUG] Response status: {response.status_code}")
|
||
|
||
if response.status_code != 200:
|
||
print(f"[ERROR] Request error: {response.status_code}, {response.text}")
|
||
raise ValueError(f"Ошибка запроса: {response.status_code}, {response.text}")
|
||
|
||
try:
|
||
data = response.json()
|
||
except json.JSONDecodeError as e:
|
||
print(f"[ERROR] Ошибка декодирования JSON: {e}")
|
||
raise ValueError(f"Ошибка декодирования JSON: {e}")
|
||
|
||
# Проверяем, что ответ содержит ключи "count" и "items"
|
||
if not isinstance(data, dict) or "count" not in data or "items" not in data:
|
||
print(f"[ERROR] Неверный формат данных: {data}")
|
||
raise ValueError(f"Неверный формат данных: {data}")
|
||
|
||
total_count = data.get("count", 0)
|
||
items = data.get("items", [])
|
||
|
||
print(f"[DEBUG] Total count: {total_count}, Items retrieved: {len(items)}")
|
||
|
||
if not isinstance(items, list):
|
||
print(f"[ERROR] Неверный тип items: {type(items)}. Ожидался list.")
|
||
raise ValueError(f"Неверный тип items: {type(items)}. Ожидался list.")
|
||
|
||
for item in items:
|
||
if not isinstance(item, dict):
|
||
print(f"[ERROR] Неверный формат элемента items: {item}")
|
||
continue
|
||
|
||
try:
|
||
await self._save_to_db(item)
|
||
except Exception as e:
|
||
print(f"[ERROR] Ошибка сохранения бронирования {item.get('id')}: {e}")
|
||
|
||
from_index += len(items)
|
||
print(f"[DEBUG] Updated from_index: {from_index}")
|
||
|
||
except Exception as e:
|
||
print(f"[ERROR] Общая ошибка в методе _fetch_data: {e}")
|
||
|
||
|
||
async def _save_to_db(self, item):
|
||
"""
|
||
Сохраняет данные о бронировании в БД.
|
||
"""
|
||
try:
|
||
print(f"[DEBUG] Fetching hotel for PMS: {self.pms_config}")
|
||
hotel = await sync_to_async(Hotel.objects.get)(pms=self.pms_config)
|
||
print(f"[DEBUG] Hotel found: {hotel.name}")
|
||
|
||
# Учитываем формат даты без 'Z'
|
||
date_format = '%Y-%m-%dT%H:%M:%S'
|
||
print(f"[DEBUG] Parsing check-in and check-out dates for reservation {item['id']}")
|
||
|
||
check_in = datetime.strptime(item["from"], date_format).replace(tzinfo=timezone.utc)
|
||
check_out = datetime.strptime(item["until"], date_format).replace(tzinfo=timezone.utc)
|
||
|
||
# Проверяем room_number и устанавливаем значение по умолчанию, если оно отсутствует
|
||
room_number = item.get("roomNumber", "") or "Unknown"
|
||
print(f"[DEBUG] Room number determined: {room_number}")
|
||
|
||
# Сохраняем бронирование
|
||
print(f"[DEBUG] Saving reservation {item['id']} to database")
|
||
reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
|
||
reservation_id=item["id"],
|
||
hotel=hotel,
|
||
defaults={
|
||
"room_number": room_number,
|
||
"room_type": item.get("roomTypeName", ""),
|
||
"check_in": check_in,
|
||
"check_out": check_out,
|
||
"status": item.get("checkInStatus", ""),
|
||
"price": item.get("reservationPrice", 0),
|
||
"discount": item.get("discount", 0),
|
||
},
|
||
)
|
||
|
||
print(f"[DEBUG] {'Created' if created else 'Updated'} reservation {item['id']}")
|
||
|
||
except KeyError as ke:
|
||
print(f"[ERROR] Ошибка обработки ключей в элементе {item}: {ke}")
|
||
except ValueError as ve:
|
||
print(f"[ERROR] Ошибка обработки данных для бронирования {item['id']}: {ve}")
|
||
except Exception as e:
|
||
print(f"[ERROR] Общая ошибка сохранения бронирования {item.get('id', 'Unknown')}: {e}")
|