Fixed the excpetion and the missing low temperature in daily forecast

tsunglung · Mar 9, 2024 · a0e7569 · a0e7569
1 parent 13c00f2
commit a0e7569
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/custom_components/opencwb/weather.py b/custom_components/opencwb/weather.py
@@ -58,6 +58,7 @@ def __init__(
         weather_coordinator: WeatherUpdateCoordinator,
     ):
         """Initialize the sensor."""
+        super().__init__(weather_coordinator)
         self._attr_name = name
         self._attr_unique_id = unique_id
         self._weather_coordinator = weather_coordinator

diff --git a/custom_components/opencwb/weather_update_coordinator.py b/custom_components/opencwb/weather_update_coordinator.py
@@ -203,11 +203,16 @@ def _convert_forecast(self, entry):
         }
 
         temperature_dict = entry.temperature("celsius")
-        if "max" in temperature_dict and "min" in temperature_dict:
+        if temperature_dict.get("max") and temperature_dict.get("min"):
             forecast[ATTR_FORECAST_TEMP] = entry.temperature(
                 "celsius").get("max")
             forecast[ATTR_FORECAST_TEMP_LOW] = entry.temperature(
                 "celsius").get("min")
+        elif temperature_dict.get("temp_max") and temperature_dict.get("temp_min"):
+            forecast[ATTR_FORECAST_TEMP] = entry.temperature(
+                "celsius").get("temp_max")
+            forecast[ATTR_FORECAST_TEMP_LOW] = entry.temperature(
+                "celsius").get("temp_min")
         else:
             forecast[ATTR_FORECAST_TEMP] = entry.temperature(
                 "celsius").get("temp")