meteocat 0.1.25 → 0.1.27

package/custom_components/meteocat/config_flow.py

@@ -1,192 +1,199 @@
-from __future__ import annotations
-
-import json
-import logging
-from pathlib import Path
-from typing import Any
-
-import voluptuous as vol
-from aiohttp import ClientError
-import aiofiles
-
-from homeassistant.config_entries import ConfigEntry, ConfigFlow, ConfigFlowResult
-from homeassistant.core import callback
-from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers import aiohttp_client, config_validation as cv
-
-from .const import (
-    DOMAIN,
-    CONF_API_KEY,
-    TOWN_NAME,
-    TOWN_ID,
-    VARIABLE_NAME,
-    VARIABLE_ID,
-    STATION_NAME,
-    STATION_ID
-)
-    
-from .options_flow import MeteocatOptionsFlowHandler
-from meteocatpy.town import MeteocatTown
-from meteocatpy.symbols import MeteocatSymbols
-from meteocatpy.variables import MeteocatVariables
-from meteocatpy.townstations import MeteocatTownStations
-from meteocatpy.exceptions import BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError
-
-_LOGGER = logging.getLogger(__name__)
-
-class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
-    """Flujo de configuración para Meteocat."""
-
-    VERSION = 1
-
-    def __init__(self):
-        self.api_key: str | None = None
-        self.municipis: list[dict[str, Any]] = []
-        self.selected_municipi: dict[str, Any] | None = None
-        self.variable_id: str | None = None
-        self.station_id: str | None = None
-        self.station_name: str | None = None
-
-    async def async_step_user(
-        self, user_input: dict[str, Any] | None = None
-    ) -> ConfigFlowResult:
-        """Primer paso: Solicitar la API Key."""
-        errors = {}
-
-        if user_input is not None:
-            self.api_key = user_input[CONF_API_KEY]
-
-            town_client = MeteocatTown(self.api_key)
-
-            try:
-                self.municipis = await town_client.get_municipis()
-            except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
-                _LOGGER.error("Error al conectar con la API de Meteocat: %s", ex)
-                errors["base"] = "cannot_connect"
-            except Exception as ex:
-                _LOGGER.error("Error inesperado al validar la API Key: %s", ex)
-                errors["base"] = "unknown"
-
-            if not errors:
-                return await self.async_step_select_municipi()
-
-        schema = vol.Schema({vol.Required(CONF_API_KEY): str})
-        return self.async_show_form(step_id="user", data_schema=schema, errors=errors)
-
-    async def async_step_select_municipi(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult:
-        """Segundo paso: Seleccionar el municipio."""
-        errors = {}
-
-        if user_input is not None:
-            selected_codi = user_input["municipi"]
-            self.selected_municipi = next(
-                (m for m in self.municipis if m["codi"] == selected_codi), None
-            )
-
-            if self.selected_municipi:
-                await self.fetch_symbols_and_variables()
-
-        if not errors and self.selected_municipi:
-            return await self.async_step_select_station()
-
-        schema = vol.Schema(
-            {vol.Required("municipi"): vol.In({m["codi"]: m["nom"] for m in self.municipis})}
-        )
-
-        return self.async_show_form(step_id="select_municipi", data_schema=schema, errors=errors)
-
-    async def fetch_symbols_and_variables(self):
-        """Descarga los símbolos y las variables después de seleccionar el municipio."""
-
-        errors = {}
-
-        # Descargar y guardar los símbolos
-        assets_dir = Path(__file__).parent / "assets"
-        assets_dir.mkdir(parents=True, exist_ok=True)
-        symbols_file = assets_dir / "symbols.json"
-        symbols_client = MeteocatSymbols(self.api_key)
-
-        try:
-            symbols_data = await symbols_client.fetch_symbols()
-            async with aiofiles.open(symbols_file, "w", encoding="utf-8") as file:
-                await file.write(json.dumps({"symbols": symbols_data}, ensure_ascii=False, indent=4))
-        except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
-            _LOGGER.error("Error al descargar o guardar los símbolos: %s", ex)
-            errors["base"] = "symbols_download_failed"
-
-        if not errors:
-            variables_client = MeteocatVariables(self.api_key)
-            try:
-                variables_data = await variables_client.get_variables()
-                self.variable_id = next(
-                    (v["codi"] for v in variables_data if v["nom"].lower() == "temperatura"), None
-                )
-                if not self.variable_id:
-                    _LOGGER.error("No se encontró la variable 'Temperatura'")
-                    errors["base"] = "variable_not_found"
-            except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
-                _LOGGER.error("Error al obtener las variables: %s", ex)
-                errors["base"] = "variables_fetch_failed"
-
-        if errors:
-            raise HomeAssistantError(errors)
-
-    async def async_step_select_station(
-        self, user_input: dict[str, Any] | None = None
-    ) -> ConfigFlowResult:
-        """Tercer paso: Seleccionar la estación para la variable seleccionada."""
-        errors = {}
-
-        townstations_client = MeteocatTownStations(self.api_key)
-        try:
-            stations_data = await townstations_client.get_town_stations(
-                self.selected_municipi["codi"], self.variable_id
-            )
-        except Exception as ex:
-            _LOGGER.error("Error al obtener las estaciones: %s", ex)
-            errors["base"] = "stations_fetch_failed"
-            stations_data = []
-
-        if user_input is not None:
-            selected_station_codi = user_input["station"]
-            selected_station = next(
-                (station for station in stations_data[0]["variables"][0]["estacions"] if station["codi"] == selected_station_codi),
-                None
-            )
-
-            if selected_station:
-                self.station_id = selected_station["codi"]
-                self.station_name = selected_station["nom"]
-
-                return self.async_create_entry(
-                    title=self.selected_municipi["nom"],
-                    data={
-                        CONF_API_KEY: self.api_key,
-                        TOWN_NAME: self.selected_municipi["nom"],
-                        TOWN_ID: self.selected_municipi["codi"],
-                        VARIABLE_NAME: "Temperatura",
-                        VARIABLE_ID: str(self.variable_id),
-                        STATION_NAME: self.station_name,
-                        STATION_ID: self.station_id
-                    },
-                )
-            else:
-                errors["base"] = "station_not_found"
-
-        schema = vol.Schema(
-            {
-                vol.Required("station"): vol.In(
-                    {station["codi"]: station["nom"] for station in stations_data[0]["variables"][0]["estacions"]}
-                )
-            }
-        )
-
-        return self.async_show_form(
-            step_id="select_station", data_schema=schema, errors=errors
-        )
-
-    @staticmethod
-    @callback
-    def async_get_options_flow(config_entry: ConfigEntry) -> MeteocatOptionsFlowHandler:
-        """Devuelve el flujo de opciones para esta configuración."""
-        return MeteocatOptionsFlowHandler(config_entry)
+from __future__ import annotations
+
+import os
+import json
+import logging
+from pathlib import Path
+from typing import Any
+
+import voluptuous as vol
+from aiohttp import ClientError
+import aiofiles
+
+from homeassistant.config_entries import ConfigEntry, ConfigFlow, ConfigFlowResult
+from homeassistant.core import callback
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers import aiohttp_client, config_validation as cv
+
+from .const import (
+    DOMAIN,
+    CONF_API_KEY,
+    TOWN_NAME,
+    TOWN_ID,
+    VARIABLE_NAME,
+    VARIABLE_ID,
+    STATION_NAME,
+    STATION_ID
+)
+    
+from .options_flow import MeteocatOptionsFlowHandler
+from meteocatpy.town import MeteocatTown
+from meteocatpy.symbols import MeteocatSymbols
+from meteocatpy.variables import MeteocatVariables
+from meteocatpy.townstations import MeteocatTownStations
+from meteocatpy.exceptions import BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError
+
+_LOGGER = logging.getLogger(__name__)
+
+class MeteocatConfigFlow(ConfigFlow, domain=DOMAIN):
+    """Flujo de configuración para Meteocat."""
+
+    VERSION = 1
+
+    def __init__(self):
+        self.api_key: str | None = None
+        self.municipis: list[dict[str, Any]] = []
+        self.selected_municipi: dict[str, Any] | None = None
+        self.variable_id: str | None = None
+        self.station_id: str | None = None
+        self.station_name: str | None = None
+
+    async def async_step_user(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Primer paso: Solicitar la API Key."""
+        errors = {}
+
+        if user_input is not None:
+            self.api_key = user_input[CONF_API_KEY]
+
+            town_client = MeteocatTown(self.api_key)
+
+            try:
+                self.municipis = await town_client.get_municipis()
+            except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
+                _LOGGER.error("Error al conectar con la API de Meteocat: %s", ex)
+                errors["base"] = "cannot_connect"
+            except Exception as ex:
+                _LOGGER.error("Error inesperado al validar la API Key: %s", ex)
+                errors["base"] = "unknown"
+
+            if not errors:
+                return await self.async_step_select_municipi()
+
+        schema = vol.Schema({vol.Required(CONF_API_KEY): str})
+        return self.async_show_form(step_id="user", data_schema=schema, errors=errors)
+
+    async def async_step_select_municipi(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult:
+        """Segundo paso: Seleccionar el municipio."""
+        errors = {}
+
+        if user_input is not None:
+            selected_codi = user_input["municipi"]
+            self.selected_municipi = next(
+                (m for m in self.municipis if m["codi"] == selected_codi), None
+            )
+
+            if self.selected_municipi:
+                await self.fetch_symbols_and_variables()
+
+        if not errors and self.selected_municipi:
+            return await self.async_step_select_station()
+
+        schema = vol.Schema(
+            {vol.Required("municipi"): vol.In({m["codi"]: m["nom"] for m in self.municipis})}
+        )
+
+        return self.async_show_form(step_id="select_municipi", data_schema=schema, errors=errors)
+
+    async def fetch_symbols_and_variables(self):
+        """Descarga los símbolos y las variables después de seleccionar el municipio."""
+
+        errors = {}
+
+        # Descargar y guardar los símbolos
+        assets_dir = Path(__file__).parent / "assets"
+        assets_dir.mkdir(parents=True, exist_ok=True)
+        symbols_file = assets_dir / "symbols.json"
+        symbols_client = MeteocatSymbols(self.api_key)
+
+        try:
+            symbols_data = await symbols_client.fetch_symbols()
+            async with aiofiles.open(symbols_file, "w", encoding="utf-8") as file:
+                await file.write(json.dumps({"symbols": symbols_data}, ensure_ascii=False, indent=4))
+        except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
+            _LOGGER.error("Error al descargar o guardar los símbolos: %s", ex)
+            errors["base"] = "symbols_download_failed"
+
+        if not errors:
+            # Configurar la ruta para la caché en la carpeta `custom_components/meteocat`
+            cache_dir = os.path.join(os.path.dirname(__file__), ".meteocat_cache")
+
+            # Crear la carpeta de caché si no existe
+            os.makedirs(cache_dir, exist_ok=True)
+
+            variables_client = MeteocatVariables(self.api_key, cache_dir=cache_dir)
+            try:
+                variables_data = await variables_client.get_variables()
+                self.variable_id = next(
+                    (v["codi"] for v in variables_data if v["nom"].lower() == "temperatura"), None
+                )
+                if not self.variable_id:
+                    _LOGGER.error("No se encontró la variable 'Temperatura'")
+                    errors["base"] = "variable_not_found"
+            except (BadRequestError, ForbiddenError, TooManyRequestsError, InternalServerError, UnknownAPIError) as ex:
+                _LOGGER.error("Error al obtener las variables: %s", ex)
+                errors["base"] = "variables_fetch_failed"
+
+        if errors:
+            raise HomeAssistantError(errors)
+
+    async def async_step_select_station(
+        self, user_input: dict[str, Any] | None = None
+    ) -> ConfigFlowResult:
+        """Tercer paso: Seleccionar la estación para la variable seleccionada."""
+        errors = {}
+
+        townstations_client = MeteocatTownStations(self.api_key)
+        try:
+            stations_data = await townstations_client.get_town_stations(
+                self.selected_municipi["codi"], self.variable_id
+            )
+        except Exception as ex:
+            _LOGGER.error("Error al obtener las estaciones: %s", ex)
+            errors["base"] = "stations_fetch_failed"
+            stations_data = []
+
+        if user_input is not None:
+            selected_station_codi = user_input["station"]
+            selected_station = next(
+                (station for station in stations_data[0]["variables"][0]["estacions"] if station["codi"] == selected_station_codi),
+                None
+            )
+
+            if selected_station:
+                self.station_id = selected_station["codi"]
+                self.station_name = selected_station["nom"]
+
+                return self.async_create_entry(
+                    title=self.selected_municipi["nom"],
+                    data={
+                        CONF_API_KEY: self.api_key,
+                        TOWN_NAME: self.selected_municipi["nom"],
+                        TOWN_ID: self.selected_municipi["codi"],
+                        VARIABLE_NAME: "Temperatura",
+                        VARIABLE_ID: str(self.variable_id),
+                        STATION_NAME: self.station_name,
+                        STATION_ID: self.station_id
+                    },
+                )
+            else:
+                errors["base"] = "station_not_found"
+
+        schema = vol.Schema(
+            {
+                vol.Required("station"): vol.In(
+                    {station["codi"]: station["nom"] for station in stations_data[0]["variables"][0]["estacions"]}
+                )
+            }
+        )
+
+        return self.async_show_form(
+            step_id="select_station", data_schema=schema, errors=errors
+        )
+
+    @staticmethod
+    @callback
+    def async_get_options_flow(config_entry: ConfigEntry) -> MeteocatOptionsFlowHandler:
+        """Devuelve el flujo de opciones para esta configuración."""
+        return MeteocatOptionsFlowHandler(config_entry)

package/custom_components/meteocat/const.py

@@ -1,55 +1,55 @@
-# Constantes generales
-DOMAIN = "meteocat"
-BASE_URL = "https://api.meteo.cat"
-CONF_API_KEY = "api_key"
-TOWN_NAME = "town_name"
-TOWN_ID = "town_id"
-VARIABLE_NAME = "variable_name"
-VARIABLE_ID = "variable_id"
-STATION_NAME = "station_name"
-STATION_ID = "station_id"
-
-# Códigos de sensores de la API
-WIND_SPEED = "wind_speed"  # Velocidad del viento
-WIND_DIRECTION = "wind_direction"  # Dirección del viento
-TEMPERATURE = "temperature"  # Temperatura
-HUMIDITY = "humidity"  # Humedad relativa
-PRESSURE = "pressure"  # Presión atmosférica
-PRECIPITATION = "precipitation"  # Precipitación
-SOLAR_GLOBAL_IRRADIANCE = "solar_global_irradiance"  # Irradiación solar global
-UV_INDEX = "uv_index"  # UV
-MAX_TEMPERATURE = "max_temperature"  # Temperatura máxima
-MIN_TEMPERATURE = "min_temperature"  # Temperatura mínima
-WIND_GUST = "wind_gust"  # Racha de viento
-STATION_TIMESTAMP = "station_timestamp"  # Código de tiempo de la estación
-
-# Definición de códigos para variables
-WIND_SPEED_CODE = 30
-WIND_DIRECTION_CODE = 31
-TEMPERATURE_CODE = 32
-HUMIDITY_CODE = 33
-PRESSURE_CODE = 34
-PRECIPITATION_CODE = 35
-SOLAR_GLOBAL_IRRADIANCE_CODE = 36
-UV_INDEX_CODE = 39
-MAX_TEMPERATURE_CODE = 40
-MIN_TEMPERATURE_CODE = 42
-WIND_GUST_CODE = 50
-
-# Mapeo de códigos 'estatCel' a condiciones de Home Assistant
-CONDITION_MAPPING = {
-    "sunny": [1],
-    "clear-night": [1],
-    "partlycloudy": [2, 3],
-    "cloudy": [4, 20, 21, 22],
-    "rainy": [5, 6, 23],
-    "pouring": [7, 8, 25],
-    "lightning-rainy": [8, 24],
-    "hail": [9],
-    "snowy": [10, 26, 27, 28],
-    "fog": [11, 12],
-    "snow-rainy": [27, 29, 30],
-}
-
-# Platforms
-PLATFORMS = ["sensor"]
+# Constantes generales
+DOMAIN = "meteocat"
+BASE_URL = "https://api.meteo.cat"
+CONF_API_KEY = "api_key"
+TOWN_NAME = "town_name"
+TOWN_ID = "town_id"
+VARIABLE_NAME = "variable_name"
+VARIABLE_ID = "variable_id"
+STATION_NAME = "station_name"
+STATION_ID = "station_id"
+
+# Códigos de sensores de la API
+WIND_SPEED = "wind_speed"  # Velocidad del viento
+WIND_DIRECTION = "wind_direction"  # Dirección del viento
+TEMPERATURE = "temperature"  # Temperatura
+HUMIDITY = "humidity"  # Humedad relativa
+PRESSURE = "pressure"  # Presión atmosférica
+PRECIPITATION = "precipitation"  # Precipitación
+SOLAR_GLOBAL_IRRADIANCE = "solar_global_irradiance"  # Irradiación solar global
+UV_INDEX = "uv_index"  # UV
+MAX_TEMPERATURE = "max_temperature"  # Temperatura máxima
+MIN_TEMPERATURE = "min_temperature"  # Temperatura mínima
+WIND_GUST = "wind_gust"  # Racha de viento
+STATION_TIMESTAMP = "station_timestamp"  # Código de tiempo de la estación
+
+# Definición de códigos para variables
+WIND_SPEED_CODE = 30
+WIND_DIRECTION_CODE = 31
+TEMPERATURE_CODE = 32
+HUMIDITY_CODE = 33
+PRESSURE_CODE = 34
+PRECIPITATION_CODE = 35
+SOLAR_GLOBAL_IRRADIANCE_CODE = 36
+UV_INDEX_CODE = 39
+MAX_TEMPERATURE_CODE = 40
+MIN_TEMPERATURE_CODE = 42
+WIND_GUST_CODE = 50
+
+# Mapeo de códigos 'estatCel' a condiciones de Home Assistant
+CONDITION_MAPPING = {
+    "sunny": [1],
+    "clear-night": [1],
+    "partlycloudy": [2, 3],
+    "cloudy": [4, 20, 21, 22],
+    "rainy": [5, 6, 23],
+    "pouring": [7, 8, 25],
+    "lightning-rainy": [8, 24],
+    "hail": [9],
+    "snowy": [10, 26, 27, 28],
+    "fog": [11, 12],
+    "snow-rainy": [27, 29, 30],
+}
+
+# Platforms
+PLATFORMS = ["sensor"]