Use EntityCategory enum instead of str

kvj · Mar 30, 2022 · c9c0531 · c9c0531
1 parent aa9deaf
commit c9c0531
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 14 deletions.
diff --git a/custom_components/nuki_ng/binary_sensor.py b/custom_components/nuki_ng/binary_sensor.py
@@ -1,4 +1,5 @@
 from homeassistant.components.binary_sensor import BinarySensorEntity
+from homeassistant.helpers.entity import EntityCategory
 
 import logging
 
@@ -48,7 +49,7 @@ def device_class(self) -> str:
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class BatteryCharging(NukiEntity, BinarySensorEntity):
@@ -64,7 +65,7 @@ def is_on(self) -> bool:
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class KeypadBatteryLow(NukiEntity, BinarySensorEntity):
@@ -83,7 +84,7 @@ def device_class(self) -> str:
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class RingAction(NukiEntity, BinarySensorEntity):
@@ -104,7 +105,7 @@ def extra_state_attributes(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class LockState(NukiEntity, BinarySensorEntity):
@@ -161,7 +162,7 @@ def extra_state_attributes(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class BridgeCallbackSet(NukiBridge, BinarySensorEntity):
@@ -186,4 +187,4 @@ def extra_state_attributes(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
diff --git a/custom_components/nuki_ng/sensor.py b/custom_components/nuki_ng/sensor.py
@@ -1,4 +1,5 @@
 from homeassistant.components.sensor import SensorEntity
+from homeassistant.helpers.entity import EntityCategory
 
 import logging
 
@@ -53,7 +54,7 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class LockState(NukiEntity, SensorEntity):
@@ -69,7 +70,7 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class RSSI(NukiEntity, SensorEntity):
@@ -94,7 +95,7 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class DoorSensorState(NukiEntity, SensorEntity):
@@ -110,7 +111,7 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class DoorSecurityState(NukiEntity, SensorEntity):
@@ -160,7 +161,7 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class BridgeVersion(NukiBridge, SensorEntity):
@@ -176,7 +177,7 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
 
 
 class LockVersion(NukiEntity, SensorEntity):
@@ -192,4 +193,4 @@ def state(self):
 
     @property
     def entity_category(self):
-        return "diagnostic"
+        return EntityCategory.DIAGNOSTIC
diff --git a/custom_components/nuki_ng/switch.py b/custom_components/nuki_ng/switch.py
@@ -1,4 +1,5 @@
 from homeassistant.components.switch import SwitchEntity
+from homeassistant.helpers.entity import EntityCategory
 
 import logging
 
@@ -86,4 +87,4 @@ def device_info(self):
 
     @property
     def entity_category(self):
-        return "config"
+        return EntityCategory.CONFIG