Use enums in zwave (#62131)

Co-authored-by: Franck Nijhof <git@frenck.dev>

homeassistant/components/zwave/binary_sensor.py

@@ -59,7 +59,7 @@ class ZWaveBinarySensor(BinarySensorEntity, ZWaveDeviceEntity):
 
     @property
     def device_class(self):
-        """Return the class of this sensor, from DEVICE_CLASSES."""
+        """Return the class of this sensor, from BinarySensorDeviceClass."""
         return self._sensor_type
 
 

homeassistant/components/zwave/cover.py

@@ -6,6 +6,7 @@ from homeassistant.components.cover import (
     DOMAIN,
     SUPPORT_CLOSE,
     SUPPORT_OPEN,
+    CoverDeviceClass,
     CoverEntity,
 )
 from homeassistant.core import callback
@@ -156,8 +157,8 @@ class ZwaveGarageDoorBase(ZWaveDeviceEntity, CoverEntity):
 
     @property
     def device_class(self):
-        """Return the class of this device, from component DEVICE_CLASSES."""
-        return "garage"
+        """Return the class of this device, from CoverDeviceClass."""
+        return CoverDeviceClass.GARAGE
 
     @property
     def supported_features(self):

homeassistant/components/zwave/sensor.py

@@ -1,6 +1,6 @@
 """Support for Z-Wave sensors."""
-from homeassistant.components.sensor import DEVICE_CLASS_BATTERY, DOMAIN, SensorEntity
-from homeassistant.const import DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
+from homeassistant.components.sensor import DOMAIN, SensorDeviceClass, SensorEntity
+from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
 from homeassistant.core import callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
 
@@ -83,7 +83,7 @@ class ZWaveMultilevelSensor(ZWaveSensor):
     def device_class(self):
         """Return the class of this device."""
         if self._units in ["C", "F"]:
-            return DEVICE_CLASS_TEMPERATURE
+            return SensorDeviceClass.TEMPERATURE
         return None
 
     @property
@@ -115,4 +115,4 @@ class ZWaveBatterySensor(ZWaveSensor):
     @property
     def device_class(self):
         """Return the class of this device."""
-        return DEVICE_CLASS_BATTERY
+        return SensorDeviceClass.BATTERY