evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use enums in zwave (#62131)

Browse source 
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
Download patch file Download diff file Expand all files Collapse all files

8
homeassistant/components/zwave/sensor.py
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