mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 20:27:08 +00:00
Add connect/disconnect callbacks to lamarzocco (#144011)
This commit is contained in:
parent
e2679004a1
commit
255beafe08
@ -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)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user