File tree 1 file changed +4
-3
lines changed
homeassistant/components/lamarzocco
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -97,14 +97,15 @@ async def _internal_async_update_data(self) -> None:
97
97
self .config_entry .async_create_background_task (
98
98
hass = self .hass ,
99
99
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 ,
101
103
),
102
104
name = "lm_websocket_task" ,
103
105
)
104
106
105
107
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 ()
108
109
109
110
self .config_entry .async_on_unload (
110
111
self .hass .bus .async_listen_once (EVENT_HOMEASSISTANT_STOP , websocket_close )
You can’t perform that action at this time.
0 commit comments