sMerge branch 'PMSManager_refactor'

antifroud/migrations/0001_initial.py

@@ -0,0 +1,129 @@
+# Generated by Django 5.1.4 on 2024-12-25 04:55
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ExternalDBSettings',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(help_text='Имя подключения для идентификации.', max_length=255, unique=True)),
+                ('host', models.CharField(help_text='Адрес сервера базы данных.', max_length=255)),
+                ('port', models.PositiveIntegerField(default=3306, help_text='Порт сервера базы данных.')),
+                ('user', models.CharField(help_text='Имя пользователя базы данных.', max_length=255)),
+                ('password', models.CharField(help_text='Пароль для подключения.', max_length=255)),
+                ('database', models.CharField(default='u1510415_wp832', help_text='Имя базы данных.', max_length=255)),
+                ('table_name', models.CharField(blank=True, default='wpts_user_activity_log', help_text='Имя таблицы для загрузки данных.', max_length=255, null=True)),
+                ('selected_fields', models.TextField(blank=True, help_text='Список полей для загрузки (через запятую).', null=True)),
+                ('is_active', models.BooleanField(default=True, help_text='Флаг активности подключения.')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+            ],
+            options={
+                'verbose_name': 'Настройка подключения к БД',
+                'verbose_name_plural': 'Настройки подключений к БД',
+            },
+        ),
+        migrations.CreateModel(
+            name='ImportedHotel',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('external_id', models.CharField(max_length=255, verbose_name='Внешний ID отеля')),
+                ('name', models.CharField(max_length=255, verbose_name='Имя отеля')),
+                ('display_name', models.CharField(blank=True, max_length=255, null=True, verbose_name='Отображаемое имя')),
+                ('created', models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')),
+                ('updated', models.DateTimeField(auto_now=True, verbose_name='Дата обновления')),
+                ('imported', models.BooleanField(default=False, verbose_name='Импортирован в основную базу')),
+            ],
+            options={
+                'verbose_name': 'Импортированный отель',
+                'verbose_name_plural': 'Импортированные отели',
+            },
+        ),
+        migrations.CreateModel(
+            name='RoomDiscrepancy',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('room_number', models.CharField(max_length=50, verbose_name='Номер комнаты')),
+                ('booking_id', models.CharField(max_length=255, verbose_name='ID бронирования')),
+                ('check_in_date_expected', models.DateField(verbose_name='Ожидаемая дата заселения')),
+                ('check_in_date_actual', models.DateField(verbose_name='Фактическая дата заселения')),
+                ('discrepancy_type', models.CharField(choices=[('early', 'Раннее заселение'), ('late', 'Позднее заселение'), ('missed', 'Неявка')], max_length=50, verbose_name='Тип несоответствия')),
+                ('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')),
+            ],
+            options={
+                'verbose_name': 'Несовпадение в заселении',
+                'verbose_name_plural': 'Несовпадения в заселении',
+            },
+        ),
+        migrations.CreateModel(
+            name='SyncLog',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created', models.DateTimeField(auto_now=True, verbose_name='Дата обновления')),
+                ('recieved_records', models.IntegerField(default=0, verbose_name='Полученные записи')),
+                ('processed_records', models.IntegerField(default=0, verbose_name='Обработанные записи')),
+            ],
+            options={
+                'verbose_name': 'Журнал синхронизации',
+                'verbose_name_plural': 'Журналы синхронизации',
+            },
+        ),
+        migrations.CreateModel(
+            name='UserActivityLog',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('external_id', models.CharField(db_index=True, max_length=255, unique=True, verbose_name='Внешний ID')),
+                ('user_id', models.BigIntegerField(blank=True, db_index=True, null=True, verbose_name='ID пользователя')),
+                ('ip', models.GenericIPAddressField(blank=True, db_index=True, null=True, verbose_name='IP-адрес')),
+                ('created', models.DateTimeField(blank=True, db_index=True, null=True, verbose_name='Дата создания')),
+                ('timestamp', models.BigIntegerField(blank=True, null=True, verbose_name='Метка времени')),
+                ('date_time', models.DateTimeField(blank=True, null=True, verbose_name='Дата и время')),
+                ('referred', models.TextField(blank=True, null=True, verbose_name='Реферальная ссылка')),
+                ('agent', models.TextField(blank=True, null=True, verbose_name='Агент пользователя')),
+                ('platform', models.CharField(blank=True, max_length=255, null=True, verbose_name='Платформа')),
+                ('version', models.CharField(blank=True, max_length=255, null=True, verbose_name='Версия')),
+                ('model', models.CharField(blank=True, max_length=255, null=True, verbose_name='Модель устройства')),
+                ('device', models.CharField(blank=True, max_length=255, null=True, verbose_name='Тип устройства')),
+                ('UAString', models.TextField(blank=True, null=True, verbose_name='User-Agent строка')),
+                ('location', models.CharField(blank=True, max_length=255, null=True, verbose_name='Местоположение')),
+                ('page_id', models.BigIntegerField(blank=True, db_index=True, null=True, verbose_name='ID страницы')),
+                ('url_parameters', models.TextField(blank=True, null=True, verbose_name='Параметры URL')),
+                ('page_title', models.TextField(blank=True, null=True, verbose_name='Заголовок страницы')),
+                ('type', models.CharField(blank=True, max_length=50, null=True, verbose_name='Тип')),
+                ('last_counter', models.IntegerField(blank=True, null=True, verbose_name='Последний счетчик')),
+                ('hits', models.IntegerField(blank=True, default='0', null=True, verbose_name='Количество обращений')),
+                ('honeypot', models.BooleanField(blank=True, null=True, verbose_name='Метка honeypot')),
+                ('reply', models.BooleanField(blank=True, null=True, verbose_name='Ответ пользователя')),
+                ('page_url', models.URLField(blank=True, null=True, verbose_name='URL страницы')),
+            ],
+            options={
+                'verbose_name': 'Регистрация посетителей',
+                'verbose_name_plural': 'Регистрации посетителей',
+            },
+        ),
+        migrations.CreateModel(
+            name='ViolationLog',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('room_number', models.CharField(blank=True, max_length=50, null=True, verbose_name='Номер комнаты')),
+                ('violation_type', models.CharField(choices=[('missed', 'Неявка'), ('early', 'Раннее заселение'), ('late', 'Позднее заселение')], max_length=50, verbose_name='Тип нарушения')),
+                ('violation_details', models.TextField(blank=True, null=True, verbose_name='Детали нарушения')),
+                ('hits', models.IntegerField(verbose_name='Срабатывания')),
+                ('detected_at', models.DateTimeField(auto_now_add=True, verbose_name='Дата обнаружения')),
+                ('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')),
+            ],
+            options={
+                'verbose_name': 'Журнал нарушений',
+                'verbose_name_plural': 'Журналы нарушений',
+            },
+        ),
+    ]

antifroud/migrations/0002_initial.py

@@ -1,4 +1,8 @@
+<<<<<<< HEAD
 # Generated by Django 5.1.4 on 2024-12-19 12:42
+=======
+# Generated by Django 5.1.4 on 2024-12-25 04:55
+>>>>>>> PMSManager_refactor
 
 import django.db.models.deletion
 from django.db import migrations, models

bot/operations/hotels.py

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

hotels/migrations/0002_initial.py

@@ -0,0 +1,47 @@
+# Generated by Django 5.1.4 on 2024-12-25 04:55
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('hotels', '0001_initial'),
+        ('pms_integration', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='hotel',
+            name='pms',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='pms_integration.pmsconfiguration', verbose_name='PMS система'),
+        ),
+        migrations.AddField(
+            model_name='fraudlog',
+            name='hotel',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='frauds', to='hotels.hotel'),
+        ),
+        migrations.AddField(
+            model_name='reservation',
+            name='hotel',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='hotels.hotel', verbose_name='Отель'),
+        ),
+        migrations.AddField(
+            model_name='guest',
+            name='reservation',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='guests', to='hotels.reservation', verbose_name='Бронирование'),
+        ),
+        migrations.AddField(
+            model_name='room',
+            name='hotel',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='rooms', to='hotels.hotel', verbose_name='Отель'),
+        ),
+        migrations.AddField(
+            model_name='userhotel',
+            name='hotel',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='hotel_users', to='hotels.hotel', verbose_name='Отель'),
+        ),
+    ]

hotels/migrations/0003_rename_external_id_hotel_external_id_pms_and_more.py

@@ -0,0 +1,54 @@
+# Generated by Django 5.1.4 on 2024-12-25 05:10
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('hotels', '0002_initial'),
+        ('users', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name='hotel',
+            old_name='external_id',
+            new_name='external_id_pms',
+        ),
+        migrations.AddField(
+            model_name='userhotel',
+            name='user',
+            field=models.ForeignKey(default=1, on_delete=django.db.models.deletion.CASCADE, related_name='user_hotels', to='users.user', verbose_name='Пользователь'),
+            preserve_default=False,
+        ),
+        migrations.AddIndex(
+            model_name='apirequestlog',
+            index=models.Index(fields=['api'], name='hotels_apir_api_id_686bb0_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='apirequestlog',
+            index=models.Index(fields=['request_time'], name='hotels_apir_request_f65147_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='fraudlog',
+            index=models.Index(fields=['reservation_id'], name='hotels_frau_reserva_5a26b7_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='fraudlog',
+            index=models.Index(fields=['detected_at'], name='hotels_frau_detecte_07e626_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='reservation',
+            index=models.Index(fields=['hotel', 'check_in', 'check_out'], name='hotels_rese_hotel_i_6c527e_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='room',
+            index=models.Index(fields=['hotel', 'number'], name='hotels_room_hotel_i_a7c4fc_idx'),
+        ),
+        migrations.AddConstraint(
+            model_name='room',
+            constraint=models.UniqueConstraint(fields=('hotel', 'number'), name='unique_hotel_room'),
+        ),
+    ]

hotels/models.py

@@ -23,6 +23,7 @@ class APIConfiguration(models.Model):
 class Hotel(models.Model):
     name = models.CharField(max_length=255, verbose_name="Название отеля")
     hotel_id = models.CharField(max_length=255, unique=True, null=True, blank=True, verbose_name="ID отеля")
+    external_id_pms = models.CharField(max_length=100, unique=False, null=True, blank=True, verbose_name="Внешний PMS ID")
     created_at = models.DateTimeField(auto_now_add=True, verbose_name="Создан")
     phone = models.CharField(
         max_length=50,

pms_integration/manager.py

@@ -51,12 +51,18 @@ class PMSIntegrationManager:
 
     def load_plugin(self):
         """
-        Загружает плагин для PMS на основе конфигурации.
+        Загружает плагин для PMS на основе конфигурации отеля.
         """
         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):
         """

pms_integration/migrations/0001_initial.py

@@ -0,0 +1,50 @@
+# Generated by Django 5.1.4 on 2024-12-25 04:55
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('hotels', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='PMSConfiguration',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('name', models.CharField(max_length=255, verbose_name='Название PMS')),
+                ('url', models.URLField(verbose_name='URL API')),
+                ('token', models.CharField(blank=True, max_length=255, null=True, verbose_name='Токен')),
+                ('public_key', models.CharField(blank=True, max_length=255, null=True, verbose_name='Публичный ключ')),
+                ('private_key', models.CharField(blank=True, max_length=255, null=True, verbose_name='Приватный ключ')),
+                ('username', models.CharField(blank=True, max_length=255, null=True, verbose_name='Логин')),
+                ('password', models.CharField(blank=True, max_length=255, null=True, verbose_name='Пароль')),
+                ('plugin_name', models.CharField(blank=True, max_length=255, null=True, verbose_name='Плагин')),
+                ('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Дата создания')),
+            ],
+            options={
+                'verbose_name': 'PMS система',
+                'verbose_name_plural': 'PMS системы',
+            },
+        ),
+        migrations.CreateModel(
+            name='PMSIntegrationLog',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('checked_at', models.DateTimeField(auto_now_add=True, verbose_name='Время проверки')),
+                ('status', models.CharField(choices=[('success', 'Успех'), ('error', 'Ошибка')], max_length=50, verbose_name='Статус')),
+                ('message', models.TextField(blank=True, null=True, verbose_name='Сообщение')),
+                ('hotel', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='hotels.hotel', verbose_name='Отель')),
+            ],
+            options={
+                'verbose_name': 'Журнал интеграции PMS',
+                'verbose_name_plural': 'Журналы интеграции PMS',
+                'indexes': [models.Index(fields=['hotel'], name='pms_integra_hotel_i_ade4da_idx'), models.Index(fields=['checked_at'], name='pms_integra_checked_938acc_idx'), models.Index(fields=['status'], name='pms_integra_status_358b64_idx')],
+            },
+        ),
+    ]

pms_integration/migrations/0002_pmsconfiguration_hotels.py

@@ -0,0 +1,19 @@
+# Generated by Django 5.1.4 on 2024-12-25 06:48
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('hotels', '0003_rename_external_id_hotel_external_id_pms_and_more'),
+        ('pms_integration', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='pmsconfiguration',
+            name='hotels',
+            field=models.ManyToManyField(related_name='pms_configurations', to='hotels.hotel', verbose_name='Отели'),
+        ),
+    ]

pms_integration/migrations/0003_remove_pmsconfiguration_hotels.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.1.4 on 2024-12-25 07:00
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('pms_integration', '0002_pmsconfiguration_hotels'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='pmsconfiguration',
+            name='hotels',
+        ),
+    ]

pms_integration/models.py

@@ -15,7 +15,6 @@ class PMSConfiguration(models.Model):
     plugin_name = models.CharField(max_length=255, blank=True, null=True, verbose_name="Плагин")
     created_at = models.DateTimeField(auto_now_add=True, verbose_name="Дата создания")
 
-
     def __str__(self):
         return self.name
 

pms_integration/plugins/bnovo_pms.py

@@ -1,136 +1,281 @@
 import requests
 import json
 from datetime import datetime, timedelta
-from .base_plugin import BasePMSPlugin
 from asgiref.sync import sync_to_async
+from .base_plugin import BasePMSPlugin
 from pms_integration.models import PMSConfiguration
+from hotels.models import Reservation, Hotel
+from touchh.utils.log import CustomLogger
+import logging
 
+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):
     """Плагин для работы с PMS Bnovo."""
 
-    def __init__(self, config):
+    def __init__(self, hotel):
-        super().__init__(config)
+        super().__init__(hotel)
-        self.api_url = config.url.rstrip("/")  # Убираем лишний `/` в конце URL
+
-        self.username = config.username
+        if not isinstance(hotel, Hotel):
-        self.password = config.password
+            logger.error("Ожидался объект Hotel, но получен другой тип.")
-        self.token = None  # SID
+            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
 
         if not self.api_url:
+            logger.error("Не указан URL для работы плагина.")
             raise ValueError("Не указан URL для работы плагина.")
         if not self.username or not self.password:
+            logger.error("Не указаны логин или пароль для авторизации.")
             raise ValueError("Не указаны логин или пароль для авторизации.")
 
     def get_default_parser_settings(self):
         """Возвращает настройки по умолчанию для обработки данных."""
+        logger.debug("Получение настроек парсера по умолчанию.")
         return {
             "date_format": "%Y-%m-%dT%H:%M:%S",
             "timezone": "UTC"
         }
 
-    async def _save_token_to_db(self, sid):
+    async def _get_stored_token(self):
-        """Сохраняет токен (SID) в базу данных."""
+        """Получение токена из конфигурации PMS отеля."""
         try:
-            await sync_to_async(PMSConfiguration.objects.update_or_create)(
+            logger.debug(f"Попытка получения токена для отеля {self.hotel.id}.")
-                plugin_name="bnovo",
+            token = self.pms_config.token
-                defaults={"token": sid}
+            logger.debug(f"Токен из базы данных: {token}")
-            )
+            return token
-            print(f"[DEBUG] Токен сохранен в БД: {sid}")
         except Exception as e:
-            print(f"[ERROR] Ошибка сохранения токена в БД: {e}")
+            logger.warning(f"Ошибка при получении токена для отеля {self.hotel.id}: {e}")
+            return None
+
+    async def _save_token_to_db(self, sid):
+        """Сохраняет токен (SID) в конфигурации PMS отеля."""
+        try:
+            logger.debug(f"Сохранение токена для отеля {self.hotel.id}: {sid}")
+            self.pms_config.token = sid
+            await sync_to_async(self.pms_config.save)()
+            logger.debug("Токен успешно сохранен.")
+        except Exception as e:
+            logger.error(f"Ошибка сохранения токена для отеля {self.hotel.id}: {e}")
 
     def _get_auth_headers(self):
         """Создает заголовки авторизации."""
+        logger.debug("Создание заголовков авторизации.")
         headers = {
             "Content-Type": "application/json",
             "Accept": "application/json",
         }
         if self.token:
             headers["Cookie"] = f"SID={self.token}"
+            logger.debug(f"Добавлен токен в заголовки: {self.token}")
         return headers
 
     async def _fetch_session(self):
-        """Получает идентификатор сессии (SID) через запрос."""
+        """Получение нового токена (SID) через запрос."""
         url = f"{self.api_url}/"
-        payload = {
+        payload = {"username": self.username, "password": self.password}
-            "username": self.username,
-            "password": self.password,
-        }
-
-        print(f"[DEBUG] URL авторизации: {url}")
-        print(f"[DEBUG] Тело запроса: {json.dumps(payload, indent=2)}")
         headers = self._get_auth_headers()
 
-        session = requests.Session()
+        logger.debug(f"Авторизация по адресу: {url} с данными: {json.dumps(payload)}")
-        response = session.post(url, json=payload, headers=headers, allow_redirects=False)
+        response = requests.post(url, json=payload, headers=headers, allow_redirects=False)
 
-        print(f"[DEBUG] Статус ответа: {response.status_code}")
+        logger.debug(f"Ответ авторизации: статус {response.status_code}, заголовки {response.headers}")
-        print(f"[DEBUG] Ответ заголовков: {response.headers}")
+        if response.status_code == 302:
-        print(f"[DEBUG] Cookies: {session.cookies}")
+            cookies = response.cookies.get_dict()
-
+            sid = cookies.get("SID")
-        if response.status_code == 302 and "SID" in session.cookies:
+            if sid:
-            sid = session.cookies.get("SID")
+                self.token = sid
-            self.token = sid
+                logger.debug(f"Получен новый SID: {sid}")
-            print(f"[DEBUG] Получен SID: {sid}")
-
-            # Правильное сохранение в БД через sync_to_async
-            try:
                 await self._save_token_to_db(sid)
-                print(f"[DEBUG] Токен сохранен в БД")
+            else:
-            except Exception as e:
+                logger.error("Не удалось извлечь SID из ответа.")
-                print(f"[ERROR] Ошибка сохранения токена в БД: {e}")
+                raise ValueError("Не удалось извлечь SID из ответа.")
         else:
-            raise ValueError(f"Не удалось получить SID из ответа: {response.text}")
+            logger.error(f"Ошибка авторизации: {response.status_code}, {response.text}")
+            raise ValueError(f"Ошибка авторизации: {response.status_code}, {response.text}")
+
+    async def _fetch_account_data(self):
+        """Получение данных аккаунта через эндпоинт /account/current."""
+        logger.info(f"Начало получения данных аккаунта для отеля {self.hotel.id}.")
+        self.token = await self._get_stored_token()
+
+        if not self.token:
+            logger.info("Токен отсутствует, выполняем авторизацию.")
+            await self._fetch_session()
+
+        url = f"{self.api_url}/account/current"
+        headers = self._get_auth_headers()
+
+        logger.debug(f"Выполнение запроса к {url}")
+        response = requests.get(url, headers=headers)
+
+        if response.status_code != 200:
+            logger.error(f"Ошибка при запросе данных аккаунта: {response.status_code}, {response.text}")
+            raise ValueError("Ошибка запроса к /account/current")
+
+        try:
+            account_data = response.json()
+            logger.debug(f"Полученные данные аккаунта:")
+        except json.JSONDecodeError as e:
+            logger.error(f"Ошибка декодирования JSON: {e}")
+            raise ValueError(f"Ошибка декодирования JSON: {e}")
+
+        return account_data
+
+    async def _fetch_and_log_account_data(self):
+        """Вызов метода _fetch_account_data и вывод результата в лог."""
+        logger.info(f"Запуск получения и логирования данных аккаунта для отеля {self.hotel.id}.")
+        try:
+            account_data = await self._fetch_account_data()
+            logger.info(f"Успешно полученные данные аккаунта:")
+            return account_data
+        except Exception as e:
+            logger.error(f"Ошибка при получении данных аккаунта: {e}")
+            raise
+
+    async def _fetch_data_with_account_info(self):
+        """Получение данных аккаунта и бронирований."""
+        logger.info(f"Запуск процесса получения данных аккаунта и бронирований для отеля {self.hotel.id}.")
+        try:
+            account_data = await self.fetch_and_log_account_data()
+            logger.info("Данные аккаунта успешно получены, продолжение с бронированиями.")
+            await self.__fetch_data()
+        except Exception as e:
+            logger.error(f"Ошибка при выполнении полной операции: {e}")
 
     async def _fetch_data(self):
-        """Получает данные о бронированиях с помощью эндпоинта `/dashboard`."""
+        """Получение данных о бронированиях с помощью эндпоинта /dashboard."""
-        await self._fetch_session()  # Авторизуемся перед каждым запросом
+        logger.info("Начало процесса получения данных о бронированиях.")
 
+        # # Вызов функции получения данных аккаунта
+        # try:
+        #     account_data = await self._fetch_and_log_account_data()
+        #     logger.info(f"Данные аккаунта успешно получены:")
+        # except Exception as e:
+        #     logger.error(f"Ошибка получения данных аккаунта: {e}")
+        #     raise
+
+        url = f"{self.api_url}/dashboard"
         now = datetime.now()
-        create_from = (now - timedelta(days=90)).strftime("%d.%m.%Y")  # Диапазон: последние 90 дней
+        create_from = (now - timedelta(days=1)).strftime("%d.%m.%Y")
         create_to = now.strftime("%d.%m.%Y")
 
         params = {
             "create_from": create_from,
             "create_to": create_to,
-            "status_ids": "1",
+            "advanced_search": 2,
-            "advanced_search": 2,  # Обязательный параметр
+            "c": 100,
-            "c": 100,  # Количество элементов на странице (максимум 100)
+            "page": 1,
-            "page": 1,  # Начальная страница
+            "order_by": "create_date.asc",
-            "order_by": "create_date.asc",  # Сортировка по возрастанию даты создания
         }
-
         headers = self._get_auth_headers()
 
-        all_bookings = []  # Для сохранения всех бронирований
+        all_bookings = []
         while True:
-            print(f"[DEBUG] Запрос к /dashboard с параметрами: {json.dumps(params, indent=2)}")
+            logger.debug(f"Запрос к {url} с параметрами: {json.dumps(params, indent=2)}")
-            response = requests.get(f"{self.api_url}/dashboard", headers=headers, params=params)
+            try:
+                response = requests.get(url, headers=headers, params=params, allow_redirects=False)
 
-            print(f"[DEBUG] Статус ответа: {response.status_code}")
+                if response.status_code == 302:
-            if response.status_code != 200:
+                    logger.warning("Получен код 302. Перенаправление.")
-                raise ValueError(f"Ошибка при получении данных: {response.status_code}, {response.text}")
+                    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.")
 
-            data = response.json()
+                if response.status_code != 200:
-            print(json.dumps(data, indent=2))
+                    logger.error(f"Ошибка при запросе: {response.status_code}, {response.text}")
-            bookings = data.get("bookings", [])
+                    raise ValueError("Ошибка запроса к /dashboard")
-            all_bookings.extend(bookings)
 
-            print(f"[DEBUG] Получено бронирований: {len(bookings)}")
+                data = response.json()
-            print(f"[DEBUG] Всего бронирований: {len(all_bookings)}")
+                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)
 
-            if current_page >= total_pages:
+                pages_info = data.get("pages", {})
-                break  # Все страницы загружены
+                current_page = pages_info.get("current_page", 1)
+                total_pages = pages_info.get("total_pages", 1)
 
-            params["page"] += 1  # Переход на следующую страницу
+                logger.debug(f"Информация о страницах: текущая {current_page}, всего {total_pages}")
+                if current_page >= total_pages:
+                    break
 
-            if not all_bookings:
+                params["page"] += 1
-                print("[DEBUG] Нет бронирований за указанный период.")
+
-        else:
+            except json.JSONDecodeError as e:
-            print(f"[DEBUG] Полученные бронирования: {json.dumps(all_bookings, indent=2)}")
+                logger.error(f"Ошибка декодирования JSON: {e}. Ответ: {response.text}")
-        return all_bookings
+                raise ValueError(f"Ошибка декодирования JSON: {e}")
+
+            except Exception as e:
+                logger.error(f"Неизвестная ошибка при обработке запроса: {e}")
+                raise
+
+        # Сопоставляем бронирования с существующими записями
+        for booking in all_bookings:
+            try:
+                booking_id = booking.get("id")
+                hotel_id = booking.get("hotel_id")
+
+                if hotel_id != str(self.hotel.external_id_pms):
+                    logger.debug(f"Бронирование {booking_id} не относится к отелю {self.hotel.external_id_pms}. Пропуск.")
+                    continue
+
+                reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
+                    external_id=booking_id,
+                    defaults = {
+                        "hotel": self.hotel,  # Объект модели Hotel
+                        "status": booking.get("status_name"),  # Статус бронирования
+                        "room_number": booking.get("current_room"),  # Номер комнаты (исправлено с create_date)
+                        "check_in": booking.get("arrival"),  # Дата заезда
+                        "check_out": booking.get("departure"),  # Дата выезда
+                        "room_type": booking.get("initial_room_type_name")  # Тип комнаты
+                    }
+                )
+                if created:
+                    logger.info(f"Создана новая запись бронирования: {reservation}")
+                else:
+                    logger.info(f"Обновлено существующее бронирование: {reservation}")
+
+            except Exception as e:
+                logger.error(f"Ошибка обработки бронирования {booking.get('id')}: {e}")
+
+        logger.info(f"Все бронирования получены и обработаны. Итоговое количество: {len(all_bookings)}")
+        return all_bookings

pms_integration/plugins/ecvi_pms.py

@@ -12,7 +12,7 @@ class EcviPMSPlugin(BasePMSPlugin):
 
     def __init__(self, pms_config):
         super().__init__(pms_config)
-
+        
         # Инициализация логгера
         self.logger = logging.getLogger(self.__class__.__name__)  # Логгер с именем класса
         handler_console = logging.StreamHandler()  # Потоковый обработчик для вывода в консоль

pms_integration/plugins/realtycalendar_pms.py

@@ -1,194 +1,3 @@
-# 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
-
-
 import requests
 import hashlib
 import json

pms_integration/plugins/travelline_pms.py

@@ -16,7 +16,7 @@ class TravelLinePMSPlugin(BasePMSPlugin):
         :param pms_config: Конфигурация PMS (объект PMSConfiguration).
         """
         super().__init__(pms_config)
-        self.api_key = pms_config.api_key
+        self.api_key = pms_config.token
         self.logger = logging.getLogger(self.__class__.__name__)
 
     def _get_headers(self):

scheduler/migrations/0001_initial.py

@@ -0,0 +1,30 @@
+# Generated by Django 5.1.4 on 2024-12-25 04:55
+
+import scheduler.models
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ScheduledTask',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('task_name', models.CharField(max_length=255, verbose_name='Название задачи')),
+                ('function_path', models.CharField(choices=scheduler.models.get_available_functions, max_length=500, verbose_name='Путь к функции (модуль.функция)')),
+                ('minutes', models.CharField(default='*', max_length=255, verbose_name='Минуты')),
+                ('hours', models.CharField(default='*', max_length=255, verbose_name='Часы')),
+                ('days', models.CharField(default='*', max_length=255, verbose_name='Дни')),
+                ('months', models.CharField(default='*', max_length=255, verbose_name='Месяцы')),
+                ('weekdays', models.JSONField(default=list, verbose_name='Дни недели')),
+                ('active', models.BooleanField(default=True, verbose_name='Активно')),
+                ('last_run', models.DateTimeField(blank=True, null=True, verbose_name='Последний запуск')),
+            ],
+        ),
+    ]

users/migrations/0001_initial.py

@@ -0,0 +1,63 @@
+# Generated by Django 5.1.4 on 2024-12-25 04:55
+
+import django.contrib.auth.models
+import django.contrib.auth.validators
+import django.db.models.deletion
+import django.utils.timezone
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('auth', '0012_alter_user_first_name_max_length'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='User',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('password', models.CharField(max_length=128, verbose_name='password')),
+                ('last_login', models.DateTimeField(blank=True, null=True, verbose_name='last login')),
+                ('is_superuser', models.BooleanField(default=False, help_text='Designates that this user has all permissions without explicitly assigning them.', verbose_name='superuser status')),
+                ('username', models.CharField(error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username')),
+                ('first_name', models.CharField(blank=True, max_length=150, verbose_name='first name')),
+                ('last_name', models.CharField(blank=True, max_length=150, verbose_name='last name')),
+                ('email', models.EmailField(blank=True, max_length=254, verbose_name='email address')),
+                ('is_staff', models.BooleanField(default=False, help_text='Designates whether the user can log into this admin site.', verbose_name='staff status')),
+                ('is_active', models.BooleanField(default=True, help_text='Designates whether this user should be treated as active. Unselect this instead of deleting accounts.', verbose_name='active')),
+                ('date_joined', models.DateTimeField(default=django.utils.timezone.now, verbose_name='date joined')),
+                ('telegram_id', models.BigIntegerField(blank=True, null=True, unique=True, verbose_name='ID Телеграм')),
+                ('chat_id', models.BigIntegerField(blank=True, null=True, unique=True, verbose_name='ID чата в телеграм')),
+                ('role', models.CharField(choices=[('admin', 'Администратор системы'), ('hotel_user', 'Сотрудник отеля')], default='hotel_user', max_length=20, verbose_name='Роль')),
+                ('confirmed', models.BooleanField(default=False, verbose_name='Подтвержден')),
+                ('groups', models.ManyToManyField(blank=True, related_name='custom_user_set', to='auth.group')),
+                ('user_permissions', models.ManyToManyField(blank=True, related_name='custom_user_set', to='auth.permission')),
+            ],
+            options={
+                'verbose_name': 'Пользователь',
+                'verbose_name_plural': 'Пользователи',
+            },
+            managers=[
+                ('objects', django.contrib.auth.models.UserManager()),
+            ],
+        ),
+        migrations.CreateModel(
+            name='NotificationSettings',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('telegram_enabled', models.BooleanField(default=True, verbose_name='Уведомления в Telegram')),
+                ('email_enabled', models.BooleanField(default=False, verbose_name='Уведомления по Email')),
+                ('email', models.EmailField(blank=True, max_length=254, null=True, verbose_name='Email для уведомлений')),
+                ('notification_time', models.TimeField(default='09:00', verbose_name='Время отправки уведомлений')),
+                ('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, to='users.user', verbose_name='Пользователь')),
+            ],
+            options={
+                'verbose_name': 'Способ оповещения',
+                'verbose_name_plural': 'Способы оповещений',
+            },
+        ),
+    ]