Fixes

plugins/UM3NetworkPrinting/src/ExportFileJob.py

@@ -0,0 +1,39 @@
+from typing import List
+
+from UM.FileHandler.FileHandler import FileHandler
+from UM.FileHandler.WriteFileJob import WriteFileJob
+from UM.Logger import Logger
+from UM.Scene.SceneNode import SceneNode
+from cura.CuraApplication import CuraApplication
+
+from .MeshFormatHandler import MeshFormatHandler
+
+
+## Job that exports the build plate to the correct file format for the target cluster.
+class ExportFileJob(WriteFileJob):
+
+    def __init__(self, file_handler: FileHandler, nodes: List[SceneNode], firmware_version: str) -> None:
+
+        self._mesh_format_handler = MeshFormatHandler(file_handler, firmware_version)
+        if not self._mesh_format_handler.is_valid:
+            Logger.log("e", "Missing file or mesh writer!")
+            return
+
+        # Determine the filename.
+        job_name = CuraApplication.getInstance().getPrintInformation().jobName
+        extension = self._mesh_format_handler.preferred_format.get("extension", "")
+        self.setFileName(f"{job_name}.{extension}")
+
+        super().__init__(self._mesh_format_handler.writer, self._mesh_format_handler.createStream(), nodes,
+                         self._mesh_format_handler.file_mode)
+
+    ## Get the mime type of the selected export file type.
+    def getMimeType(self) -> str:
+        return self._mesh_format_handler.mime_type
+
+    ## Get the job result as bytes as that is what we need to upload to the cluster.
+    def getOutput(self) -> bytes:
+        output = self.getStream().getvalue()
+        if isinstance(output, str):
+            output = output.encode("utf-8")
+        return output