Add connect/disconnect callbacks to lamarzocco (#144011)

This commit is contained in:
Josef Zweck 2025-05-01 22:29:44 +02:00 committed by GitHub
parent e2679004a1
commit 255beafe08
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -97,14 +97,15 @@ class LaMarzoccoConfigUpdateCoordinator(LaMarzoccoUpdateCoordinator):
self.config_entry.async_create_background_task( self.config_entry.async_create_background_task(
hass=self.hass, hass=self.hass,
target=self.device.connect_dashboard_websocket( target=self.device.connect_dashboard_websocket(
update_callback=lambda _: self.async_set_updated_data(None) update_callback=lambda _: self.async_set_updated_data(None),
connect_callback=self.async_update_listeners,
disconnect_callback=self.async_update_listeners,
), ),
name="lm_websocket_task", name="lm_websocket_task",
) )
async def websocket_close(_: Any | None = None) -> None: async def websocket_close(_: Any | None = None) -> None:
if self.device.websocket.connected: await self.device.websocket.disconnect()
await self.device.websocket.disconnect()
self.config_entry.async_on_unload( self.config_entry.async_on_unload(
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, websocket_close) self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, websocket_close)