diff --git a/homeassistant/components/hunterdouglas_powerview/cover.py b/homeassistant/components/hunterdouglas_powerview/cover.py index 22636b7e3c4..f8644b15a5d 100644 --- a/homeassistant/components/hunterdouglas_powerview/cover.py +++ b/homeassistant/components/hunterdouglas_powerview/cover.py @@ -14,11 +14,11 @@ import async_timeout from homeassistant.components.cover import ( ATTR_POSITION, - DEVICE_CLASS_SHADE, SUPPORT_CLOSE, SUPPORT_OPEN, SUPPORT_SET_POSITION, SUPPORT_STOP, + CoverDeviceClass, CoverEntity, ) from homeassistant.core import callback @@ -145,7 +145,7 @@ class PowerViewShade(ShadeEntity, CoverEntity): @property def device_class(self): """Return device class.""" - return DEVICE_CLASS_SHADE + return CoverDeviceClass.SHADE @property def name(self): diff --git a/homeassistant/components/hunterdouglas_powerview/sensor.py b/homeassistant/components/hunterdouglas_powerview/sensor.py index c789dd150da..1251ee778dc 100644 --- a/homeassistant/components/hunterdouglas_powerview/sensor.py +++ b/homeassistant/components/hunterdouglas_powerview/sensor.py @@ -1,13 +1,10 @@ """Support for hunterdouglass_powerview sensors.""" from aiopvapi.resources.shade import factory as PvShade -from homeassistant.components.sensor import SensorEntity -from homeassistant.const import ( - DEVICE_CLASS_BATTERY, - ENTITY_CATEGORY_DIAGNOSTIC, - PERCENTAGE, -) +from homeassistant.components.sensor import SensorDeviceClass, SensorEntity +from homeassistant.const import PERCENTAGE from homeassistant.core import callback +from homeassistant.helpers.entity import EntityCategory from .const import ( COORDINATOR, @@ -53,7 +50,7 @@ async def async_setup_entry(hass, entry, async_add_entities): class PowerViewShadeBatterySensor(ShadeEntity, SensorEntity): """Representation of an shade battery charge sensor.""" - _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC + _attr_entity_category = EntityCategory.DIAGNOSTIC @property def native_unit_of_measurement(self): @@ -68,7 +65,7 @@ class PowerViewShadeBatterySensor(ShadeEntity, SensorEntity): @property def device_class(self): """Shade battery Class.""" - return DEVICE_CLASS_BATTERY + return SensorDeviceClass.BATTERY @property def unique_id(self):