SmartSolTech/Touchh
1
0
Fork 0
Code Issues Pull Requests 2 Actions Packages Projects Releases Wiki Activity

new

Browse Source
This commit is contained in:
trevor
2024-12-27 09:54:58 +09:00
parent 32f142727e
commit 9bff204fb4
5 changed files with 156 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bot/operations/hotels.py
View File

@@ -151,7 +151,7 @@ async def check_pms(update, context):
if hasattr(pms_manager.plugin, 'fetch_data') and callable(pms_manager.plugin.fetch_data): if hasattr(pms_manager.plugin, 'fetch_data') and callable(pms_manager.plugin.fetch_data):
# Плагин поддерживает метод fetch_data # Плагин поддерживает метод fetch_data
report = await pms_manager.plugin._fetch_data() report = await pms_manager.plugin._fetch_data()
logger.debug(f"REPORT: {report}, TYPE: {type(report)}") logger.debug(f"TYPE: {type(report)}")
else: else:
await query.edit_message_text("Подходящий способ интеграции с PMS не найден.") await query.edit_message_text("Подходящий способ интеграции с PMS не найден.")
return return

14
pms_integration/manager.py
View File

@@ -51,12 +51,18 @@ class PMSIntegrationManager:
def load_plugin(self): def load_plugin(self):
""" """
Загружает плагин для PMS на основе конфигурации. Загружает плагин для PMS на основе конфигурации отеля.
""" """
plugins = PluginLoader.load_plugins() plugins = PluginLoader.load_plugins()
if self.pms_config.plugin_name not in plugins: if not self.hotel:
raise ValueError(f"Плагин для PMS {self.pms_config.plugin_name} не найден.") raise ValueError("Отель не загружен. Пожалуйста, вызовите load_hotel перед загрузкой плагина.")
self.plugin = plugins[self.pms_config.plugin_name](self.pms_config) 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)
def fetch_data(self): def fetch_data(self):
""" """

5
pms_integration/models.py
View File

@@ -14,11 +14,6 @@ class PMSConfiguration(models.Model):
password = models.CharField(max_length=255, blank=True, null=True, verbose_name="Пароль") password = models.CharField(max_length=255, blank=True, null=True, verbose_name="Пароль")
plugin_name = models.CharField(max_length=255, blank=True, null=True, verbose_name="Плагин") plugin_name = models.CharField(max_length=255, blank=True, null=True, verbose_name="Плагин")
created_at = models.DateTimeField(auto_now_add=True, verbose_name="Дата создания") created_at = models.DateTimeField(auto_now_add=True, verbose_name="Дата создания")
hotels = models.ManyToManyField(
'hotels.Hotel',
related_name='pms_configurations',
verbose_name="Отели"
)
def __str__(self): def __str__(self):
return self.name return self.name

244
pms_integration/plugins/bnovo_pms.py
View File

