Websocket: changes to include closing websocket

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: