Update sensor.py

Correct bug causing energy spikes on energy dashboard
hif2k1 · Dec 7, 2022 · e76fc18 · e76fc18
1 parent 830935e
commit e76fc18
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/custom_components/battery_sim/sensor.py b/custom_components/battery_sim/sensor.py
@@ -113,6 +113,11 @@ async def async_added_to_hass(self):
         if state:
             try: 
                 self._handle._sensors[self._sensor_type] = float(state.state)
+                self._last_reset = dt_util.as_utc(
+                    dt_util.parse_datetime(state.attributes.get(ATTR_LAST_RESET))
+                )
+                self._available = True
+                await self.async_update_ha_state(True)
             except:
                 _LOGGER.debug("Sensor state not restored properly.")
                 if self._sensor_type == GRID_IMPORT_SIM:
@@ -435,4 +440,4 @@ def state(self):
         return self.handle._sensors[BATTERY_MODE]
 
     def update(self):
-        """Not used"""
+        """Not used"""