Skip to content

Commit bd80b95

Browse files
committed
Simplify energy_log_update()
improve docstrings
1 parent 6ce576d commit bd80b95

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

plugwise_usb/nodes/circle.py

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
365365

366366
# Always request last energy log records at initial startup
367367
if not self._last_energy_log_requested:
368-
self._last_energy_log_requested, _ = await self.energy_log_update(
368+
self._last_energy_log_requested = await self.energy_log_update(
369369
self._current_log_address
370370
)
371371

@@ -379,7 +379,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
379379
return None
380380

381381
# Try collecting energy-stats for _current_log_address
382-
result, _ = await self.energy_log_update(self._current_log_address)
382+
result = await self.energy_log_update(self._current_log_address)
383383
if not result:
384384
_LOGGER.debug(
385385
"async_energy_update | %s | Log rollover | energy_log_update failed",
@@ -393,7 +393,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
393393
_prev_log_address, _ = calc_log_address(
394394
self._current_log_address, 1, -4
395395
)
396-
result, _ = await self.energy_log_update(_prev_log_address)
396+
result = await self.energy_log_update(_prev_log_address)
397397
if not result:
398398
_LOGGER.debug(
399399
"async_energy_update | %s | Log rollover | energy_log_update %s failed",
@@ -414,7 +414,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
414414
return self._energy_counters.energy_statistics
415415

416416
if len(missing_addresses) == 1:
417-
result, _ = await self.energy_log_update(missing_addresses[0])
417+
result = await self.energy_log_update(missing_addresses[0])
418418
if result:
419419
await self.power_update()
420420
_LOGGER.debug(
@@ -464,11 +464,11 @@ async def _get_initial_energy_logs(self) -> None:
464464
total_addresses = 11
465465
log_address = self._current_log_address
466466
while total_addresses > 0:
467-
result, empty_log = await self.energy_log_update(log_address)
468-
if result and empty_log:
467+
result = await self.energy_log_update(log_address)
468+
if not result:
469469
# Handle case with None-data in all address slots
470470
_LOGGER.debug(
471-
"Energy None-data collected from log address %s, stopping collection",
471+
"Energy None-data of outdated data collected from log address %s, stopping collection",
472472
log_address,
473473
)
474474
break
@@ -507,28 +507,28 @@ async def get_missing_energy_logs(self) -> None:
507507
if self._cache_enabled:
508508
await self._energy_log_records_save_to_cache()
509509

510-
async def energy_log_update(self, address: int | None) -> tuple[bool, bool]:
510+
async def energy_log_update(self, address: int | None) -> bool:
511511
"""Request energy log statistics from node. Returns true if successful."""
512-
empty_log = False
513512
result = False
514513
if address is None:
515-
return result, empty_log
514+
return result
516515

517516
_LOGGER.debug(
518-
"Request of energy log at address %s for node %s",
517+
"Requesting EnergyLogs from node %s address %s",
519518
str(address),
520519
self.name,
521520
)
522521
request = CircleEnergyLogsRequest(self._send, self._mac_in_bytes, address)
523522
if (response := await request.send()) is None:
524523
_LOGGER.debug(
525-
"Retrieving of energy log at address %s for node %s failed",
526-
str(address),
524+
"Retrieving EnergyLogs data from node %s failed",
527525
self._mac_in_str,
528526
)
529-
return result, empty_log
527+
return result
530528

531-
_LOGGER.debug("EnergyLogs data from %s, address=%s", self._mac_in_str, address)
529+
_LOGGER.debug(
530+
"EnergyLogs data from node %s, address=%s", self._mac_in_str, address
531+
)
532532
await self._available_update_state(True, response.timestamp)
533533
energy_record_update = False
534534

@@ -543,12 +543,10 @@ async def energy_log_update(self, address: int | None) -> tuple[bool, bool]:
543543
)
544544
if log_timestamp is None or log_pulses is None:
545545
self._energy_counters.add_empty_log(response.log_address, _slot)
546-
empty_log = True
547546
continue
548547
elif not self._check_timestamp_is_recent(address, _slot, log_timestamp):
549548
# Don't store an old log-record, store am empty record instead
550549
self._energy_counters.add_empty_log(response.log_address, _slot)
551-
empty_log = True
552550
continue
553551

554552
if await self._energy_log_record_update_state(
@@ -578,7 +576,7 @@ async def energy_log_update(self, address: int | None) -> tuple[bool, bool]:
578576
)
579577
await self.save_cache()
580578

581-
return result, empty_log
579+
return result
582580

583581
def _check_timestamp_is_recent(self, address, slot, timestamp) -> bool:
584582
"""Check if the timestamp of the received log-record is recent.

0 commit comments

Comments
 (0)