fixing state class for layers (#197)

marcolivierarsenault · Sep 6, 2023 · ca6ba95 · ca6ba95
1 parent 687d1ab
commit ca6ba95
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/custom_components/moonraker/sensor.py b/custom_components/moonraker/sensor.py
@@ -8,6 +8,7 @@
     SensorDeviceClass,
     SensorEntity,
     SensorEntityDescription,
+    SensorStateClass,
 )
 from homeassistant.const import PERCENTAGE, UnitOfLength, UnitOfTemperature, UnitOfTime
 from homeassistant.core import callback
@@ -268,13 +269,15 @@ class MoonrakerSensorDescription(SensorEntityDescription):
         ),
         subscriptions=[("print_stats", "info", "total_layer")],
         icon="mdi:layers-triple",
+        state_class=SensorStateClass.MEASUREMENT,
     ),
     MoonrakerSensorDescription(
         key="current_layer",
         name="Current Layer",
         value_fn=lambda sensor: calculate_current_layer(sensor.coordinator.data),
         subscriptions=[("print_stats", "info", "current_layer")],
         icon="mdi:layers-edit",
+        state_class=SensorStateClass.MEASUREMENT,
     ),
     MoonrakerSensorDescription(
         key="toolhead_position_x",
@@ -421,6 +424,7 @@ async def async_setup_history_sensors(coordinator, entry, async_add_entities):
             subscriptions=[],
             icon="mdi:numeric",
             unit="Jobs",
+            state_class=SensorStateClass.TOTAL_INCREASING,
         ),
         MoonrakerSensorDescription(
             key="total_print_time",
@@ -442,6 +446,7 @@ async def async_setup_history_sensors(coordinator, entry, async_add_entities):
             subscriptions=[],
             icon="mdi:clock-outline",
             unit=UnitOfLength.METERS,
+            state_class=SensorStateClass.TOTAL_INCREASING,
         ),
         MoonrakerSensorDescription(
             key="longest_print",