Merge pull request #62 from basilfx/feature/statistics

Set sensor state class to enable statistics
basilfx · Nov 16, 2022 · 3206684 · 3206684
2 parents 4f1e990 + 5cc1256
commit 3206684
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/custom_components/biketrax/sensor.py b/custom_components/biketrax/sensor.py
@@ -10,6 +10,7 @@
     SensorDeviceClass,
     SensorEntity,
     SensorEntityDescription,
+    SensorStateClass,
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
@@ -50,6 +51,7 @@ class BikeTraxSensorEntityDescription(
         key="battery_level",
         name="Battery level",
         native_unit_of_measurement=PERCENTAGE,
+        state_class=SensorStateClass.MEASUREMENT,
     ),
     BikeTraxSensorEntityDescription(
         coordinator=DATA_DEVICE,
@@ -58,6 +60,7 @@ class BikeTraxSensorEntityDescription(
         key="total_distance",
         name="Total distance",
         native_unit_of_measurement=LENGTH_KILOMETERS,
+        state_class=SensorStateClass.TOTAL_INCREASING,
     ),
     BikeTraxSensorEntityDescription(
         coordinator=DATA_DEVICE,
@@ -66,6 +69,7 @@ class BikeTraxSensorEntityDescription(
         key="speed",
         name="Current speed",
         native_unit_of_measurement=SPEED_KILOMETERS_PER_HOUR,
+        state_class=SensorStateClass.MEASUREMENT,
     ),
     BikeTraxSensorEntityDescription(
         coordinator=DATA_SUBSCRIPTION,