1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
# Copyright (c) 2022 Ultimaker B.V.
# Uranium is released under the terms of the LGPLv3 or higher.
from PyQt6.QtCore import QObject, QUrl, pyqtSignal, pyqtProperty, pyqtSlot # For typing.
from typing import Optional
from UM.Logger import Logger
from UM.FileHandler.FileHandler import FileHandler
from UM.FileHandler.ReadFileJob import ReadFileJob # For typing.
from typing import TYPE_CHECKING, cast
from UM.Workspace.WorkspaceReader import WorkspaceReader
if TYPE_CHECKING:
from UM.Qt.QtApplication import QtApplication
## Central class for reading and writing workspaces.
# This class is created by Application and handles reading and writing workspace files.
class WorkspaceFileHandler(FileHandler):
def __init__(self, application: "QtApplication", writer_type: str = "workspace_writer", reader_type: str = "workspace_reader", parent: QObject = None) -> None:
super().__init__(application, writer_type, reader_type, parent)
self.workspace_reader = None # type: Optional[WorkspaceReader]
self._enabled = True
self.enabledChanged.emit()
enabledChanged = pyqtSignal()
def setEnabled(self, enabled: bool) -> None:
if self._enabled != enabled:
self._enabled = enabled
self.enabledChanged.emit()
@pyqtProperty(bool, notify = enabledChanged)
def enabled(self):
return self._enabled
def readerRead(self, reader: WorkspaceReader, file_name: str, **kwargs):
self.workspace_reader = reader
results = None
try:
results = reader.read(file_name)
except Exception:
Logger.logException("w", "An exception occurred while loading workspace [%s]" % file_name)
return results
def _readLocalFile(self, file: QUrl, add_to_recent_files_hint: bool = True) -> None:
from UM.FileHandler.ReadFileJob import ReadFileJob
filename = file.toLocalFile()
job = ReadFileJob(filename, handler = self, add_to_recent_files = add_to_recent_files_hint)
job.finished.connect(self._readWorkspaceFinished)
job.start()
def _readWorkspaceFinished(self, job: ReadFileJob) -> None:
# Add the loaded nodes to the scene.
result = job.getResult()
if isinstance(result, tuple):
nodes, metadata = result
else:
nodes = result
metadata = {}
if nodes is not None: # Job was not a failure.
self._application.resetWorkspace()
for node in nodes:
# We need to prevent circular dependency, so do some just in time importing.
from UM.Operations.AddSceneNodeOperation import AddSceneNodeOperation
op = AddSceneNodeOperation(node, self._application.getController().getScene().getRoot())
op.push()
self._application.getWorkspaceMetadataStorage().setAllData(metadata)
self._application.workspaceLoaded.emit(cast(WorkspaceReader, self.workspace_reader).workspaceName())
|
