Improve

Author: bouwew

plugwise_usb/__init__.py

@@ -218,8 +218,7 @@ async def energy_reset_request(self, mac: str) -> bool:
             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:
+        if (result := await self.set_energy_intervals(mac, 60, 0)):
             return True
 
         return False

plugwise_usb/network/__init__.py

@@ -6,13 +6,13 @@
 
 from asyncio import Task, create_task, gather, sleep
 from collections.abc import Callable, Coroutine
-from datetime import datetime, timedelta
+from datetime import UTC, datetime, timedelta
 import logging
 from typing import Any
 
 from ..api import NodeEvent, NodeType, PlugwiseNode, StickEvent
 from ..connection import StickController
-from ..constants import UTC, UTF8
+from ..constants import UTF8
 from ..exceptions import CacheError, MessageError, NodeError, StickError, StickTimeout
 from ..messages.requests import (
     CircleClockSetRequest,
@@ -545,7 +545,7 @@ async def allow_join_requests(self, state: bool) -> None:
     async def energy_reset_request(self, mac: str) -> None:
         """Send an energy-reset to a Node."""
         request = CircleClockSetRequest(
-            self._send,
+            self._controller.send,
             self._mac_in_bytes,
             datetime.now(tz=UTC),
             self._node_protocols.max,