Add async_track_state_report_event

Author: andrew-codechimp

custom_components/battery_notes/sensor.py

@@ -42,8 +42,10 @@
 )
 from homeassistant.helpers.event import (
     EventStateChangedData,
+    EventStateReportedData,
     async_track_entity_registry_updated_event,
     async_track_state_change_event,
+    async_track_state_report_event,
 )
 from homeassistant.helpers.reload import async_setup_reload_service
 from homeassistant.helpers.typing import StateType
@@ -385,6 +387,25 @@ async def async_state_changed_listener(
 
         self.async_write_ha_state()
 
+    @callback
+    async def async_state_reported_listener(
+        self, event: Event[EventStateChangedData] | None = None
+    ) -> None:
+        # pylint: disable=unused-argument
+        """Handle child updates."""
+
+        if not self.coordinator.wrapped_battery:
+            return
+
+        self.coordinator.last_reported = datetime.utcnow()
+
+        _LOGGER.debug(
+            "Entity id %s has been reported.",
+            self.coordinator.wrapped_battery.entity_id,
+        )
+
+        await self.coordinator.async_request_refresh()
+
     async def _register_entity_id_change_listener(
         self,
         entity_id: str,
@@ -425,6 +446,14 @@ async def _entity_rename_listener(event: Event[er.EventEntityRegistryUpdatedData
                     )
                 )
 
+                self.async_on_remove(
+                    async_track_state_report_event(
+                        self.hass,
+                        [self.coordinator.wrapped_battery.entity_id],
+                        self.async_state_reported_listener,
+                    )
+                )
+
         @callback
         def _filter_entity_id(event_data: Mapping[str, Any]) -> bool:
             """Only dispatch the listener for update events concerning the source entity."""
@@ -451,6 +480,13 @@ async def _async_state_changed_listener(
             """Handle child updates."""
             await self.async_state_changed_listener(event)
 
+        @callback
+        async def _async_state_reported_listener(
+            event: Event[EventStateReportedData] | None = None,
+        ) -> None:
+            """Handle child updates."""
+            await self.async_state_reported_listener(event)
+
         if self.coordinator.wrapped_battery:
             self.async_on_remove(
                 async_track_state_change_event(
@@ -460,6 +496,14 @@ async def _async_state_changed_listener(
                 )
             )
 
+            self.async_on_remove(
+                async_track_state_report_event(
+                    self.hass,
+                    [self.coordinator.wrapped_battery.entity_id],
+                    _async_state_reported_listener,
+                )
+            )
+
             await self._register_entity_id_change_listener(
                 self.entity_id,
                 self.coordinator.wrapped_battery.entity_id,