From 5cc5813bd6608f4413b625190826660681ff142e Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Tue, 25 Jan 2022 18:36:22 -1000 Subject: [PATCH] Add state class to hunterdouglas_powerview (#64936) --- .../hunterdouglas_powerview/sensor.py | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/homeassistant/components/hunterdouglas_powerview/sensor.py b/homeassistant/components/hunterdouglas_powerview/sensor.py index 641a9dfcc91..43e438041f2 100644 --- a/homeassistant/components/hunterdouglas_powerview/sensor.py +++ b/homeassistant/components/hunterdouglas_powerview/sensor.py @@ -1,7 +1,11 @@ """Support for hunterdouglass_powerview sensors.""" from aiopvapi.resources.shade import factory as PvShade -from homeassistant.components.sensor import SensorDeviceClass, SensorEntity +from homeassistant.components.sensor import ( + SensorDeviceClass, + SensorEntity, + SensorStateClass, +) from homeassistant.config_entries import ConfigEntry from homeassistant.const import PERCENTAGE from homeassistant.core import HomeAssistant, callback @@ -55,22 +59,15 @@ class PowerViewShadeBatterySensor(ShadeEntity, SensorEntity): """Representation of an shade battery charge sensor.""" _attr_entity_category = EntityCategory.DIAGNOSTIC - - @property - def native_unit_of_measurement(self): - """Return the unit of measurement.""" - return PERCENTAGE + _attr_native_unit_of_measurement = PERCENTAGE + _attr_device_class = SensorDeviceClass.BATTERY + _attr_state_class = SensorStateClass.MEASUREMENT @property def name(self): """Name of the shade battery.""" return f"{self._shade_name} Battery" - @property - def device_class(self): - """Shade battery Class.""" - return SensorDeviceClass.BATTERY - @property def unique_id(self): """Shade battery Uniqueid."""