add some classes to binary sensor (#110)

custom_components/multimatic/binary_sensor.py

@@ -114,6 +114,11 @@ def entity_category(self) -> str | None:
         """Return the category of the entity, if any."""
         return ENTITY_CATEGORY_DIAGNOSTIC
 
+    @property
+    def device_class(self):
+        """Return the class of this device, from component DEVICE_CLASSES."""
+        return BinarySensorDeviceClass.RUNNING
+
 
 class RoomWindow(MultimaticEntity, BinarySensorEntity):
     """multimatic window binary sensor."""
@@ -357,6 +362,11 @@ def entity_category(self) -> str | None:
         """Return the category of the entity, if any."""
         return ENTITY_CATEGORY_DIAGNOSTIC
 
+    @property
+    def device_class(self):
+        """Return the class of this device, from component DEVICE_CLASSES."""
+        return BinarySensorDeviceClass.UPDATE
+
 
 class BoxOnline(VRBoxEntity):
     """Check if box is online."""
@@ -527,7 +537,7 @@ def __init__(self, coordinator: MultimaticCoordinator) -> None:
     @property
     def is_on(self):
         """Return true if the binary sensor is on."""
-        return self.coordinator.data and self.coordinator.data.is_applied
+        return self.coordinator.data is not None and self.coordinator.data.is_applied
 
     @property
     def state_attributes(self):
@@ -554,6 +564,11 @@ def entity_category(self) -> str | None:
         """Return the category of the entity, if any."""
         return ENTITY_CATEGORY_DIAGNOSTIC
 
+    @property
+    def device_class(self):
+        """Return the class of this device, from component DEVICE_CLASSES."""
+        return BinarySensorDeviceClass.OCCUPANCY
+
 
 class QuickModeSensor(MultimaticEntity, BinarySensorEntity):
     """Binary sensor for holiday mode."""
@@ -591,3 +606,8 @@ def available(self) -> bool:
     def entity_category(self) -> str | None:
         """Return the category of the entity, if any."""
         return ENTITY_CATEGORY_DIAGNOSTIC
+
+    @property
+    def device_class(self):
+        """Return the class of this device, from component DEVICE_CLASSES."""
+        return BinarySensorDeviceClass.RUNNING