Use enums in zwave (#62131)

Co-authored-by: Franck Nijhof <git@frenck.dev>
This commit is contained in:
Robert Hillis 2021-12-19 07:30:16 -05:00 committed by GitHub
parent abb36ff45f
commit 867cbeedb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 7 deletions

View File

@ -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

View File

@ -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):

View File

@ -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