@@ -4,20 +4,51 @@ from datetime import datetime, timedelta
from asgiref.sync import sync_to_async from asgiref.sync import sync_to_async
from .base_plugin import BasePMSPlugin from .base_plugin import BasePMSPlugin
from pms_integration.models import PMSConfiguration from pms_integration.models import PMSConfiguration
from hotels.models import Reservation, Hotel
from touchh.utils.log import CustomLogger from touchh.utils.log import CustomLogger
from pms_integration.models import PMSConfiguration import logging
from hotels.models import Hotel, Reservation
logger = CustomLogger(name="BnovoPMS Plugin", log_level="DEBUG").get_logger() import logging
# Настройка логирования
logging.basicConfig(
level=logging.WARNING, # Установите уровень логов для всех обработчиков
format='%(asctime)s [%(levelname)s] %(message)s',
handlers=[
logging.FileHandler("bnovo_plugin.log"), # Логи пишутся в файл
logging.StreamHandler() # Логи выводятся в консоль
]
)
# Создаем отдельный логгер для консоли с уровнем INFO
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.WARNING) # Меняем уровень логов для консоли
console_handler.setFormatter(logging.Formatter('%(asctime)s [%(levelname)s] %(message)s'))
# Основной логгер
logger = logging.getLogger("BnovoPMS Plugin")
logger.addHandler(console_handler)
logger.setLevel(logging.WARNING) # Основной уровень логов
class BnovoPMSPlugin(BasePMSPlugin): class BnovoPMSPlugin(BasePMSPlugin):
"""Плагин для работы с PMS Bnovo.""" """Плагин для работы с PMS Bnovo."""
def __init__(self, config): def __init__(self, hotel):
super().__init__(config) super().__init__(hotel)
self.api_url = config.url.rstrip("/")
self.username = config.username if not isinstance(hotel, Hotel):
self.password = config.password logger.error("Ожидался объект Hotel, но получен другой тип.")
raise ValueError("Для инициализации плагина требуется объект Hotel.")
self.hotel = hotel
self.pms_config = hotel.pms # Связь отеля с PMSConfiguration
if not self.pms_config:
logger.error(f"Отель {hotel.id} не связан с конфигурацией PMS.")
raise ValueError(f"Отель {hotel.id} не связан с конфигурацией PMS.")
self.api_url = self.pms_config.url.rstrip("/")
self.username = self.pms_config.username
self.password = self.pms_config.password
self.token = None self.token = None
if not self.api_url: if not self.api_url:
@@ -35,27 +66,26 @@ class BnovoPMSPlugin(BasePMSPlugin):
"timezone": "UTC" "timezone": "UTC"
} }
async def _get_stored_token(self, hotel_id): async def _get_stored_token(self):
"""Получение токена из базы данных для конкретного отеля.""" """Получение токена из конфигурации PMS отеля."""
try: try:
logger.debug(f"Попытка получения токена из базы данных для отеля: {hotel_id}.") logger.debug(f"Попытка получения токена для отеля {self.hotel.id}.")
config = await sync_to_async(PMSConfiguration.objects.get)(plugin_name="bnovo", hotel_id=hotel_id) token = self.pms_config.token
logger.debug(f"Токен из базы данных: {config.token}") logger.debug(f"Токен из базы данных: {token}")
return config.token return token
except PMSConfiguration.DoesNotExist: except Exception as e:
logger.warning(f"Токен отсутствует в базе данных для отеля: {hotel_id}.") logger.warning(f"Ошибка при получении токена для отеля {self.hotel.id}: {e}")
return None return None
async def _save_token_to_db(self, sid, hotel_id): async def _save_token_to_db(self, sid):
"""Сохраняет токен (SID) в базу данных для конкретного отеля.""" """Сохраняет токен (SID) в конфигурации PMS отеля."""
try: try:
logger.debug(f"Сохранение токена в базу данных для отеля {hotel_id}: {sid}") logger.debug(f"Сохранение токена для отеля {self.hotel.id}: {sid}")
await sync_to_async(PMSConfiguration.objects.update_or_create)( self.pms_config.token = sid
plugin_name="bnovo", hotel_id=hotel_id, defaults={"token": sid} await sync_to_async(self.pms_config.save)()
)
logger.debug("Токен успешно сохранен.") logger.debug("Токен успешно сохранен.")
except Exception as e: except Exception as e:
logger.error(f"Ошибка сохранения токена в базу данных для отеля {hotel_id}: {e}") logger.error(f"Ошибка сохранения токена для отеля {self.hotel.id}: {e}")
def _get_auth_headers(self): def _get_auth_headers(self):
"""Создает заголовки авторизации.""" """Создает заголовки авторизации."""
@@ -69,8 +99,8 @@ class BnovoPMSPlugin(BasePMSPlugin):
logger.debug(f"Добавлен токен в заголовки: {self.token}") logger.debug(f"Добавлен токен в заголовки: {self.token}")
return headers return headers
async def _fetch_session(self, hotel_id): async def _fetch_session(self):
"""Получение нового токена (SID) через запрос для конкретного отеля.""" """Получение нового токена (SID) через запрос."""
url = f"{self.api_url}/" url = f"{self.api_url}/"
payload = {"username": self.username, "password": self.password} payload = {"username": self.username, "password": self.password}
headers = self._get_auth_headers() headers = self._get_auth_headers()
@@ -85,7 +115,7 @@ class BnovoPMSPlugin(BasePMSPlugin):
if sid: if sid:
self.token = sid self.token = sid
logger.debug(f"Получен новый SID: {sid}") logger.debug(f"Получен новый SID: {sid}")
await self._save_token_to_db(sid, hotel_id) await self._save_token_to_db(sid)
else: else:
logger.error("Не удалось извлечь SID из ответа.") logger.error("Не удалось извлечь SID из ответа.")
raise ValueError("Не удалось извлечь SID из ответа.") raise ValueError("Не удалось извлечь SID из ответа.")
@@ -93,14 +123,14 @@ class BnovoPMSPlugin(BasePMSPlugin):
logger.error(f"Ошибка авторизации: {response.status_code}, {response.text}") logger.error(f"Ошибка авторизации: {response.status_code}, {response.text}")
raise ValueError(f"Ошибка авторизации: {response.status_code}, {response.text}") raise ValueError(f"Ошибка авторизации: {response.status_code}, {response.text}")
async def _fetch_account_data(self, hotel_id): async def _fetch_account_data(self):
"""Получение данных аккаунта через эндпоинт /account/current для конкретного отеля.""" """Получение данных аккаунта через эндпоинт /account/current."""
logger.info(f"Начало получения данных аккаунта для отеля: {hotel_id}.") logger.info(f"Начало получения данных аккаунта для отеля {self.hotel.id}.")
self.token = await self._get_stored_token(hotel_id) self.token = await self._get_stored_token()
if not self.token: if not self.token:
logger.info("Токен отсутствует, выполняем авторизацию.") logger.info("Токен отсутствует, выполняем авторизацию.")
await self._fetch_session(hotel_id) await self._fetch_session()
url = f"{self.api_url}/account/current" url = f"{self.api_url}/account/current"
headers = self._get_auth_headers() headers = self._get_auth_headers()
@@ -114,29 +144,29 @@ class BnovoPMSPlugin(BasePMSPlugin):
try: try:
account_data = response.json() account_data = response.json()
logger.debug(f"Полученные данные аккаунта: {json.dumps(account_data, indent=2)}") logger.debug(f"Полученные данные аккаунта:")
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
logger.error(f"Ошибка декодирования JSON: {e}") logger.error(f"Ошибка декодирования JSON: {e}")
raise ValueError(f"Ошибка декодирования JSON: {e}") raise ValueError(f"Ошибка декодирования JSON: {e}")
return account_data return account_data
async def fetch_and_log_account_data(self, hotel_id): async def _fetch_and_log_account_data(self):
"""Вызов метода _fetch_account_data и вывод результата в лог.""" """Вызов метода _fetch_account_data и вывод результата в лог."""
logger.info(f"Запуск получения и логирования данных аккаунта для отеля: {hotel_id}.") logger.info(f"Запуск получения и логирования данных аккаунта для отеля {self.hotel.id}.")
try: try:
account_data = await self._fetch_account_data(hotel_id) account_data = await self._fetch_account_data()
logger.info(f"Успешно полученные данные аккаунта: {json.dumps(account_data, indent=2)}") logger.info(f"Успешно полученные данные аккаунта:")
return account_data return account_data
except Exception as e: except Exception as e:
logger.error(f"Ошибка при получении данных аккаунта: {e}") logger.error(f"Ошибка при получении данных аккаунта: {e}")
raise raise
async def fetch_data_with_account_info(self, hotel_id): async def _fetch_data_with_account_info(self):
"""Получение данных аккаунта и бронирований для конкретного отеля.""" """Получение данных аккаунта и бронирований."""
logger.info(f"Запуск процесса получения данных аккаунта и бронирований для отеля: {hotel_id}.") logger.info(f"Запуск процесса получения данных аккаунта и бронирований для отеля {self.hotel.id}.")
try: try:
account_data = await self.fetch_and_log_account_data(hotel_id) account_data = await self.fetch_and_log_account_data()
logger.info("Данные аккаунта успешно получены, продолжение с бронированиями.") logger.info("Данные аккаунта успешно получены, продолжение с бронированиями.")
await self.__fetch_data() await self.__fetch_data()
except Exception as e: except Exception as e:
@@ -144,21 +174,19 @@ class BnovoPMSPlugin(BasePMSPlugin):
async def _fetch_data(self): async def _fetch_data(self):
"""Получение данных о бронированиях с помощью эндпоинта /dashboard.""" """Получение данных о бронированиях с помощью эндпоинта /dashboard."""
# Проверяем наличие токена в базе данных logger.info("Начало процесса получения данных о бронированиях.")
logger.info("Начало процесса получения данных.")
self.token = await self._get_stored_token()
if not self.token: # # Вызов функции получения данных аккаунта
logger.info("Токен отсутствует, выполняем авторизацию.") # try:
await self._fetch_session() # account_data = await self._fetch_and_log_account_data()
# logger.info(f"Данные аккаунта успешно получены:")
# except Exception as e:
accounts = await self._fetch_account_data() # logger.error(f"Ошибка получения данных аккаунта: {e}")
print(f'\n------\nACCOUNTS: {accounts}\n-------\n') # raise
url = f"{self.api_url}/dashboard" url = f"{self.api_url}/dashboard"
now = datetime.now() now = datetime.now()
create_from = (now - timedelta(days=90)).strftime("%d.%m.%Y") create_from = (now - timedelta(days=1)).strftime("%d.%m.%Y")
create_to = now.strftime("%d.%m.%Y") create_to = now.strftime("%d.%m.%Y")
params = { params = {
@@ -171,69 +199,83 @@ class BnovoPMSPlugin(BasePMSPlugin):
} }
headers = self._get_auth_headers() headers = self._get_auth_headers()
logger.debug(f"Начальный запрос к {url} с параметрами: {json.dumps(params, indent=2)}")
all_bookings = [] all_bookings = []
while True: while True:
logger.debug(f"Выполнение запроса к {url}") logger.debug(f"Запрос к {url} с параметрами: {json.dumps(params, indent=2)}")
response = requests.get(url, headers=headers, params=params)
logger.debug(f"Ответ запроса: статус {response.status_code}, тело {response.text}")
if response.status_code != 200:
logger.error(f"Ошибка при запросе: {response.status_code}, {response.text}")
raise ValueError("Ошибка запроса к /dashboard")
try: try:
response = requests.get(url, headers=headers, params=params, allow_redirects=False)
if response.status_code == 302:
logger.warning("Получен код 302. Перенаправление.")
redirected_url = response.headers.get("Location")
if redirected_url:
logger.debug(f"Перенаправление на {redirected_url}")
url = redirected_url
continue
else:
logger.error("Ответ с кодом 302 не содержит заголовка Location.")
raise ValueError("Перенаправление без указанного URL.")
if response.status_code != 200:
logger.error(f"Ошибка при запросе: {response.status_code}, {response.text}")
raise ValueError("Ошибка запроса к /dashboard")
data = response.json() data = response.json()
logger.debug(f"Полученные данные: {json.dumps(data, indent=2)}") bookings = data.get("bookings", [])
rooms = data.get("rooms", [])
logger.debug(f'bookings: {bookings}\n rooms: {rooms}')
all_bookings.extend(bookings)
logger.info(f"Получено бронирований: {len(bookings)}. Всего: {len(all_bookings)}.")
pages_info = data.get("pages", {})
current_page = pages_info.get("current_page", 1)
total_pages = pages_info.get("total_pages", 1)
logger.debug(f"Информация о страницах: текущая {current_page}, всего {total_pages}")
if current_page >= total_pages:
break
params["page"] += 1
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
logger.error(f"Ошибка декодирования JSON: {e}") logger.error(f"Ошибка декодирования JSON: {e}. Ответ: {response.text}")
raise ValueError(f"Ошибка декодирования JSON: {e}") raise ValueError(f"Ошибка декодирования JSON: {e}")
bookings = data.get("bookings", []) except Exception as e:
all_bookings.extend(bookings) logger.error(f"Неизвестная ошибка при обработке запроса: {e}")
raise
logger.info(f"Получено бронирований за запрос: {len(bookings)}. Всего: {len(all_bookings)}.")
# Проверяем наличие следующей страницы
pages_info = data.get("pages", {})
current_page = pages_info.get("current_page", 1)
total_pages = pages_info.get("total_pages", 1)
logger.debug(f"Информация о страницах: текущая {current_page}, всего {total_pages}")
if current_page >= total_pages:
break
params["page"] += 1
# Сопоставляем бронирования с существующими записями # Сопоставляем бронирования с существующими записями
for booking in all_bookings: for booking in all_bookings:
booking_id = booking.get("id") try:
hotel_id = booking.get("hotel_id") booking_id = booking.get("id")
if not booking_id or not hotel_id: hotel_id = booking.get("hotel_id")
logger.warning("У бронирования отсутствует id или hotel_id. Пропуск.")
continue
if hotel_id != str(self.config.hotel.external_id_pms): if hotel_id != str(self.hotel.external_id_pms):
logger.debug(f"Бронирование {booking_id} не относится к отелю {self.config.hotel.external_id_pms}. Пропуск.") logger.debug(f"Бронирование {booking_id} не относится к отелю {self.hotel.external_id_pms}. Пропуск.")
continue continue
reservation, created = await sync_to_async(Reservation.objects.update_or_create)( reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
external_id=booking_id, external_id=booking_id,
defaults={ defaults = {
"hotel": self.config.hotel, "hotel": self.hotel, # Объект модели Hotel
"status": booking.get("status_name"), "status": booking.get("status_name"), # Статус бронирования
"create_date": booking.get("create_date"), "room_number": booking.get("current_room"), # Номер комнаты (исправлено с create_date)
"arrival": booking.get("arrival"), "check_in": booking.get("arrival"), # Дата заезда
"departure": booking.get("departure"), "check_out": booking.get("departure"), # Дата выезда
"room_type": booking.get("initial_room_type_name"), "room_type": booking.get("initial_room_type_name") # Тип комнаты
"data": booking }
} )
) if created:
logger.info(f"Создана новая запись бронирования: {reservation}")
else:
logger.info(f"Обновлено существующее бронирование: {reservation}")
if created: except Exception as e:
logger.info(f"Создана новая запись бронирования: {reservation}") logger.error(f"Ошибка обработки бронирования {booking.get('id')}: {e}")
else:
logger.info(f"Обновлено существующее бронирование: {reservation}")
logger.info(f"Все бронирования получены и обработаны. Итоговое количество: {len(all_bookings)}") logger.info(f"Все бронирования получены и обработаны. Итоговое количество: {len(all_bookings)}")
return all_bookings return all_bookings