From 1d59f5c4b71db88dd2feba53a6bca1563e3a9822 Mon Sep 17 00:00:00 2001 From: Jesse Sopel Date: Tue, 5 Nov 2024 20:31:30 -0500 Subject: [PATCH] fix: Restore mug callback --- custom_components/ember_mug/coordinator.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/custom_components/ember_mug/coordinator.py b/custom_components/ember_mug/coordinator.py index 7ccc37d..f0731e9 100644 --- a/custom_components/ember_mug/coordinator.py +++ b/custom_components/ember_mug/coordinator.py @@ -78,6 +78,10 @@ async def _async_setup(self) -> None: f"An error occurred updating {self.mug.model_name}: {e=}", ) from e + self.mug.register_callback( + self._async_handle_callback, + ) + async def _async_update_data(self) -> MugData: """Poll the device.""" _LOGGER.debug("Updating") @@ -166,6 +170,12 @@ def handle_bluetooth_event( self.mug.ble_event_callback(service_info.device, service_info.advertisement) self.hass.loop.create_task(close_stale_connections(service_info.device)) + @callback + def _async_handle_callback(self, mug_data: MugData) -> None: + """Handle a Bluetooth event.""" + _LOGGER.debug("Callback called in Home Assistant") + self.async_set_updated_data(mug_data) + def refresh_from_mug(self) -> None: """Update stored data from mug data and trigger entities.""" self.async_set_updated_data(self.mug.data)