plugins devel
This commit is contained in:
@@ -1,129 +1,10 @@
|
||||
# import requests
|
||||
# import hashlib
|
||||
# import json
|
||||
# from .base_plugin import BasePMSPlugin
|
||||
# from datetime import datetime, timedelta
|
||||
# from asgiref.sync import sync_to_async
|
||||
|
||||
|
||||
# class RealtyCalendarPlugin(BasePMSPlugin):
|
||||
# """Плагин для импорта данных из системы RealtyCalendar
|
||||
# """
|
||||
# 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("/") # Убираем лишний `/` в конце URL
|
||||
|
||||
# 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):
|
||||
# """
|
||||
# Возвращает отсортированный по имени список ключей.
|
||||
# """
|
||||
# return sorted(list(obj.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)
|
||||
# return string + self.private_key
|
||||
|
||||
# def _generate_md5(self, string):
|
||||
# """
|
||||
# Генерирует MD5-хеш от строки.
|
||||
# """
|
||||
# return hashlib.md5(string.encode("utf-8")).hexdigest()
|
||||
|
||||
# def _generate_sign(self, data):
|
||||
# """
|
||||
# Генерирует подпись для данных запроса.
|
||||
# """
|
||||
# data_string = self._generate_data_string(data)
|
||||
# return self._generate_md5(data_string)
|
||||
|
||||
# def fetch_data(self):
|
||||
# """
|
||||
# Выполняет запрос к API RealtyCalendar для получения данных о бронированиях.
|
||||
# """
|
||||
# base_url = f"https://realtycalendar.ru/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)
|
||||
|
||||
# # Отправляем запрос
|
||||
# print(f"URL запроса: {base_url}")
|
||||
# print(f"Заголовки: {headers}")
|
||||
# print(f"Данные запроса: {data}")
|
||||
|
||||
# response = requests.post(url=base_url, headers=headers, json=data)
|
||||
|
||||
# # Логируем результат
|
||||
# print(f"Статус ответа: {response.status_code}")
|
||||
# print(f"Ответ: {response.text}")
|
||||
|
||||
# # Проверяем успешность запроса
|
||||
# if response.status_code == 200:
|
||||
# return response.json().get("bookings", [])
|
||||
# else:
|
||||
# raise ValueError(f"Ошибка API RealtyCalendar: {response.status_code}, {response.text}")
|
||||
|
||||
# async def _save_to_db(self, data, hotel_id):
|
||||
# """
|
||||
# Сохраняет данные о бронированиях в базу данных.
|
||||
# """
|
||||
# from hotels.models import Reservation, Hotel
|
||||
|
||||
# hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
|
||||
# for item in data:
|
||||
# try:
|
||||
# reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
|
||||
# reservation_id=item["id"],
|
||||
# hotel=hotel,
|
||||
# defaults={
|
||||
# "room_number": item.get("apartment_id", ""), # ID квартиры
|
||||
# "check_in": datetime.strptime(item["begin_date"], "%Y-%m-%d"), # Дата заезда
|
||||
# "check_out": datetime.strptime(item["end_date"], "%Y-%m-%d"), # Дата выезда
|
||||
# "status": item.get("status", ""), # Статус бронирования
|
||||
# "price": item.get("amount", 0), # Сумма оплаты
|
||||
# "client_name": item["client"].get("fio", ""), # Имя клиента
|
||||
# "client_email": item["client"].get("email", ""), # Email клиента
|
||||
# "client_phone": item["client"].get("phone", ""), # Телефон клиента
|
||||
# }
|
||||
# )
|
||||
# print(f"{'Создана' if created else 'Обновлена'} запись: {reservation}")
|
||||
# except Exception as e:
|
||||
# print(f"Ошибка при сохранении бронирования ID {item['id']}: {e}")
|
||||
|
||||
|
||||
import requests
|
||||
import hashlib
|
||||
import json
|
||||
from .base_plugin import BasePMSPlugin
|
||||
from datetime import datetime, timedelta
|
||||
from asgiref.sync import sync_to_async
|
||||
|
||||
from math import ceil
|
||||
|
||||
class RealtyCalendarPlugin(BasePMSPlugin):
|
||||
"""Плагин для импорта данных из системы RealtyCalendar
|
||||
@@ -196,6 +77,7 @@ class RealtyCalendarPlugin(BasePMSPlugin):
|
||||
data = {
|
||||
"begin_date": (now - timedelta(days=7)).strftime("%Y-%m-%d"),
|
||||
"end_date": now.strftime("%Y-%m-%d"),
|
||||
"room_number": ""
|
||||
}
|
||||
|
||||
print(f"[DEBUG] Даты выборки: {data}")
|
||||
@@ -223,32 +105,53 @@ class RealtyCalendarPlugin(BasePMSPlugin):
|
||||
raise ValueError(f"Ошибка API RealtyCalendar: {response.status_code}, {response.text}")
|
||||
|
||||
|
||||
async def _save_to_db(self, data, hotel_id):
|
||||
async def _save_to_db(self, data, hotel_id, batch_size=50):
|
||||
"""
|
||||
Сохраняет данные о бронированиях в базу данных.
|
||||
Сохраняет данные о бронированиях в базу данных партиями.
|
||||
"""
|
||||
from hotels.models import Reservation, Hotel
|
||||
|
||||
hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
|
||||
print(f"[DEBUG] Загружен отель: {hotel.name}")
|
||||
try:
|
||||
hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
|
||||
self.logger.info(f"Загружен отель: {hotel.name}")
|
||||
|
||||
for item in data:
|
||||
print(f"[DEBUG] Обработка бронирования: {item}")
|
||||
try:
|
||||
reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
|
||||
reservation_id=item["id"],
|
||||
hotel=hotel,
|
||||
defaults={
|
||||
"room_number": item.get("apartment_id", ""), # ID квартиры
|
||||
"check_in": datetime.strptime(item["begin_date"], "%Y-%m-%d"), # Дата заезда
|
||||
"check_out": datetime.strptime(item["end_date"], "%Y-%m-%d"), # Дата выезда
|
||||
"status": item.get("status", ""), # Статус бронирования
|
||||
"price": item.get("amount", 0), # Сумма оплаты
|
||||
"client_name": item["client"].get("fio", ""), # Имя клиента
|
||||
"client_email": item["client"].get("email", ""), # Email клиента
|
||||
"client_phone": item["client"].get("phone", ""), # Телефон клиента
|
||||
}
|
||||
)
|
||||
print(f"[DEBUG] {'Создана' if created else 'Обновлена'} запись: {reservation}")
|
||||
except Exception as e:
|
||||
print(f"[DEBUG] Ошибка при сохранении бронирования ID {item['id']}: {e}")
|
||||
# Разделение данных на батчи
|
||||
total_records = len(data)
|
||||
batches = [data[i:i + batch_size] for i in range(0, total_records, batch_size)]
|
||||
self.logger.info(f"Обработка {total_records} записей в {len(batches)} партиях...")
|
||||
|
||||
for batch_index, batch in enumerate(batches):
|
||||
self.logger.info(f"Обработка партии {batch_index + 1}/{len(batches)}")
|
||||
|
||||
for item in batch:
|
||||
try:
|
||||
if item.get("is_delete", False):
|
||||
self.logger.info(f"Пропущена запись с ID {item.get('id')} (удалена).")
|
||||
continue
|
||||
|
||||
client_data = item.get("client", {})
|
||||
if not item.get("id") or not item.get("begin_date") or not item.get("end_date"):
|
||||
self.logger.warning(f"Пропущена запись с неполными данными: {item}")
|
||||
continue
|
||||
|
||||
reservation_defaults = {
|
||||
"room_number": item.get("apartment_id", ""),
|
||||
"check_in": datetime.strptime(item["begin_date"], "%Y-%m-%d"),
|
||||
"check_out": datetime.strptime(item["end_date"], "%Y-%m-%d"),
|
||||
"status": item.get("status", ""),
|
||||
"price": item.get("amount", 0),
|
||||
"client_name": client_data.get("fio", ""),
|
||||
"client_email": client_data.get("email", ""),
|
||||
"client_phone": client_data.get("phone", ""),
|
||||
}
|
||||
|
||||
await sync_to_async(Reservation.objects.update_or_create)(
|
||||
reservation_id=item["id"],
|
||||
hotel=hotel,
|
||||
defaults=reservation_defaults
|
||||
)
|
||||
self.logger.info(f"Сохранена запись для бронирования ID {item['id']}.")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при обработке бронирования ID {item.get('id', 'неизвестно')}: {e}")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при обработке данных: {e}")
|
||||
116
pms_integration/plugins/trevelline_pms.py
Normal file
116
pms_integration/plugins/trevelline_pms.py
Normal file
@@ -0,0 +1,116 @@
|
||||
import logging
|
||||
import requests
|
||||
from datetime import datetime
|
||||
from .base_plugin import BasePMSPlugin
|
||||
|
||||
class TravelLinePMSPlugin(BasePMSPlugin):
|
||||
"""
|
||||
Плагин для интеграции с PMS TravelLine.
|
||||
"""
|
||||
BASE_URL = "https://partner.tlintegration.com/api/webpms/v1"
|
||||
|
||||
def __init__(self, pms_config):
|
||||
"""
|
||||
Инициализация плагина с конфигурацией PMS.
|
||||
|
||||
:param pms_config: Конфигурация PMS (объект PMSConfiguration).
|
||||
"""
|
||||
super().__init__(pms_config)
|
||||
self.api_key = pms_config.api_key
|
||||
self.logger = logging.getLogger(self.__class__.__name__)
|
||||
|
||||
def _get_headers(self):
|
||||
"""
|
||||
Возвращает заголовки для запросов.
|
||||
"""
|
||||
return {
|
||||
"Authorization": f"Bearer {self.api_key}",
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
|
||||
def _fetch_data(self):
|
||||
"""
|
||||
Получение данных из API TravelLine (поиск бронирований).
|
||||
|
||||
:return: Список номеров бронирований.
|
||||
"""
|
||||
url = f"{self.BASE_URL}/bookings"
|
||||
params = {
|
||||
"roomId": self.pms_config.room_id,
|
||||
"modifiedFrom": self.pms_config.modified_from,
|
||||
"modifiedTo": self.pms_config.modified_to,
|
||||
"state": self.pms_config.state,
|
||||
"affectsPeriodFrom": self.pms_config.affects_period_from,
|
||||
"affectsPeriodTo": self.pms_config.affects_period_to,
|
||||
}
|
||||
params = {k: v for k, v in params.items() if v is not None}
|
||||
|
||||
try:
|
||||
response = requests.get(url, headers=self._get_headers(), params=params)
|
||||
response.raise_for_status()
|
||||
self.logger.info("Данные успешно получены из API TravelLine.")
|
||||
return response.json().get("bookingNumbers", [])
|
||||
except requests.RequestException as e:
|
||||
self.logger.error(f"Ошибка при запросе к API TravelLine: {e}")
|
||||
return []
|
||||
|
||||
def fetch_data(self):
|
||||
"""
|
||||
Обертка для получения данных из API TravelLine с дополнительной обработкой.
|
||||
|
||||
:return: Список номеров бронирований.
|
||||
"""
|
||||
return self._fetch_data()
|
||||
|
||||
def get_default_parser_settings(self):
|
||||
"""
|
||||
Возвращает настройки парсера по умолчанию.
|
||||
"""
|
||||
return {
|
||||
"field_mapping": {
|
||||
"reservation_id": "bookingNumber",
|
||||
"check_in": "actualCheckInDateTime",
|
||||
"check_out": "actualCheckOutDateTime",
|
||||
"room_number": "roomId",
|
||||
"status": "state",
|
||||
},
|
||||
"date_format": "%Y-%m-%dT%H:%M"
|
||||
}
|
||||
|
||||
def process_data(self, booking_number, room_stay_id, action, actual_date_time):
|
||||
"""
|
||||
Обработка данных для заселения или выселения проживания.
|
||||
|
||||
:param booking_number: Номер бронирования.
|
||||
:param room_stay_id: Идентификатор проживания.
|
||||
:param action: Действие ("check-in" или "check-out").
|
||||
:param actual_date_time: Фактические дата и время.
|
||||
:return: Ответ API.
|
||||
"""
|
||||
if action not in ["check-in", "check-out"]:
|
||||
raise ValueError("Invalid action. Must be 'check-in' or 'check-out'.")
|
||||
|
||||
url = f"{self.BASE_URL}/bookings/{booking_number}/room-stays/{room_stay_id}/{action}"
|
||||
payload = {
|
||||
f"actual{action.capitalize()}DateTime": actual_date_time
|
||||
}
|
||||
|
||||
try:
|
||||
response = requests.post(url, headers=self._get_headers(), json=payload)
|
||||
response.raise_for_status()
|
||||
self.logger.info(f"Успешно выполнено действие '{action}' для бронирования {booking_number}.")
|
||||
return response.json()
|
||||
except requests.RequestException as e:
|
||||
self.logger.error(f"Ошибка при выполнении действия '{action}': {e}")
|
||||
return {}
|
||||
|
||||
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.info(f"Плагин {self.__class__.__name__} успешно прошел валидацию.")
|
||||
return True
|
||||
Reference in New Issue
Block a user