Implement energy-reset functions

Author: bouwew

plugwise_usb/__init__.py

@@ -10,7 +10,7 @@
 from collections.abc import Callable, Coroutine
 from functools import wraps
 import logging
-from typing import Any, TypeVar, cast, Final
+from typing import Any, Final, TypeVar, cast
 
 from .api import NodeEvent, PlugwiseNode, StickEvent
 from .connection import StickController
@@ -210,6 +210,20 @@ async def set_accept_join_request(self, state: bool) -> bool:
             raise NodeError(f"Failed setting accept joining: {exc}") from exc
         return True
 
+    async def energy_reset_request(self, mac: str) -> bool:
+        """Send an energy-reset request to a Node."""
+        try:
+            await self._network.energy_reset_request(mac)
+        except (MessageError, NodeError) as exc:
+            raise NodeError(f"{exc}") from exc
+
+        # Follow up by an energy-intervals (re)set
+        result = await self.set_energy_intervals(mac, 60, 0)
+        if result:
+            return True
+
+        return False
+
     async def set_energy_intervals(
         self, mac: str, cons_interval: int, prod_interval: int
     ) -> bool:
@@ -295,7 +309,7 @@ async def connect(self, port: str | None = None) -> None:
                 "Unable to connect. " +
                 "Path to USB-Stick is not defined, set port property first"
             )
-        
+
         await self._controller.connect_to_stick(
             self._port,
         )

plugwise_usb/network/__init__.py

@@ -12,11 +12,12 @@
 
 from ..api import NodeEvent, NodeType, PlugwiseNode, StickEvent
 from ..connection import StickController
-from ..constants import UTF8
+from ..constants import UTC, UTF8
 from ..exceptions import CacheError, MessageError, NodeError, StickError, StickTimeout
 from ..messages.requests import (
-    CirclePlusAllowJoiningRequest,
+    CircleClockSetRequest,
     CircleMeasureIntervalRequest,
+    CirclePlusAllowJoiningRequest,
     NodePingRequest,
 )
 from ..messages.responses import (
@@ -541,6 +542,23 @@ async def allow_join_requests(self, state: bool) -> None:
         _LOGGER.debug("Sent AllowJoiningRequest to Circle+ with state=%s", state)
         self.accept_join_request = state
 
+    async def energy_reset_request(self, mac: str) -> None:
+        """Send an energy-reset to a Node."""
+        request = CircleClockSetRequest(
+            self._send,
+            self._mac_in_bytes,
+            datetime.now(tz=UTC),
+            self._node_protocols.max,
+            True,
+        )
+        if (response := await request.send()) is None:
+            raise NodeError(f"Energy-reset for {mac} failed")
+
+        if response.ack_id != NodeResponseType.CLOCK_ACCEPTED:
+            raise MessageError(
+                f"Unknown NodeResponseType '{response.response_type.name}' received"
+            )
+
     async def set_energy_intervals(
         self, mac: str, consumption: int, production: int
     ) -> None: