Optimized code by ChatGPT

Author: bouwew

plugwise_usb/nodes/circle.py

@@ -656,25 +656,24 @@ async def _energy_log_records_load_from_cache(self) -> bool:
             return False
 
         collected_logs = _collect_records(cache_data)
-        # Sort and prune the records loaded from cache
-        sorted_logs: dict[int, dict[int, tuple[datetime, int]]] = {}
+
+        # Cutoff timestamp for filtering
         skip_before = datetime.now(tz=UTC) - timedelta(hours=MAX_LOG_HOURS)
-        sorted_addresses = sorted(collected_logs.keys(), reverse=True)
-        for address in sorted_addresses:
-            sorted_slots = sorted(collected_logs[address].keys(), reverse=True)
-            for slot in sorted_slots:
-                if collected_logs[address][slot][0] > skip_before:
-                    if sorted_logs.get(address) is None:
-                        sorted_logs[address] = {}
-                    sorted_logs[address][slot] = collected_logs[address][slot]
-
-        for address, data in sorted_logs.items():
-            for slot, pulse_data in data.items():
+
+        # Iterate in reverse sorted order directly
+        for address in sorted(collected_logs, reverse=True):
+            slots = collected_logs[address]
+            filtered_slots = {
+                slot: data
+                for slot, data in sorted(slots.items(), reverse=True)
+                if data[0] > skip_before
+            }
+            for slot, (timestamp, pulses) in filtered_slots.items():
                 self._energy_counters.add_pulse_log(
                     address=address,
                     slot=slot,
-                    pulses=pulse_data[1],
-                    timestamp=pulse_data[0],
+                    pulses=pulses,
+                    timestamp=timestamp,
                     import_only=True,
                 )