From 320df10a26c3c4cd9ffb6011c24e343576c65478 Mon Sep 17 00:00:00 2001 From: Stefan Agner Date: Fri, 4 Feb 2022 11:40:38 +0100 Subject: [PATCH] Use _attr_last_reset to set last_reset (#65648) --- homeassistant/components/iotawatt/sensor.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/iotawatt/sensor.py b/homeassistant/components/iotawatt/sensor.py index acc4577fa8a..c3c173f778e 100644 --- a/homeassistant/components/iotawatt/sensor.py +++ b/homeassistant/components/iotawatt/sensor.py @@ -8,7 +8,6 @@ import logging from iotawattpy.sensor import Sensor from homeassistant.components.sensor import ( - ATTR_LAST_RESET, SensorDeviceClass, SensorEntity, SensorEntityDescription, @@ -211,6 +210,12 @@ class IotaWattSensor(update_coordinator.CoordinatorEntity, SensorEntity): else: self.hass.async_create_task(self.async_remove()) return + + if (begin := self._sensor_data.getBegin()) and ( + last_reset := dt.parse_datetime(begin) + ): + self._attr_last_reset = last_reset + super()._handle_coordinator_update() @property @@ -220,8 +225,6 @@ class IotaWattSensor(update_coordinator.CoordinatorEntity, SensorEntity): attrs = {"type": data.getType()} if attrs["type"] == "Input": attrs["channel"] = data.getChannel() - if (begin := data.getBegin()) and (last_reset := dt.parse_datetime(begin)): - attrs[ATTR_LAST_RESET] = last_reset.isoformat() return attrs