Be a bit more defensive on empty data

Signed-off-by: Olivier Mehani <[email protected]>
LeighCurran · Aug 9, 2023 · c715e32 · c715e32
1 parent 20bc429
commit c715e32
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/auroraplus/sensor.py b/custom_components/auroraplus/sensor.py
@@ -237,10 +237,12 @@ async def async_update(self):
                 float(self._api.EstimatedBalance), self._rounding)
         elif self._sensor == SENSOR_DOLLARVALUEUSAGE:
             self._state = round(
-                self._api.DollarValueUsage['Total'], self._rounding)
+                self._api.DollarValueUsage.get('Total', float('nan')),
+                self._rounding)
         elif self._sensor == SENSOR_KILOWATTHOURUSAGE:
             self._state = round(
-                self._api.KilowattHourUsage['Total'], self._rounding)
+                self._api.KilowattHourUsage.get('Total', float('nan')),
+                self._rounding)
         elif self._sensor.startswith(SENSOR_KILOWATTHOURUSAGETARIFF):
             pass