"""Support for Met.no weather service.""" from __future__ import annotations from types import MappingProxyType from typing import TYPE_CHECKING, Any from homeassistant.components.weather import ( ATTR_FORECAST_CONDITION, ATTR_FORECAST_TIME, ATTR_WEATHER_CLOUD_COVERAGE, ATTR_WEATHER_DEW_POINT, ATTR_WEATHER_HUMIDITY, ATTR_WEATHER_PRESSURE, ATTR_WEATHER_TEMPERATURE, ATTR_WEATHER_WIND_BEARING, ATTR_WEATHER_WIND_GUST_SPEED, ATTR_WEATHER_WIND_SPEED, DOMAIN as WEATHER_DOMAIN, Forecast, SingleCoordinatorWeatherEntity, WeatherEntityFeature, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME, UnitOfPrecipitationDepth, UnitOfPressure, UnitOfSpeed, UnitOfTemperature, ) from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import entity_registry as er, sun from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.util.unit_system import METRIC_SYSTEM from .const import ( ATTR_CONDITION_CLEAR_NIGHT, ATTR_CONDITION_SUNNY, ATTR_MAP, CONDITIONS_MAP, CONF_TRACK_HOME, DOMAIN, FORECAST_MAP, ) from .coordinator import MetDataUpdateCoordinator DEFAULT_NAME = "Met.no" async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Add a weather entity from a config_entry.""" coordinator: MetDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id] entity_registry = er.async_get(hass) name: str | None is_metric = hass.config.units is METRIC_SYSTEM if config_entry.data.get(CONF_TRACK_HOME, False): name = hass.config.location_name else: name = config_entry.data.get(CONF_NAME, DEFAULT_NAME) if TYPE_CHECKING: assert isinstance(name, str) entities = [MetWeather(coordinator, config_entry, False, name, is_metric)] # Add hourly entity to legacy config entries if entity_registry.async_get_entity_id( WEATHER_DOMAIN, DOMAIN, _calculate_unique_id(config_entry.data, True), ): name = f"{name} hourly" entities.append(MetWeather(coordinator, config_entry, True, name, is_metric)) async_add_entities(entities) def _calculate_unique_id(config: MappingProxyType[str, Any], hourly: bool) -> str: """Calculate unique ID.""" name_appendix = "" if hourly: name_appendix = "-hourly" if config.get(CONF_TRACK_HOME): return f"home{name_appendix}" return f"{config[CONF_LATITUDE]}-{config[CONF_LONGITUDE]}{name_appendix}" def format_condition(condition: str) -> str: """Return condition from dict CONDITIONS_MAP.""" for key, value in CONDITIONS_MAP.items(): if condition in value: return key return condition class MetWeather(SingleCoordinatorWeatherEntity[MetDataUpdateCoordinator]): """Implementation of a Met.no weather condition.""" _attr_attribution = ( "Weather forecast from met.no, delivered by the Norwegian " "Meteorological Institute." ) _attr_has_entity_name = True _attr_native_temperature_unit = UnitOfTemperature.CELSIUS _attr_native_precipitation_unit = UnitOfPrecipitationDepth.MILLIMETERS _attr_native_pressure_unit = UnitOfPressure.HPA _attr_native_wind_speed_unit = UnitOfSpeed.KILOMETERS_PER_HOUR _attr_supported_features = ( WeatherEntityFeature.FORECAST_DAILY | WeatherEntityFeature.FORECAST_HOURLY ) def __init__( self, coordinator: MetDataUpdateCoordinator, config_entry: ConfigEntry, hourly: bool, name: str, is_metric: bool, ) -> None: """Initialise the platform with a data instance and site.""" super().__init__(coordinator) self._attr_unique_id = _calculate_unique_id(config_entry.data, hourly) self._config = config_entry.data self._is_metric = is_metric self._hourly = hourly self._attr_entity_registry_enabled_default = not hourly self._attr_device_info = DeviceInfo( name="Forecast", entry_type=DeviceEntryType.SERVICE, identifiers={(DOMAIN, config_entry.entry_id)}, manufacturer="Met.no", model="Forecast", configuration_url="https://www.met.no/en", ) self._attr_track_home = self._config.get(CONF_TRACK_HOME, False) self._attr_name = name @property def condition(self) -> str | None: """Return the current condition.""" condition = self.coordinator.data.current_weather_data.get("condition") if condition is None: return None if condition == ATTR_CONDITION_SUNNY and not sun.is_up(self.hass): condition = ATTR_CONDITION_CLEAR_NIGHT return format_condition(condition) @property def native_temperature(self) -> float | None: """Return the temperature.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_TEMPERATURE] ) @property def native_pressure(self) -> float | None: """Return the pressure.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_PRESSURE] ) @property def humidity(self) -> float | None: """Return the humidity.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_HUMIDITY] ) @property def native_wind_speed(self) -> float | None: """Return the wind speed.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_WIND_SPEED] ) @property def wind_bearing(self) -> float | str | None: """Return the wind direction.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_WIND_BEARING] ) @property def native_wind_gust_speed(self) -> float | None: """Return the wind gust speed in native units.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_WIND_GUST_SPEED] ) @property def cloud_coverage(self) -> float | None: """Return the cloud coverage.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_CLOUD_COVERAGE] ) @property def native_dew_point(self) -> float | None: """Return the dew point.""" return self.coordinator.data.current_weather_data.get( ATTR_MAP[ATTR_WEATHER_DEW_POINT] ) def _forecast(self, hourly: bool) -> list[Forecast] | None: """Return the forecast array.""" if hourly: met_forecast = self.coordinator.data.hourly_forecast else: met_forecast = self.coordinator.data.daily_forecast required_keys = {"temperature", ATTR_FORECAST_TIME} ha_forecast: list[Forecast] = [] for met_item in met_forecast: if not set(met_item).issuperset(required_keys): continue ha_item = { k: met_item[v] for k, v in FORECAST_MAP.items() if met_item.get(v) is not None } if ha_item.get(ATTR_FORECAST_CONDITION): ha_item[ATTR_FORECAST_CONDITION] = format_condition( ha_item[ATTR_FORECAST_CONDITION] ) ha_forecast.append(ha_item) # type: ignore[arg-type] return ha_forecast @property def forecast(self) -> list[Forecast] | None: """Return the forecast array.""" return self._forecast(self._hourly) @callback def _async_forecast_daily(self) -> list[Forecast] | None: """Return the daily forecast in native units.""" return self._forecast(False) @callback def _async_forecast_hourly(self) -> list[Forecast] | None: """Return the hourly forecast in native units.""" return self._forecast(True)