Skip to content

Commit 255beaf

Browse files
authored
Add connect/disconnect callbacks to lamarzocco (#144011)
1 parent e267900 commit 255beaf

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

homeassistant/components/lamarzocco/coordinator.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -97,14 +97,15 @@ async def _internal_async_update_data(self) -> None:
9797
self.config_entry.async_create_background_task(
9898
hass=self.hass,
9999
target=self.device.connect_dashboard_websocket(
100-
update_callback=lambda _: self.async_set_updated_data(None)
100+
update_callback=lambda _: self.async_set_updated_data(None),
101+
connect_callback=self.async_update_listeners,
102+
disconnect_callback=self.async_update_listeners,
101103
),
102104
name="lm_websocket_task",
103105
)
104106

105107
async def websocket_close(_: Any | None = None) -> None:
106-
if self.device.websocket.connected:
107-
await self.device.websocket.disconnect()
108+
await self.device.websocket.disconnect()
108109

109110
self.config_entry.async_on_unload(
110111
self.hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, websocket_close)

0 commit comments

Comments
 (0)