mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-23 06:09:38 +08:00
45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
from cura.MachineAction import MachineAction
|
|
from PyQt5.QtCore import pyqtSlot, pyqtSignal, pyqtProperty
|
|
|
|
from UM.i18n import i18nCatalog
|
|
from UM.Application import Application
|
|
catalog = i18nCatalog("cura")
|
|
|
|
import UM.Settings.InstanceContainer
|
|
|
|
class UMOUpgradeSelection(MachineAction):
|
|
def __init__(self):
|
|
super().__init__("UMOUpgradeSelection", catalog.i18nc("@action", "Select upgrades"))
|
|
self._qml_url = "UMOUpgradeSelectionMachineAction.qml"
|
|
|
|
def _reset(self):
|
|
self.heatedBedChanged.emit()
|
|
|
|
heatedBedChanged = pyqtSignal()
|
|
|
|
@pyqtProperty(bool, notify = heatedBedChanged)
|
|
def hasHeatedBed(self):
|
|
global_container_stack = Application.getInstance().getGlobalContainerStack()
|
|
if global_container_stack:
|
|
return global_container_stack.getProperty("machine_heated_bed", "value")
|
|
|
|
@pyqtSlot(bool)
|
|
def setHeatedBed(self, heated_bed = True):
|
|
global_container_stack = Application.getInstance().getGlobalContainerStack()
|
|
if global_container_stack:
|
|
variant = global_container_stack.findContainer({"type": "variant"})
|
|
if variant:
|
|
if variant.getId() == "empty_variant":
|
|
variant_index = global_container_stack.getContainerIndex(variant)
|
|
variant = self._createVariant(global_container_stack, variant_index)
|
|
variant.setProperty("machine_heated_bed", "value", heated_bed)
|
|
self.heatedBedChanged.emit()
|
|
|
|
def _createVariant(self, global_container_stack, variant_index):
|
|
# Create and switch to a variant to store the settings in
|
|
new_variant = UM.Settings.InstanceContainer(global_container_stack.getName() + "_variant")
|
|
new_variant.addMetaDataEntry("type", "variant")
|
|
new_variant.setDefinition(global_container_stack.getBottom())
|
|
UM.Settings.ContainerRegistry.getInstance().addContainer(new_variant)
|
|
global_container_stack.replaceContainer(variant_index, new_variant)
|
|
return new_variant |