meteocat 2.0.3 → 2.2.2

package/CHANGELOG.md

@@ -1,11 +1,29 @@
+## [2.2.2](https://github.com/figorr/meteocat/compare/v2.2.1...v2.2.2) (2025-02-04)
+
+
+### Bug Fixes
+
+* 2.2.2 ([3667e5d](https://github.com/figorr/meteocat/commit/3667e5d65069ee0079ebbaebeab6c929be8b9630))
+* fix version 2.2.1 ([359cc7f](https://github.com/figorr/meteocat/commit/359cc7f7a9f1f025890bedc5177785016e9968d1))
+
+# [2.1.0](https://github.com/figorr/meteocat/compare/v2.0.3...v2.1.0) (2025-01-27)
+
+
+### Bug Fixes
+
+* 2.1.0 ([759c293](https://github.com/figorr/meteocat/commit/759c2932fea8e4f174a753eca5b4c09f9d2549b6))
+
+
+### Features
+
+* fix version ([596ee59](https://github.com/figorr/meteocat/commit/596ee59a835ee913238f66d40869ce27a53da4e9))
+
 ## [2.0.3](https://github.com/figorr/meteocat/compare/v2.0.2...v2.0.3) (2025-01-27)
 
+* 2.0.3
 
 ### Bug Fixes
 
-* 2.0.2 ([efe8e9a](https://github.com/figorr/meteocat/commit/efe8e9aad133c7dff8fc7eabce7e5ab598b2e037))
-* 2.0.3 ([12a5470](https://github.com/figorr/meteocat/commit/12a5470994dbe95176a75bc05a530b33868b8141))
-* fix version 2.0.2 ([264c946](https://github.com/figorr/meteocat/commit/264c94659445d40601415eed22b7075f74d36823))
 
 ## [2.0.2](https://github.com/figorr/meteocat/compare/v2.0.1...v2.0.2) (2025-01-27)
 

package/custom_components/meteocat/__init__.py

@@ -22,6 +22,8 @@ from .coordinator import (
     MeteocatTempForecastCoordinator,
     MeteocatAlertsCoordinator,
     MeteocatAlertsRegionCoordinator,
+    MeteocatQuotesCoordinator,
+    MeteocatQuotesFileCoordinator,
 )
 
 from .const import DOMAIN, PLATFORMS
@@ -29,7 +31,7 @@ from .const import DOMAIN, PLATFORMS
 _LOGGER = logging.getLogger(__name__)
 
 # Versión
-__version__ = "2.0.3"
+__version__ = "2.2.2"
 
 # Definir el esquema de configuración CONFIG_SCHEMA
 CONFIG_SCHEMA = vol.Schema(
@@ -129,6 +131,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         alerts_region_coordinator = MeteocatAlertsRegionCoordinator(hass=hass, entry_data=entry_data)
         await alerts_region_coordinator.async_config_entry_first_refresh()
 
+        quotes_coordinator = MeteocatQuotesCoordinator(hass=hass, entry_data=entry_data)
+        await quotes_coordinator.async_config_entry_first_refresh()
+
+        quotes_file_coordinator = MeteocatQuotesFileCoordinator(hass=hass, entry_data=entry_data)
+        await quotes_file_coordinator.async_config_entry_first_refresh()
+
     except Exception as err:  # Capturar todos los errores
         _LOGGER.exception(f"Error al inicializar los coordinadores: {err}")
         return False
@@ -147,6 +155,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         "temp_forecast_coordinator": temp_forecast_coordinator,
         "alerts_coordinator": alerts_coordinator,
         "alerts_region_coordinator": alerts_region_coordinator,
+        "quotes_coordinator": quotes_coordinator,
+        "quotes_file_coordinator": quotes_file_coordinator,
         **entry_data,
     }
 
@@ -213,6 +223,9 @@ async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
     alerts_file = files_folder / "alerts.json"
     alerts_region_file = files_folder / f"alerts_{region_id}.json"
 
+    # Archivos JSON de cuotas
+    quotes_file = files_folder / f"quotes.json"
+
     # Validar la ruta base
     if not custom_components_path.exists():
         _LOGGER.warning(f"La ruta {custom_components_path} no existe. No se realizará la limpieza.")
@@ -226,6 +239,7 @@ async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
     safe_remove(forecast_hourly_data_file)
     safe_remove(forecast_daily_data_file)
     safe_remove(alerts_file)
+    safe_remove(quotes_file)
     safe_remove(alerts_region_file)
     safe_remove(assets_folder, is_folder=True)
     safe_remove(files_folder, is_folder=True)

package/custom_components/meteocat/config_flow.py

@@ -1,14 +1,18 @@
 from __future__ import annotations
 
 import os
+import asyncio
 import json
 import logging
 from pathlib import Path
 from typing import Any
+from datetime import datetime, timezone
+from zoneinfo import ZoneInfo
 
 import voluptuous as vol
 from aiohttp import ClientError
 import aiofiles
+import unicodedata
 
 from homeassistant.config_entries import ConfigEntry, ConfigFlow, ConfigFlowResult
 from homeassistant.core import callback
@@ -34,7 +38,10 @@ from .const import (
     PROVINCE_NAME,
     STATION_STATUS,
     LIMIT_XEMA,
-	LIMIT_PREDICCIO
+	LIMIT_PREDICCIO,
+    LIMIT_XDDE,
+    LIMIT_BASIC,
+    LIMIT_QUOTA
 )
     
 from .options_flow import MeteocatOptionsFlowHandler
@@ -43,11 +50,19 @@ from meteocatpy.symbols import MeteocatSymbols
 from meteocatpy.variables import MeteocatVariables
 from meteocatpy.townstations import MeteocatTownStations
 from meteocatpy.infostation import MeteocatInfoStation
+from meteocatpy.quotes import MeteocatQuotes
 
 from meteocatpy.exceptions import BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError
 
 _LOGGER = logging.getLogger(__name__)
 
+TIMEZONE = ZoneInfo("Europe/Madrid")
+
+def normalize_name(name):
+    """Normaliza el nombre eliminando acentos y convirtiendo a minúsculas."""
+    name = unicodedata.normalize("NFKD", name).encode("ASCII", "ignore").decode("utf-8")
+    return name.lower()
+
 class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
     """Flujo de configuración para Meteocat."""
 
@@ -62,6 +77,89 @@ class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
         self.station_name: str | None = None
         self._cache = {}
 
+    async def fetch_and_save_quotes(self, api_key):
+        """Obtiene las cuotas de la API de Meteocat y las guarda en quotes.json."""
+        meteocat_quotes = MeteocatQuotes(api_key)
+        quotes_dir = os.path.join(
+            self.hass.config.path(),
+            "custom_components",
+            "meteocat",
+            "files"
+        )
+        os.makedirs(quotes_dir, exist_ok=True)
+        quotes_file = os.path.join(quotes_dir, "quotes.json")
+
+        try:
+            data = await asyncio.wait_for(
+                meteocat_quotes.get_quotes(),
+                timeout=30
+            )
+            
+            # Modificar los nombres de los planes con normalización
+            plan_mapping = {
+                "xdde_": "XDDE",
+                "prediccio_": "Prediccio",
+                "referencia basic": "Basic",
+                "xema_": "XEMA",
+                "quota": "Quota"
+            }
+
+            modified_plans = []
+            for plan in data["plans"]:
+                normalized_nom = normalize_name(plan["nom"])
+                new_name = next((v for k, v in plan_mapping.items() if normalized_nom.startswith(k)), plan["nom"])
+
+                modified_plans.append({
+                    "nom": new_name,
+                    "periode": plan["periode"],
+                    "maxConsultes": plan["maxConsultes"],
+                    "consultesRestants": plan["consultesRestants"],
+                    "consultesRealitzades": plan["consultesRealitzades"]
+                })
+
+            # Añadir la clave 'actualitzat' con la fecha y hora actual de la zona horaria local
+            current_time = datetime.now(timezone.utc).astimezone(TIMEZONE).isoformat()
+            data_with_timestamp = {
+                "actualitzat": {
+                    "dataUpdate": current_time
+                },
+                "client": data["client"],
+                "plans": modified_plans
+            }
+
+            # Guardar los datos en el archivo JSON
+            async with aiofiles.open(quotes_file, "w", encoding="utf-8") as file:
+                await file.write(json.dumps(data_with_timestamp, ensure_ascii=False, indent=4))
+            
+            _LOGGER.info("Cuotas guardadas exitosamente en %s", quotes_file)
+
+        except Exception as ex:
+            _LOGGER.error("Error al obtener o guardar las cuotas: %s", ex)
+            raise HomeAssistantError("No se pudieron obtener las cuotas de la API")
+    
+    async def create_alerts_file(self):
+        """Crea el archivo alerts.json si no existe."""
+        alerts_dir = os.path.join(
+            self.hass.config.path(),
+            "custom_components",
+            "meteocat",
+            "files"
+        )
+        os.makedirs(alerts_dir, exist_ok=True)
+        alerts_file = os.path.join(alerts_dir, "alerts.json")
+
+        if not os.path.exists(alerts_file):
+            initial_data = {
+                "actualitzat": {
+                    "dataUpdate": "1970-01-01T00:00:00+00:00"
+                },
+                "dades": []
+            }
+            async with aiofiles.open(alerts_file, "w", encoding="utf-8") as file:
+                await file.write(json.dumps(initial_data, ensure_ascii=False, indent=4))
+            
+            _LOGGER.info("Archivo alerts.json creado en %s", alerts_file)
+
     async def async_step_user(
         self, user_input: dict[str, Any] | None = None
     ) -> ConfigFlowResult:
@@ -75,6 +173,10 @@ class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
 
             try:
                 self.municipis = await town_client.get_municipis()
+                # Aquí obtenemos y guardamos las cuotas
+                await self.fetch_and_save_quotes(self.api_key)
+                # Aquí creamos el archivo alerts.json si no existe
+                await self.create_alerts_file()
             except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
                 _LOGGER.error("Error al conectar con la API de Meteocat: %s", ex)
                 errors["base"] = "cannot_connect"
@@ -231,6 +333,9 @@ class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
         if user_input is not None:
             self.limit_xema = user_input.get(LIMIT_XEMA, 750)
             self.limit_prediccio = user_input.get(LIMIT_PREDICCIO, 100)
+            self.limit_xdde = user_input.get(LIMIT_XDDE, 250)
+            self.limit_quota = user_input.get(LIMIT_QUOTA, 300)
+            self.limit_basic = user_input.get(LIMIT_BASIC, 300)
             return self.async_create_entry(
                 title=self.selected_municipi["nom"],
                 data={
@@ -252,6 +357,9 @@ class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
                     STATION_STATUS: str(self.station_status),
                     LIMIT_XEMA: self.limit_xema,
                     LIMIT_PREDICCIO: self.limit_prediccio,
+                    LIMIT_XDDE: self.limit_xdde,
+                    LIMIT_QUOTA: self.limit_quota,
+                    LIMIT_BASIC: self.limit_basic,
                 },
             )
 
@@ -259,6 +367,9 @@ class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
             {
                 vol.Required(LIMIT_XEMA, default=750): cv.positive_int,
                 vol.Required(LIMIT_PREDICCIO, default=100): cv.positive_int,
+                vol.Required(LIMIT_XDDE, default=250): cv.positive_int,
+                vol.Required(LIMIT_QUOTA, default=300): cv.positive_int,
+                vol.Required(LIMIT_BASIC, default=2000): cv.positive_int,
             }
         )
 

package/custom_components/meteocat/const.py

@@ -18,6 +18,9 @@ PROVINCE_ID = "province_id"
 PROVINCE_NAME = "province_name"
 LIMIT_XEMA = "limit_xema"
 LIMIT_PREDICCIO = "limit_prediccio"
+LIMIT_XDDE = "limit_xdde"
+LIMIT_BASIC = "limit_basic"
+LIMIT_QUOTA = "limit_quota"
 STATION_STATUS = "station_status"
 HOURLY_FORECAST_FILE_STATUS = "hourly_forecast_file_status"
 DAILY_FORECAST_FILE_STATUS = "daily_forecast_file_status"
@@ -32,6 +35,12 @@ ALERT_COLD = "alert_cold"
 ALERT_WARM = "alert_warm"
 ALERT_WARM_NIGHT = "alert_warm_night"
 ALERT_SNOW = "alert_snow"
+QUOTA_FILE_STATUS = "quota_file_status"
+QUOTA_XDDE = "quota_xdde"
+QUOTA_PREDICCIO = "quota_prediccio"
+QUOTA_BASIC = "quota_basic"
+QUOTA_XEMA = "quota_xema"
+QUOTA_QUERIES = "quota_queries"
 
 from homeassistant.const import Platform
 
@@ -44,6 +53,7 @@ DEFAULT_VALIDITY_DAYS = 1  # Número de días a partir de los cuales se consider
 DEFAULT_VALIDITY_HOURS = 5  # Hora a partir de la cual la API tiene la información actualizada de predicciones disponible para descarga
 DEFAULT_VALIDITY_MINUTES = 0  # Minutos a partir de los cuales la API tiene la información actualizada de predicciones disponible para descarga
 DEFAULT_ALERT_VALIDITY_TIME = 120  # Minutos a partir de los cuales las alertas están obsoletas y se se debe proceder a una nueva llamada a la API
+DEFAULT_QUOTES_VALIDITY_TIME = 240 # Minutos a partir de los cuales los datos de cuotas están obsoletos y se se debe proceder a una nueva llamada a la API
 
 # Multiplicadores para la duración de validez basada en limit_prediccio
 ALERT_VALIDITY_MULTIPLIER_100 = 12  # para limit_prediccio <= 100