From e3ea20eca7532d6b842c0109d7728be101d9c836 Mon Sep 17 00:00:00 2001
From: Alvaro Duarte <59612788+ad-ha@users.noreply.github.com>
Date: Tue, 8 Feb 2022 14:25:12 +0100
Subject: [PATCH 1/2] Update sensor.py

Add RestoreEntity subclass and async procedure to restore data upon HA restart
---
 custom_components/ide/sensor.py | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/custom_components/ide/sensor.py b/custom_components/ide/sensor.py
index 9b76385..23e751f 100644
--- a/custom_components/ide/sensor.py
+++ b/custom_components/ide/sensor.py
@@ -23,18 +23,20 @@
 )
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.typing import (
     ConfigType,
     DiscoveryInfoType,
 )
 import homeassistant.helpers.config_validation as cv
 
-__VERSION__ = "0.1.1"
+__VERSION__ = "0.2.1"
 
 from oligo.asyncio import AsyncIber
 from oligo.exception import IberException
 
 DOMAIN = "ide"
+DATA_UPDATED = "i-DE Sensor Update"
 
 # Validation of the user's configuration
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
@@ -89,7 +91,7 @@ async def async_setup_platform(
     )
 
 
-class IDESensor(SensorEntity):
+class IDESensor(RestoreEntity, SensorEntity):
     """Representation of a Sensor."""
 
     _attr_state_class = SensorStateClass.TOTAL_INCREASING
@@ -110,6 +112,26 @@ def __init__(self, config, name, variable, unit, deviceclass, stateclass):
         self.username = config[CONF_USERNAME]
         self.password = config[CONF_PASSWORD]
 
+    async def async_added_to_hass(self) -> None:
+        """Handle entity which will be added."""
+        await super().async_added_to_hass()
+        state = await self.async_get_last_state()
+        if not state:
+            return
+        self._state = state.state
+        _LOGGER.warning(
+            "i-DE Meter could not be updated. Last data restored {}".format(self._state)
+        )
+
+        async_dispatcher_connect(
+            self.hass, DATA_UPDATED, self._schedule_immediate_update
+        )
+
+    @callback
+        """Schedule Update when possible"""
+    def _schedule_immediate_update(self):
+        self.async_schedule_update_ha_state(True)
+        
     @property
     def name(self):
         """Return the name of the sensor."""

From a80fbd223daa67c4321414ab1e5ca5b60577307a Mon Sep 17 00:00:00 2001
From: Alvaro Duarte <59612788+ad-ha@users.noreply.github.com>
Date: Tue, 8 Feb 2022 14:28:34 +0100
Subject: [PATCH 2/2] Update sensor.py

---
 custom_components/ide/sensor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/custom_components/ide/sensor.py b/custom_components/ide/sensor.py
index 23e751f..bac5123 100644
--- a/custom_components/ide/sensor.py
+++ b/custom_components/ide/sensor.py
@@ -128,8 +128,8 @@ async def async_added_to_hass(self) -> None:
         )
 
     @callback
-        """Schedule Update when possible"""
     def _schedule_immediate_update(self):
+        """Schedule Update when possible"""
         self.async_schedule_update_ha_state(True)
         
     @property