Add sensor class to boiler usage sensors

Author: danielgospodinow

custom_components/eldom/manifest.json

@@ -11,5 +11,5 @@
   "requirements": ["pyeldom==0.4.1"],
   "ssdp": [],
   "zeroconf": [],
-  "version": "3.1.2"
+  "version": "3.2.0"
 }

custom_components/eldom/sensor.py

@@ -2,7 +2,11 @@
 
 import logging
 
-from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorStateClass,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import UnitOfEnergy
 from homeassistant.core import HomeAssistant, callback
@@ -120,6 +124,11 @@ def device_class(self) -> SensorDeviceClass:
         """Return the device class of the sensor."""
         return SensorDeviceClass.ENERGY
 
+    @property
+    def state_class(self) -> SensorStateClass:
+        """Return the state class of the sensor."""
+        return SensorStateClass.TOTAL_INCREASING
+
     @property
     def native_unit_of_measurement(self) -> str:
         """Return the unit of measurement."""
@@ -178,6 +187,11 @@ def device_class(self) -> SensorDeviceClass:
         """Return the device class of the sensor."""
         return SensorDeviceClass.ENERGY
 
+    @property
+    def state_class(self) -> SensorStateClass:
+        """Return the state class of the sensor."""
+        return SensorStateClass.TOTAL_INCREASING
+
     @property
     def native_unit_of_measurement(self) -> str:
         """Return the unit of measurement."""