From 6f991e20ebe94523e6dd8801284ccd243424d9be Mon Sep 17 00:00:00 2001
From: David <139538558+io-debug@users.noreply.github.com>
Date: Fri, 20 Oct 2023 13:25:10 +0200
Subject: [PATCH] Added data update after set_peak shaving (#40)

---
 custom_components/senec/service.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/custom_components/senec/service.py b/custom_components/senec/service.py
index bac313c..74d503b 100644
--- a/custom_components/senec/service.py
+++ b/custom_components/senec/service.py
@@ -1,6 +1,8 @@
 """ Services for SENEC Device"""
 
 from datetime import datetime
+from homeassistant.util import slugify
+from homeassistant.config_entries import ConfigEntry
 
 
 class SenecService():
@@ -26,11 +28,8 @@ async def set_peakshaving(self, call):
             new_peak_shaving = {"mode": mode, "capacity": capacity, "end_time": end_time}
             await self._coordinator.senec.set_peak_shaving(new_peak_shaving)
 
-            # Force update
-            # registry = entity_registry.async_get(self._hass)
-            # peakshaving_mode_key = f"sensor.{slugify(ConfigEntry.title)}_peakshaving_mode".lower()
-            # entity = registry.async_get(peakshaving_mode_key)
-            # entity.async_schedule_update_ha_state(force_refresh=True)
+            # Force update of data
+            await self._coordinator.async_refresh()
 
             return True
         except ValueError: