From f7bc456bd24780dee2741c3cf7021edafdecf638 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Sat, 22 May 2021 18:17:35 +0200 Subject: [PATCH] Define sensor entity attributes as class variables (#50942) --- homeassistant/components/sensor/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/sensor/__init__.py b/homeassistant/components/sensor/__init__.py index a0532a65cf3..2857a3cc5c2 100644 --- a/homeassistant/components/sensor/__init__.py +++ b/homeassistant/components/sensor/__init__.py @@ -96,15 +96,18 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: class SensorEntity(Entity): """Base class for sensor entities.""" + _attr_state_class: str | None = None + _attr_last_reset: datetime | None = None + @property def state_class(self) -> str | None: """Return the state class of this entity, from STATE_CLASSES, if any.""" - return None + return self._attr_state_class @property def last_reset(self) -> datetime | None: """Return the time when the sensor was last reset, if any.""" - return None + return self._attr_last_reset @property def capability_attributes(self) -> Mapping[str, Any] | None: