add more native_unit_of_measurement and state_class

jcgoette · Dec 11, 2021 · 6bbadbe · 6bbadbe
1 parent e506612
commit 6bbadbe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/custom_components/babybuddy/const.py b/custom_components/babybuddy/const.py
@@ -107,6 +107,7 @@ class BabyBuddyEntityDescription(SensorEntityDescription, SwitchEntityDescriptio
         icon="mdi:sleep",
         key=ATTR_SLEEP,
         native_unit_of_measurement=TIME_MINUTES,
+        state_class=STATE_CLASS_MEASUREMENT,
         state_key=lambda value: int(
             time_period_str(value[ATTR_DURATION]).total_seconds() / 60
         ),
@@ -127,6 +128,8 @@ class BabyBuddyEntityDescription(SensorEntityDescription, SwitchEntityDescriptio
     BabyBuddyEntityDescription(
         icon="mdi:baby",
         key=ATTR_TUMMY_TIMES,
+        native_unit_of_measurement=TIME_MINUTES,
+        state_class=STATE_CLASS_MEASUREMENT,
         state_key=lambda value: int(
             time_period_str(value[ATTR_DURATION]).total_seconds() / 60
         ),