mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-13 11:58:02 +08:00
7ffa770fb4
Made typing more generic to work with an ContainerStack to compensate. Made AbstractMachine getMachines a classmethod so it can be called with ContainerStacks. CURA-9514
45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
from typing import List
|
|
|
|
from UM.Settings.ContainerStack import ContainerStack
|
|
from cura.PrinterOutput.PrinterOutputDevice import ConnectionType
|
|
from cura.Settings.GlobalStack import GlobalStack
|
|
from UM.MimeTypeDatabase import MimeType, MimeTypeDatabase
|
|
from UM.Settings.ContainerRegistry import ContainerRegistry
|
|
|
|
|
|
class AbstractMachine(GlobalStack):
|
|
""" Represents a group of machines of the same type. This allows the user to select settings before selecting a printer. """
|
|
|
|
def __init__(self, container_id: str) -> None:
|
|
super().__init__(container_id)
|
|
self.setMetaDataEntry("type", "abstract_machine")
|
|
|
|
@classmethod
|
|
def getMachines(cls, abstract_machine: ContainerStack) -> List[ContainerStack]:
|
|
""" Fetches containers for all machines that match definition with an abstract machine.
|
|
|
|
:param abstractMachine: The abstract machine stack.
|
|
:return: A list of Containers or an empty list if stack is not an "abstract_machine"
|
|
"""
|
|
if not abstract_machine.getMetaDataEntry("type") == "abstract_machine":
|
|
return []
|
|
|
|
from cura.CuraApplication import CuraApplication # In function to avoid circular import
|
|
application = CuraApplication.getInstance()
|
|
registry = application.getContainerRegistry()
|
|
|
|
printer_type = abstract_machine.definition.getId()
|
|
|
|
return [machine for machine in registry.findContainerStacks(type="machine") if machine.definition.id == printer_type and ConnectionType.CloudConnection in machine.configuredConnectionTypes]
|
|
|
|
|
|
## private:
|
|
_abstract_machine_mime = MimeType(
|
|
name = "application/x-cura-abstract-machine",
|
|
comment = "Cura Abstract Machine",
|
|
suffixes = ["global.cfg"]
|
|
)
|
|
|
|
MimeTypeDatabase.addMimeType(_abstract_machine_mime)
|
|
ContainerRegistry.addContainerTypeByName(AbstractMachine, "abstract_machine", _abstract_machine_mime.name)
|