mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-09 08:59:02 +08:00
100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
# 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")
|
|
|
|
|
|
class UltimakerNetworkedPrinterAction(MachineAction):
|
|
"""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.
|
|
"""
|
|
|
|
# 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]
|
|
|
|
def needsUserInteraction(self) -> bool:
|
|
"""Override the default value."""
|
|
|
|
return False
|
|
|
|
@pyqtSlot(name = "startDiscovery")
|
|
def startDiscovery(self) -> None:
|
|
"""Start listening to network discovery events via the plugin."""
|
|
|
|
self._networkPlugin.discoveredDevicesChanged.connect(self._onDeviceDiscoveryChanged)
|
|
self.discoveredDevicesChanged.emit() # trigger at least once to populate the list
|
|
|
|
@pyqtSlot(name = "reset")
|
|
def reset(self) -> None:
|
|
"""Reset the discovered devices."""
|
|
|
|
self.discoveredDevicesChanged.emit() # trigger to reset the list
|
|
|
|
@pyqtSlot(name = "restartDiscovery")
|
|
def restartDiscovery(self) -> None:
|
|
"""Reset the discovered devices."""
|
|
|
|
self._networkPlugin.startDiscovery()
|
|
self.discoveredDevicesChanged.emit() # trigger to reset the list
|
|
|
|
@pyqtSlot(str, str, name = "removeManualDevice")
|
|
def removeManualDevice(self, key: str, address: str) -> None:
|
|
"""Remove a manually added device."""
|
|
|
|
self._networkPlugin.removeManualDevice(key, address)
|
|
|
|
@pyqtSlot(str, str, name = "setManualDevice")
|
|
def setManualDevice(self, key: str, address: str) -> None:
|
|
"""Add a new manual device. Can replace an existing one by key."""
|
|
|
|
if key != "":
|
|
self._networkPlugin.removeManualDevice(key)
|
|
if address != "":
|
|
self._networkPlugin.addManualDevice(address)
|
|
|
|
@pyqtProperty("QVariantList", notify = discoveredDevicesChanged)
|
|
def foundDevices(self):
|
|
"""Get the devices discovered in the local network sorted by name."""
|
|
|
|
discovered_devices = list(self._networkPlugin.getDiscoveredDevices().values())
|
|
discovered_devices.sort(key = lambda d: d.name)
|
|
return discovered_devices
|
|
|
|
@pyqtSlot(QObject, name = "associateActiveMachineWithPrinterDevice")
|
|
def associateActiveMachineWithPrinterDevice(self, device: LocalClusterOutputDevice) -> None:
|
|
"""Connect a device selected in the list with the active machine."""
|
|
|
|
self._networkPlugin.associateActiveMachineWithPrinterDevice(device)
|
|
|
|
def _onDeviceDiscoveryChanged(self) -> None:
|
|
"""Callback for when the list of discovered devices in the plugin was changed."""
|
|
|
|
self.discoveredDevicesChanged.emit()
|
|
|
|
@property
|
|
def _networkPlugin(self) -> UM3OutputDevicePlugin:
|
|
"""Get the network manager from the plugin."""
|
|
|
|
if not self._network_plugin:
|
|
output_device_manager = CuraApplication.getInstance().getOutputDeviceManager()
|
|
network_plugin = output_device_manager.getOutputDevicePlugin("UM3NetworkPrinting")
|
|
self._network_plugin = cast(UM3OutputDevicePlugin, network_plugin)
|
|
return self._network_plugin
|