Use new DeviceClass enum in home_connect (#61759)

Co-authored-by: epenet <epenet@users.noreply.github.com>
pull/61823/head
epenet 2021-12-14 19:59:47 +01:00 committed by GitHub
parent 67ddfcb218
commit 6bf41325ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions

View File

@ -7,12 +7,12 @@ import homeconnect
from homeconnect.api import HomeConnectError
from homeassistant import config_entries, core
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.const import (
ATTR_DEVICE_CLASS,
ATTR_ICON,
CONF_DEVICE,
CONF_ENTITIES,
DEVICE_CLASS_TIMESTAMP,
PERCENTAGE,
TIME_SECONDS,
)
@ -159,7 +159,7 @@ class DeviceWithPrograms(HomeConnectDevice):
device.
"""
sensors = {
"Remaining Program Time": (None, None, DEVICE_CLASS_TIMESTAMP, 1),
"Remaining Program Time": (None, None, SensorDeviceClass.TIMESTAMP, 1),
"Duration": (TIME_SECONDS, "mdi:update", None, 1),
"Program Progress": (PERCENTAGE, "mdi:progress-clock", None, 1),
}

View File

@ -3,8 +3,8 @@
from datetime import timedelta
import logging
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import CONF_ENTITIES, DEVICE_CLASS_TIMESTAMP
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import CONF_ENTITIES
import homeassistant.util.dt as dt_util
from .const import ATTR_VALUE, BSH_OPERATION_STATE, DOMAIN
@ -57,7 +57,7 @@ class HomeConnectSensor(HomeConnectEntity, SensorEntity):
if self._key not in status:
self._state = None
else:
if self.device_class == DEVICE_CLASS_TIMESTAMP:
if self.device_class == SensorDeviceClass.TIMESTAMP:
if ATTR_VALUE not in status[self._key]:
self._state = None
elif (