Initil after Upgrade

custom_components/powercalc/sensors/daily_energy.py

@@ -75,7 +75,7 @@ DAILY_FIXED_ENERGY_SCHEMA = vol.Schema(
 _LOGGER = logging.getLogger(__name__)
 
 
-async def create_daily_fixed_energy_sensor(
+def create_daily_fixed_energy_sensor(
     hass: HomeAssistant,
     sensor_config: ConfigType,
     source_entity: SourceEntity | None = None,
@@ -134,7 +134,10 @@ async def create_daily_fixed_energy_power_sensor(
         return None
 
     power_value: float = mode_config.get(CONF_VALUE)  # type: ignore
-    if mode_config.get(CONF_UNIT_OF_MEASUREMENT) == UnitOfEnergy.KILO_WATT_HOUR and not isinstance(power_value, Template):
+    if mode_config.get(CONF_UNIT_OF_MEASUREMENT) == UnitOfEnergy.KILO_WATT_HOUR and not isinstance(
+        power_value,
+        Template,
+    ):
         power_value = power_value * 1000 / 24
 
     power_sensor_config = sensor_config.copy()
@@ -244,7 +247,9 @@ class DailyEnergySensor(RestoreEntity, SensorEntity, EnergySensor):
             self.hass,
             refresh,
             timedelta(seconds=self._update_frequency),
+            cancel_on_shutdown=True,
         )
+        self.async_on_remove(self._update_timer_removal)
 
     def calculate_delta(self, elapsed_seconds: int = 0) -> Decimal:
         if self._last_delta_calculate is None:
@@ -267,7 +272,11 @@ class DailyEnergySensor(RestoreEntity, SensorEntity, EnergySensor):
                 )
                 return Decimal(0)
 
-        wh_per_day = value * (self._on_time.total_seconds() / 3600) if self._user_unit_of_measurement == UnitOfPower.WATT else value * 1000
+        wh_per_day = (
+            value * (self._on_time.total_seconds() / 3600)
+            if self._user_unit_of_measurement == UnitOfPower.WATT
+            else value * 1000
+        )
 
         # Convert Wh to the native measurement unit
         energy_per_day = wh_per_day