Clear energy_collection from cache when resetting energy-logging

Author: bouwew

plugwise_usb/network/__init__.py

@@ -14,6 +14,7 @@
 from ..connection import StickController
 from ..constants import ENERGY_NODE_TYPES, UTF8
 from ..exceptions import CacheError, MessageError, NodeError, StickError, StickTimeout
+from ..node.helpers.cache import NodeCache
 from ..helpers.util import validate_mac
 from ..messages.requests import (
     CircleClockSetRequest,
@@ -34,6 +35,7 @@
 from ..nodes import get_plugwise_node
 from .registry import StickNetworkRegister
 
+CACHE_ENERGY_COLLECTION = "energy_collection"
 _LOGGER = logging.getLogger(__name__)
 # endregion
 
@@ -562,6 +564,11 @@ async def energy_reset_request(self, mac: str) -> None:
                 f"Unexpected NodeResponseType {response.ack_id!r} received as response to CircleClockSetRequest"
             )
 
+        # Clear the cached energy_collection
+        if self._cache_enabled:
+            node_cache = NodeCache(mac, "")
+            node_cache.remove_state(CACHE_ENERGY_COLLECTION)
+
     async def set_energy_intervals(
         self, mac: str, consumption: int, production: int
     ) -> None: