import requests
import json
from datetime import datetime, timedelta
from asgiref.sync import sync_to_async
from .base_plugin import BasePMSPlugin
from hotels.models import Reservation
from hotels.models import Hotel


class Shelter(BasePMSPlugin):
    """
    Плагин для PMS Shelter Coud.
    """
    def __init__(self, config):
        super().__init__(config)
        self.token = config.token

    def get_default_parser_settings(self):
        """
        Возвращает настройки по умолчанию для обработки данных.
        """
        return {
            "date_format": "%Y-%m-%dT%H:%M:%S",
            "timezone": "UTC"
        }

    def _fetch_data(self):
        """
        Выполняет запрос к API PMS для получения данных.
        """
        url = 'https://pms.frontdesk24.ru/sheltercloudapi/Reservations/ByFilter'
        headers = {
            'accept': 'text/plain',
            'Authorization': f'Bearer {self.token}',
            'Content-Type': 'application/json',
        }

        from_index = 0
        count_per_request = 50
        total_count = None
        all_items = []
        now = datetime.now()
        start_date = (now - timedelta(days=60)).strftime('%Y-%m-%dT%H:%M:%SZ')
        end_date = (now + timedelta(days=60)).strftime('%Y-%m-%dT%H:%M:%SZ')

        while total_count is None or from_index < total_count:
            data = {
                "from": start_date,
                "until": end_date,
                "pagination": {
                    "from": from_index,
                    "count": count_per_request
                }
            }

            response = requests.post(url, headers=headers, data=json.dumps(data))
            if response.status_code == 200:
                response_data = response.json()
                items = response_data.get("items", [])
                all_items.extend(items)

                if total_count is None:
                    total_count = response_data.get("count", 0)

                from_index += len(items)
            else:
                raise ValueError(f'Shelter API Error: {response.status_code}')

        return all_items

    async def _save_to_db(self, data, hotel_id):
        """
        Сохраняет данные о бронированиях в таблицу Reservation.
        :param data: Список данных о бронированиях.
        :param hotel_id: ID отеля, к которому относятся бронирования.
        """
        hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
        for item in data:
            print(f"Данные для сохранения: {item}")

            try:
                reservation, created = await sync_to_async(Reservation.objects.update_or_create)(
                    reservation_id=item["id"],
                    hotel=hotel,
                    defaults={
                        "room_number": item.get("roomNumber", ""),  # Номер комнаты
                        "room_type": item.get("roomTypeName", ""),  # Тип комнаты
                        "check_in": datetime.strptime(item["from"], '%Y-%m-%dT%H:%M:%S'),  # Дата заезда
                        "check_out": datetime.strptime(item["until"], '%Y-%m-%dT%H:%M:%S'),  # Дата выезда
                        "status": item.get("checkInStatus", ""),  # Статус бронирования
                        "price": item.get("reservationPrice", 0),  # Цена
                        "discount": item.get("discount", 0),  # Скидка
                    }
                )
                if created:
                    print(f"Создана запись: {reservation}")
                else:
                    print(f"Обновлена запись: {reservation}")
            except Exception as e:
                print(f"Ошибка при сохранении бронирования ID {item['id']}: {e}")