Add additional logging

Author: andrew-codechimp

custom_components/battery_notes/device.py

@@ -97,6 +97,13 @@ async def async_setup(self) -> bool:
             device_class = entity.device_class or entity.original_device_class
             if device_class == SensorDeviceClass.BATTERY and entity.unit_of_measurement == PERCENTAGE:
                 self.wrapped_battery = entity
+            else:
+                _LOGGER.debug(
+                    "%s is not a battery entity device_class: %s unit_of_measurement: %s",
+                    source_entity_id,
+                    device_class,
+                    entity.unit_of_measurement,
+                )
 
             if entity.device_id:
                 device_entry = device_registry.async_get(entity.device_id)
@@ -128,6 +135,7 @@ async def async_setup(self) -> bool:
                     continue
 
                 self.wrapped_battery = entity_registry.async_get(entity.entity_id)
+                break
 
             device_entry = device_registry.async_get(device_id)
             if device_entry:

custom_components/battery_notes/sensor.py

@@ -298,6 +298,12 @@ def __init__(
             self._attr_translation_placeholders = {"device_name": ""}
             self.entity_id = f"sensor.{coordinator.device_name.lower()}_{description.key}"
 
+        _LOGGER.debug(
+            "Setting up %s with wrapped battery %s",
+            self.entity_id,
+            self.coordinator.wrapped_battery.entity_id
+        )
+
         self.entity_description = description
         self._attr_unique_id = unique_id
         self.device = device