Battery low attributes (#3577)

Author: andrew-codechimp

custom_components/battery_notes/binary_sensor.py

@@ -66,10 +66,14 @@
 from . import PLATFORMS
 from .common import validate_is_float
 from .const import (
+    ATTR_BATTERY_LAST_REPLACED,
     ATTR_BATTERY_LOW_THRESHOLD,
     ATTR_BATTERY_QUANTITY,
     ATTR_BATTERY_TYPE,
     ATTR_BATTERY_TYPE_AND_QUANTITY,
+    ATTR_DEVICE_ID,
+    ATTR_DEVICE_NAME,
+    ATTR_SOURCE_ENTITY_ID,
     CONF_SOURCE_ENTITY_ID,
     DOMAIN,
 )
@@ -330,12 +334,27 @@ class BatteryNotesBatteryLowBaseSensor(
 ):
     """Low battery binary sensor base."""
 
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        coordinator: BatteryNotesCoordinator,
+    ):
+        """Initialize the low battery binary sensor."""
+
+        super().__init__(coordinator=coordinator)
+
+        self.enable_replaced = hass.data[MY_KEY].enable_replaced
+
     _unrecorded_attributes = frozenset(
         {
             ATTR_BATTERY_LOW_THRESHOLD,
             ATTR_BATTERY_QUANTITY,
             ATTR_BATTERY_TYPE,
             ATTR_BATTERY_TYPE_AND_QUANTITY,
+            ATTR_BATTERY_LAST_REPLACED,
+            ATTR_DEVICE_ID,
+            ATTR_SOURCE_ENTITY_ID,
+            ATTR_DEVICE_NAME,
         }
     )
 
@@ -351,6 +370,14 @@ def extra_state_attributes(self) -> dict[str, Any] | None:
             ATTR_BATTERY_TYPE_AND_QUANTITY: self.coordinator.battery_type_and_quantity,
         }
 
+        if self.enable_replaced:
+            attrs[ATTR_BATTERY_LAST_REPLACED] = self.coordinator.last_replaced
+
+        # Other attributes that should follow battery, attribute list is unsorted
+        attrs[ATTR_DEVICE_ID] = self.coordinator.device_id or ""
+        attrs[ATTR_SOURCE_ENTITY_ID] = self.coordinator.source_entity_id or ""
+        attrs[ATTR_DEVICE_NAME] = self.coordinator.device_name
+
         super_attrs = super().extra_state_attributes
         if super_attrs:
             attrs.update(super_attrs)
@@ -382,7 +409,7 @@ def __init__(
         self._attr_unique_id = unique_id
         self._template_attrs: dict[Template, list[_TemplateAttribute]] = {}
 
-        super().__init__(coordinator=coordinator)
+        super().__init__(hass=hass, coordinator=coordinator)
 
         if coordinator.device_id and (
             device_entry := device_registry.async_get(coordinator.device_id)
@@ -612,7 +639,7 @@ def __init__(
         self.entity_description = description
         self._attr_unique_id = unique_id
 
-        super().__init__(coordinator=coordinator)
+        super().__init__(hass=hass, coordinator=coordinator)
 
         if coordinator.device_id and (
             device_entry := device_registry.async_get(coordinator.device_id)
@@ -707,7 +734,7 @@ def __init__(
         self.entity_description = description
         self._attr_unique_id = unique_id
 
-        super().__init__(coordinator=coordinator)
+        super().__init__(hass=hass, coordinator=coordinator)
 
         if coordinator.device_id and (
             device_entry := device_registry.async_get(coordinator.device_id)