From e9305a20a60c8506ee74da37a1e80764843f5fe4 Mon Sep 17 00:00:00 2001
From: ChrisTerBeke <c.terbeke@ultimaker.com>
Date: Fri, 9 Aug 2019 12:45:38 +0200
Subject: [PATCH] cleanup not host message

---
 .../src/UltimakerNetworkedPrinterOutputDevice.py  | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/plugins/UM3NetworkPrinting/src/UltimakerNetworkedPrinterOutputDevice.py b/plugins/UM3NetworkPrinting/src/UltimakerNetworkedPrinterOutputDevice.py
index b310fa90f5..4752c88f49 100644
--- a/plugins/UM3NetworkPrinting/src/UltimakerNetworkedPrinterOutputDevice.py
+++ b/plugins/UM3NetworkPrinting/src/UltimakerNetworkedPrinterOutputDevice.py
@@ -216,18 +216,19 @@ class UltimakerNetworkedPrinterOutputDevice(NetworkedPrinterOutputDevice):
             if self._active_printer and self._active_printer.key == removed_printer.key:
                 self.setActivePrinter(None)
 
-        # Check if this is actually a group host.
-        if len(new_printers) < 1 and self.isConnected():
-            NotClusterHostMessage(self).show()
-            self.close()
-            CuraApplication.getInstance().getOutputDeviceManager().removeOutputDevice(self.key)
-            return
-
         self._printers = new_printers
         if self._printers and not self.activePrinter:
             self.setActivePrinter(self._printers[0])
 
         self.printersChanged.emit()
+        self._checkIfClusterHost()
+
+    ## Check is this device is a cluster host and takes the needed actions when it is not.
+    def _checkIfClusterHost(self):
+        if len(self._printers) < 1 and self.isConnected():
+            NotClusterHostMessage(self).show()
+            self.close()
+            CuraApplication.getInstance().getOutputDeviceManager().removeOutputDevice(self.key)
 
     ## Updates the local list of print jobs with the list received from the cluster.
     #  \param remote_jobs: The print jobs received from the cluster.