Use enums in wolflink (#61988)

This commit is contained in:
Robert Hillis 2021-12-16 03:10:14 -05:00 committed by GitHub
parent f691b0a1da
commit 806366a0c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -9,14 +9,8 @@ from wolf_smartset.models import (
Temperature, Temperature,
) )
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import ( from homeassistant.const import PRESSURE_BAR, TEMP_CELSIUS, TIME_HOURS
DEVICE_CLASS_PRESSURE,
DEVICE_CLASS_TEMPERATURE,
PRESSURE_BAR,
TEMP_CELSIUS,
TIME_HOURS,
)
from homeassistant.helpers.update_coordinator import CoordinatorEntity from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import COORDINATOR, DEVICE_ID, DOMAIN, PARAMETERS, STATES from .const import COORDINATOR, DEVICE_ID, DOMAIN, PARAMETERS, STATES
@ -106,7 +100,7 @@ class WolfLinkTemperature(WolfLinkSensor):
@property @property
def device_class(self): def device_class(self):
"""Return the device_class.""" """Return the device_class."""
return DEVICE_CLASS_TEMPERATURE return SensorDeviceClass.TEMPERATURE
@property @property
def native_unit_of_measurement(self): def native_unit_of_measurement(self):
@ -120,7 +114,7 @@ class WolfLinkPressure(WolfLinkSensor):
@property @property
def device_class(self): def device_class(self):
"""Return the device_class.""" """Return the device_class."""
return DEVICE_CLASS_PRESSURE return SensorDeviceClass.PRESSURE
@property @property
def native_unit_of_measurement(self): def native_unit_of_measurement(self):