fix[weather::forecasts]: add forecast data to attributes to load it w…

…hile re-start Actually simulate behaviour HA 2024.3
IATkachenko · Mar 30, 2024 · cc5a1a4 · cc5a1a4
1 parent a9d801d
commit cc5a1a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/custom_components/yandex_weather/const.py b/custom_components/yandex_weather/const.py
@@ -31,7 +31,7 @@
 ATTR_MIN_FORECAST_TEMPERATURE = "min_forecast_temperature"
 ATTR_API_FORECAST_ICONS = "forecast_icons"
 
-ATTR_FORECAST_DATA = "forecast_data"
+ATTR_FORECAST_DATA = "forecast"  # just to be able to load saved forecast after restart
 
 CONF_UPDATES_PER_DAY = "updates_per_day"
 CONF_IMAGE_SOURCE = "image_source"

diff --git a/custom_components/yandex_weather/weather.py b/custom_components/yandex_weather/weather.py
@@ -218,6 +218,7 @@ def _handle_coordinator_update(self) -> None:
             "wind_gust": self.coordinator.data.get(ATTR_API_WIND_GUST),
             "yandex_condition": self.coordinator.data.get(ATTR_API_YA_CONDITION),
             "forecast_icons": self.coordinator.data.get(ATTR_API_FORECAST_ICONS),
+            ATTR_FORECAST_DATA: self._twice_daily_forecast,
         }
         try:
             self._attr_extra_state_attributes["temp_water"] = self.coordinator.data.get(