Add hourly forecast to open_meteo (#113622)
* Add hourly forecast to open_meteo * Ran ruff formatting again
This commit is contained in:
parent
b7527feb5f
commit
a1eef4732f
2 changed files with 70 additions and 9 deletions
|
@ -5,6 +5,7 @@ from __future__ import annotations
|
||||||
from open_meteo import (
|
from open_meteo import (
|
||||||
DailyParameters,
|
DailyParameters,
|
||||||
Forecast,
|
Forecast,
|
||||||
|
HourlyParameters,
|
||||||
OpenMeteo,
|
OpenMeteo,
|
||||||
OpenMeteoError,
|
OpenMeteoError,
|
||||||
PrecipitationUnit,
|
PrecipitationUnit,
|
||||||
|
@ -45,6 +46,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
||||||
DailyParameters.WIND_DIRECTION_10M_DOMINANT,
|
DailyParameters.WIND_DIRECTION_10M_DOMINANT,
|
||||||
DailyParameters.WIND_SPEED_10M_MAX,
|
DailyParameters.WIND_SPEED_10M_MAX,
|
||||||
],
|
],
|
||||||
|
hourly=[
|
||||||
|
HourlyParameters.PRECIPITATION,
|
||||||
|
HourlyParameters.TEMPERATURE_2M,
|
||||||
|
HourlyParameters.WEATHER_CODE,
|
||||||
|
],
|
||||||
precipitation_unit=PrecipitationUnit.MILLIMETERS,
|
precipitation_unit=PrecipitationUnit.MILLIMETERS,
|
||||||
temperature_unit=TemperatureUnit.CELSIUS,
|
temperature_unit=TemperatureUnit.CELSIUS,
|
||||||
timezone="UTC",
|
timezone="UTC",
|
||||||
|
|
|
@ -5,6 +5,12 @@ from __future__ import annotations
|
||||||
from open_meteo import Forecast as OpenMeteoForecast
|
from open_meteo import Forecast as OpenMeteoForecast
|
||||||
|
|
||||||
from homeassistant.components.weather import (
|
from homeassistant.components.weather import (
|
||||||
|
ATTR_FORECAST_CONDITION,
|
||||||
|
ATTR_FORECAST_NATIVE_PRECIPITATION,
|
||||||
|
ATTR_FORECAST_NATIVE_TEMP,
|
||||||
|
ATTR_FORECAST_NATIVE_TEMP_LOW,
|
||||||
|
ATTR_FORECAST_NATIVE_WIND_SPEED,
|
||||||
|
ATTR_FORECAST_WIND_BEARING,
|
||||||
Forecast,
|
Forecast,
|
||||||
SingleCoordinatorWeatherEntity,
|
SingleCoordinatorWeatherEntity,
|
||||||
WeatherEntityFeature,
|
WeatherEntityFeature,
|
||||||
|
@ -15,6 +21,7 @@ from homeassistant.core import HomeAssistant, callback
|
||||||
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
|
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
||||||
|
from homeassistant.util import dt as dt_util
|
||||||
|
|
||||||
from .const import DOMAIN, WMO_TO_HA_CONDITION_MAP
|
from .const import DOMAIN, WMO_TO_HA_CONDITION_MAP
|
||||||
|
|
||||||
|
@ -39,7 +46,9 @@ class OpenMeteoWeatherEntity(
|
||||||
_attr_native_precipitation_unit = UnitOfPrecipitationDepth.MILLIMETERS
|
_attr_native_precipitation_unit = UnitOfPrecipitationDepth.MILLIMETERS
|
||||||
_attr_native_temperature_unit = UnitOfTemperature.CELSIUS
|
_attr_native_temperature_unit = UnitOfTemperature.CELSIUS
|
||||||
_attr_native_wind_speed_unit = UnitOfSpeed.KILOMETERS_PER_HOUR
|
_attr_native_wind_speed_unit = UnitOfSpeed.KILOMETERS_PER_HOUR
|
||||||
_attr_supported_features = WeatherEntityFeature.FORECAST_DAILY
|
_attr_supported_features = (
|
||||||
|
WeatherEntityFeature.FORECAST_DAILY | WeatherEntityFeature.FORECAST_HOURLY
|
||||||
|
)
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
@ -95,31 +104,77 @@ class OpenMeteoWeatherEntity(
|
||||||
return None
|
return None
|
||||||
|
|
||||||
forecasts: list[Forecast] = []
|
forecasts: list[Forecast] = []
|
||||||
|
|
||||||
daily = self.coordinator.data.daily
|
daily = self.coordinator.data.daily
|
||||||
for index, time in enumerate(self.coordinator.data.daily.time):
|
for index, date in enumerate(self.coordinator.data.daily.time):
|
||||||
forecast = Forecast(
|
forecast = Forecast(
|
||||||
datetime=time.isoformat(),
|
datetime=date.isoformat(),
|
||||||
)
|
)
|
||||||
|
|
||||||
if daily.weathercode is not None:
|
if daily.weathercode is not None:
|
||||||
forecast["condition"] = WMO_TO_HA_CONDITION_MAP.get(
|
forecast[ATTR_FORECAST_CONDITION] = WMO_TO_HA_CONDITION_MAP.get(
|
||||||
daily.weathercode[index]
|
daily.weathercode[index]
|
||||||
)
|
)
|
||||||
|
|
||||||
if daily.precipitation_sum is not None:
|
if daily.precipitation_sum is not None:
|
||||||
forecast["native_precipitation"] = daily.precipitation_sum[index]
|
forecast[ATTR_FORECAST_NATIVE_PRECIPITATION] = daily.precipitation_sum[
|
||||||
|
index
|
||||||
|
]
|
||||||
|
|
||||||
if daily.temperature_2m_max is not None:
|
if daily.temperature_2m_max is not None:
|
||||||
forecast["native_temperature"] = daily.temperature_2m_max[index]
|
forecast[ATTR_FORECAST_NATIVE_TEMP] = daily.temperature_2m_max[index]
|
||||||
|
|
||||||
if daily.temperature_2m_min is not None:
|
if daily.temperature_2m_min is not None:
|
||||||
forecast["native_templow"] = daily.temperature_2m_min[index]
|
forecast[ATTR_FORECAST_NATIVE_TEMP_LOW] = daily.temperature_2m_min[
|
||||||
|
index
|
||||||
|
]
|
||||||
|
|
||||||
if daily.wind_direction_10m_dominant is not None:
|
if daily.wind_direction_10m_dominant is not None:
|
||||||
forecast["wind_bearing"] = daily.wind_direction_10m_dominant[index]
|
forecast[ATTR_FORECAST_WIND_BEARING] = (
|
||||||
|
daily.wind_direction_10m_dominant[index]
|
||||||
|
)
|
||||||
|
|
||||||
if daily.wind_speed_10m_max is not None:
|
if daily.wind_speed_10m_max is not None:
|
||||||
forecast["native_wind_speed"] = daily.wind_speed_10m_max[index]
|
forecast[ATTR_FORECAST_NATIVE_WIND_SPEED] = daily.wind_speed_10m_max[
|
||||||
|
index
|
||||||
|
]
|
||||||
|
|
||||||
|
forecasts.append(forecast)
|
||||||
|
|
||||||
|
return forecasts
|
||||||
|
|
||||||
|
@callback
|
||||||
|
def _async_forecast_hourly(self) -> list[Forecast] | None:
|
||||||
|
"""Return the daily forecast in native units."""
|
||||||
|
if self.coordinator.data.hourly is None:
|
||||||
|
return None
|
||||||
|
|
||||||
|
forecasts: list[Forecast] = []
|
||||||
|
|
||||||
|
# Can have data in the past: https://github.com/open-meteo/open-meteo/issues/699
|
||||||
|
today = dt_util.utcnow()
|
||||||
|
|
||||||
|
hourly = self.coordinator.data.hourly
|
||||||
|
for index, datetime in enumerate(self.coordinator.data.hourly.time):
|
||||||
|
if dt_util.as_utc(datetime) < today:
|
||||||
|
continue
|
||||||
|
|
||||||
|
forecast = Forecast(
|
||||||
|
datetime=datetime.isoformat(),
|
||||||
|
)
|
||||||
|
|
||||||
|
if hourly.weather_code is not None:
|
||||||
|
forecast[ATTR_FORECAST_CONDITION] = WMO_TO_HA_CONDITION_MAP.get(
|
||||||
|
hourly.weather_code[index]
|
||||||
|
)
|
||||||
|
|
||||||
|
if hourly.precipitation is not None:
|
||||||
|
forecast[ATTR_FORECAST_NATIVE_PRECIPITATION] = hourly.precipitation[
|
||||||
|
index
|
||||||
|
]
|
||||||
|
|
||||||
|
if hourly.temperature_2m is not None:
|
||||||
|
forecast[ATTR_FORECAST_NATIVE_TEMP] = hourly.temperature_2m[index]
|
||||||
|
|
||||||
forecasts.append(forecast)
|
forecasts.append(forecast)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue