Use new DeviceClass enums in iaqualink (#61805)
Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
parent
c02aae58fb
commit
2c26eae9b2
2 changed files with 5 additions and 5 deletions
|
@ -1,7 +1,7 @@
|
||||||
"""Support for Aqualink temperature sensors."""
|
"""Support for Aqualink temperature sensors."""
|
||||||
from homeassistant.components.binary_sensor import (
|
from homeassistant.components.binary_sensor import (
|
||||||
DEVICE_CLASS_COLD,
|
|
||||||
DOMAIN,
|
DOMAIN,
|
||||||
|
BinarySensorDeviceClass,
|
||||||
BinarySensorEntity,
|
BinarySensorEntity,
|
||||||
)
|
)
|
||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
|
@ -40,5 +40,5 @@ class HassAqualinkBinarySensor(AqualinkEntity, BinarySensorEntity):
|
||||||
def device_class(self) -> str:
|
def device_class(self) -> str:
|
||||||
"""Return the class of the binary sensor."""
|
"""Return the class of the binary sensor."""
|
||||||
if self.name == "Freeze Protection":
|
if self.name == "Freeze Protection":
|
||||||
return DEVICE_CLASS_COLD
|
return BinarySensorDeviceClass.COLD
|
||||||
return None
|
return None
|
||||||
|
|
|
@ -1,9 +1,9 @@
|
||||||
"""Support for Aqualink temperature sensors."""
|
"""Support for Aqualink temperature sensors."""
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from homeassistant.components.sensor import DOMAIN, SensorEntity
|
from homeassistant.components.sensor import DOMAIN, SensorDeviceClass, SensorEntity
|
||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
from homeassistant.const import DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, TEMP_FAHRENHEIT
|
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from . import AqualinkEntity
|
from . import AqualinkEntity
|
||||||
|
@ -55,5 +55,5 @@ class HassAqualinkSensor(AqualinkEntity, SensorEntity):
|
||||||
def device_class(self) -> str | None:
|
def device_class(self) -> str | None:
|
||||||
"""Return the class of the sensor."""
|
"""Return the class of the sensor."""
|
||||||
if self.dev.name.endswith("_temp"):
|
if self.dev.name.endswith("_temp"):
|
||||||
return DEVICE_CLASS_TEMPERATURE
|
return SensorDeviceClass.TEMPERATURE
|
||||||
return None
|
return None
|
||||||
|
|
Loading…
Add table
Reference in a new issue