Skip to content

Commit 25f059f

Browse files
committed
Simplify energy_log_update()
improve docstrings
1 parent 2554d2d commit 25f059f

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
@@ -366,7 +366,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
366366

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

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

382382
# Try collecting energy-stats for _current_log_address
383-
result, _ = await self.energy_log_update(self._current_log_address)
383+
result = await self.energy_log_update(self._current_log_address)
384384
if not result:
385385
_LOGGER.debug(
386386
"async_energy_update | %s | Log rollover | energy_log_update failed",
@@ -394,7 +394,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
394394
_prev_log_address, _ = calc_log_address(
395395
self._current_log_address, 1, -4
396396
)
397-
result, _ = await self.energy_log_update(_prev_log_address)
397+
result = await self.energy_log_update(_prev_log_address)
398398
if not result:
399399
_LOGGER.debug(
400400
"async_energy_update | %s | Log rollover | energy_log_update %s failed",
@@ -415,7 +415,7 @@ async def energy_update(self) -> EnergyStatistics | None: # noqa: PLR0911 PLR09
415415
return self._energy_counters.energy_statistics
416416

417417
if len(missing_addresses) == 1:
418-
result, _ = await self.energy_log_update(missing_addresses[0])
418+
result = await self.energy_log_update(missing_addresses[0])
419419
if result:
420420
await self.power_update()
421421
_LOGGER.debug(
@@ -465,11 +465,11 @@ async def _get_initial_energy_logs(self) -> None:
465465
total_addresses = 11
466466
log_address = self._current_log_address
467467
while total_addresses > 0:
468-
result, empty_log = await self.energy_log_update(log_address)
469-
if result and empty_log:
468+
result = await self.energy_log_update(log_address)
469+
if not result:
470470
# Handle case with None-data in all address slots
471471
_LOGGER.debug(
472-
"Energy None-data collected from log address %s, stopping collection",
472+
"Energy None-data of outdated data collected from log address %s, stopping collection",
473473
log_address,
474474
)
475475
break
@@ -508,28 +508,28 @@ async def get_missing_energy_logs(self) -> None:
508508
if self._cache_enabled:
509509
await self._energy_log_records_save_to_cache()
510510

511-
async def energy_log_update(self, address: int | None) -> tuple[bool, bool]:
511+
async def energy_log_update(self, address: int | None) -> bool:
512512
"""Request energy log statistics from node. Returns true if successful."""
513-
empty_log = False
514513
result = False
515514
if address is None:
516-
return result, empty_log
515+
return result
517516

518517
_LOGGER.debug(
519-
"Request of energy log at address %s for node %s",
518+
"Requesting EnergyLogs from node %s address %s",
520519
str(address),
521520
self.name,
522521
)
523522
request = CircleEnergyLogsRequest(self._send, self._mac_in_bytes, address)
524523
if (response := await request.send()) is None:
525524
_LOGGER.debug(
526-
"Retrieving of energy log at address %s for node %s failed",
527-
str(address),
525+
"Retrieving EnergyLogs data from node %s failed",
528526
self._mac_in_str,
529527
)
530-
return result, empty_log
528+
return result
531529

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

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

555553
if await self._energy_log_record_update_state(
@@ -579,7 +577,7 @@ async def energy_log_update(self, address: int | None) -> tuple[bool, bool]:
579577
)
580578
await self.save_cache()
581579

582-
return result, empty_log
580+
return result
583581

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

0 commit comments

Comments
 (0)