From b251a2ebbb16e7c848ab931710b0eb748918019c Mon Sep 17 00:00:00 2001 From: Jordan Ruthe Date: Mon, 15 Feb 2021 14:20:39 -0500 Subject: [PATCH] Websocket: changes to include closing websocket --- ks_includes/KlippyWebsocket.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/ks_includes/KlippyWebsocket.py b/ks_includes/KlippyWebsocket.py index 7acf6660..6b15d9fe 100644 --- a/ks_includes/KlippyWebsocket.py +++ b/ks_includes/KlippyWebsocket.py @@ -43,6 +43,7 @@ class KlippyWebsocket(threading.Thread): self._screen = screen self._callback = callback self.klippy = MoonrakerApi(self) + self.closing = False self._url = "%s:%s" % (host, port) @@ -73,6 +74,9 @@ class KlippyWebsocket(threading.Thread): except Exception: logger.debug("Error starting web socket") + def close(self): + self.closing = True + def is_connected(self): return self.connected @@ -133,6 +137,11 @@ class KlippyWebsocket(threading.Thread): logger.debug("Connection already closed") return + if self.closing == True: + logger.debug("Closing websocket") + self.ws.stop() + return + logger.info("Moonraker Websocket Closed") self.connected = False if self.timeout == None: