Add legacy 'Connect over Network' button back

plugins/UM3NetworkPrinting/src/UltimakerNetworkedPrinterAction.py

@@ -0,0 +1,91 @@
+# Copyright (c) 2019 Ultimaker B.V.
+# Cura is released under the terms of the LGPLv3 or higher.
+from typing import Optional, cast
+
+from PyQt5.QtCore import pyqtSlot, pyqtSignal, pyqtProperty, QObject
+
+from UM import i18nCatalog
+from cura.CuraApplication import CuraApplication
+from cura.MachineAction import MachineAction
+
+from .UM3OutputDevicePlugin import UM3OutputDevicePlugin
+from .Network.LocalClusterOutputDevice import LocalClusterOutputDevice
+
+
+I18N_CATALOG = i18nCatalog("cura")
+
+
+## Machine action that allows to connect the active machine to a networked devices.
+#  TODO: in the future this should be part of the new discovery workflow baked into Cura.
+class UltimakerNetworkedPrinterAction(MachineAction):
+    
+    # Signal emitted when discovered devices have changed.
+    discoveredDevicesChanged = pyqtSignal()
+
+    def __init__(self) -> None:
+        super().__init__("DiscoverUM3Action", I18N_CATALOG.i18nc("@action", "Connect via Network"))
+        self._qml_url = "resources/qml/DiscoverUM3Action.qml"
+        self._network_plugin = None  # type: Optional[UM3OutputDevicePlugin]
+
+    ## Override the default value.
+    def needsUserInteraction(self) -> bool:
+        return False
+
+    ## Start listening to network discovery events via the plugin.
+    @pyqtSlot(name = "startDiscovery")
+    def startDiscovery(self) -> None:
+        network_plugin = self._getNetworkPlugin()
+        network_plugin.discoveredDevicesChanged.connect(self._onDeviceDiscoveryChanged)
+        self.discoveredDevicesChanged.emit()  # trigger at least once to populate the list
+
+    ## Reset the discovered devices.
+    @pyqtSlot(name = "reset")
+    def reset(self) -> None:
+        self.restartDiscovery()
+
+    ## Reset the discovered devices.
+    @pyqtSlot(name = "restartDiscovery")
+    def restartDiscovery(self) -> None:
+        network_plugin = self._getNetworkPlugin()
+        network_plugin.startDiscovery()
+        self.discoveredDevicesChanged.emit()  # trigger to reset the list
+
+    ## Remove a manually added device.
+    @pyqtSlot(str, str, name = "removeManualDevice")
+    def removeManualDevice(self, key: str, address: str) -> None:
+        network_plugin = self._getNetworkPlugin()
+        network_plugin.removeManualDevice(key, address)
+
+    ## Add a new manual device. Can replace an existing one by key.
+    @pyqtSlot(str, str, name = "setManualDevice")
+    def setManualDevice(self, key: str, address: str) -> None:
+        network_plugin = self._getNetworkPlugin()
+        if key != "":
+            network_plugin.removeManualDevice(key)
+        if address != "":
+            network_plugin.addManualDevice(address)
+
+    ## Get the devices discovered in the local network sorted by name.
+    @pyqtProperty("QVariantList", notify = discoveredDevicesChanged)
+    def foundDevices(self):
+        network_plugin = self._getNetworkPlugin()
+        discovered_devices = list(network_plugin.getDiscoveredDevices().values())
+        discovered_devices.sort(key = lambda d: d.name)
+        return discovered_devices
+
+    ## Connect a device selected in the list with the active machine.
+    @pyqtSlot(QObject, name = "associateActiveMachineWithPrinterDevice")
+    def associateActiveMachineWithPrinterDevice(self, device: LocalClusterOutputDevice) -> None:
+        network_plugin = self._getNetworkPlugin()
+        network_plugin.associateActiveMachineWithPrinterDevice(device)
+
+    ## Callback for when the list of discovered devices in the plugin was changed.
+    def _onDeviceDiscoveryChanged(self) -> None:
+        self.discoveredDevicesChanged.emit()
+
+    ## Get the network manager from the plugin.
+    def _getNetworkPlugin(self) -> UM3OutputDevicePlugin:
+        if not self._network_plugin:
+            plugin = CuraApplication.getInstance().getOutputDeviceManager().getOutputDevicePlugin("UM3NetworkPrinting")
+            self._network_plugin = cast(UM3OutputDevicePlugin, plugin)
+        return self._network_plugin