diff --git a/moonraker/websockets.py b/moonraker/websockets.py index 5e2b66f..30f67cf 100644 --- a/moonraker/websockets.py +++ b/moonraker/websockets.py @@ -172,6 +172,8 @@ class WebsocketManager: # Register events self.server.register_event_handler( "server:klippy_ready", self._handle_klippy_ready) + self.server.register_event_handler( + "server:klippy_shutdown", self._handle_klippy_shutdown) self.server.register_event_handler( "server:klippy_disconnect", self._handle_klippy_disconnect) self.server.register_event_handler( @@ -188,6 +190,9 @@ class WebsocketManager: async def _handle_klippy_ready(self): await self.notify_websockets("klippy_ready") + async def _handle_klippy_shutdown(self): + await self.notify_websockets("klippy_shutdown") + async def _handle_klippy_disconnect(self): await self.notify_websockets("klippy_disconnected")