Fix statistic bug in Tibber sensor (home-assistant#116112)

* Handle keyError in Tibber sensor Signed-off-by: Daniel Hjelseth Høyer <[email protected]> * Constant Signed-off-by: Daniel Hjelseth Høyer <[email protected]> --------- Signed-off-by: Daniel Hjelseth Høyer <[email protected]>
wittypluck · Apr 24, 2024 · 830e8d7 · 830e8d7
1 parent 67021be
commit 830e8d7
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/homeassistant/components/tibber/sensor.py b/homeassistant/components/tibber/sensor.py
@@ -53,6 +53,8 @@
 
 from .const import DOMAIN as TIBBER_DOMAIN, MANUFACTURER
 
+FIVE_YEARS = 5 * 365 * 24
+
 _LOGGER = logging.getLogger(__name__)
 
 ICON = "mdi:currency-usd"
@@ -724,9 +726,16 @@ async def _insert_statistics(self) -> None:
                         None,
                         {"sum"},
                     )
-                    first_stat = stat[statistic_id][0]
-                    _sum = cast(float, first_stat["sum"])
-                    last_stats_time = first_stat["start"]
+                    if statistic_id in stat:
+                        first_stat = stat[statistic_id][0]
+                        _sum = cast(float, first_stat["sum"])
+                        last_stats_time = first_stat["start"]
+                    else:
+                        hourly_data = await home.get_historic_data(
+                            FIVE_YEARS, production=is_production
+                        )
+                        _sum = 0.0
+                        last_stats_time = None
 
                 statistics = []