PrintSettingConverter.py

# LGPLv3

import re
import os.path

from typing import cast

from PyQt6.QtCore import QDateTime
from PyQt6.QtCore import QObject

from UM.i18n import i18nCatalog
from UM.Extension import Extension
from UM.Application import Application
from UM.Qt.Duration import DurationFormat
from UM.PluginRegistry import PluginRegistry
from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator
from UM.Scene.SceneNode import SceneNode
from UM.Version import Version

from cura.CuraApplication import CuraApplication
from cura.Settings.ExtruderManager import ExtruderManager
from cura.UI.ObjectsModel import ObjectsModel

from PyQt6.QtWidgets import QFileDialog, QMessageBox

    # Retrieve print setting values from Cura/Uranium
def getPrintSettings(filename_format):
    application = cast(CuraApplication, Application.getInstance())
    print_information = application.getPrintInformation()
    machine_manager = application.getMachineManager()
    global_stack = machine_manager.activeMachine
    print_settings = dict()
    multi_extruder_settings = dict()

    objectCount = application.getPreferences().getValue("gcode_filename_format_plus/object_count")

    try:
        first_extruder_stack = ExtruderManager.getInstance().getActiveExtruderStacks()[0]
        active_extruder_stacks = ExtruderManager.getInstance().getActiveExtruderStacks()
        base_name = print_information.baseName
        abbr_machine = print_information._abbr_machine
        printer_name = global_stack.getName()
        profile_name = machine_manager.activeQualityOrQualityChangesName
        print_time = print_information.currentPrintTime.getDisplayString(DurationFormat.Format.ISO8601)
        print_time_days = print_information.currentPrintTime.days
        print_time_hours = print_information.currentPrintTime.hours
        print_time_hours_all = print_time_days * 24 + print_time_hours
        print_time_minutes = print_information.currentPrintTime.minutes
        print_time_seconds = print_information.currentPrintTime.seconds
        material_weight = print_information.materialWeights
        material_length = print_information.materialLengths
        material_cost = print_information.materialCosts
        object_count = objectCount
        cura_version = Version(Application.getInstance().getVersion())

        nodes_to_arrange = []
        for node in DepthFirstIterator(application.getController().getScene().getRoot()):
            if not isinstance(node, SceneNode):
                continue

            if not node.getMeshData() and not node.callDecoration("isGroup"):
                continue  # Node that doesn't have a mesh and is not a group.

            parent_node = node.getParent()
            if parent_node and parent_node.callDecoration("isGroup"):
                continue  # Grouped nodes don't need resetting as their parent (the group) is reset)

            if not node.callDecoration("isSliceable") and not node.callDecoration("isGroup"):
                continue  # i.e. node with layer data

            bounding_box = node.getBoundingBox()
            # Skip nodes that are too big
            if bounding_box is None or bounding_box.width < application._volume.getBoundingBox().width or bounding_box.depth < application._volume.getBoundingBox().depth:
                nodes_to_arrange.append(node)

        scale_vector = nodes_to_arrange[0].getScale()
        scale_x = round(scale_vector.x * 100, 2)
        scale_y = round(scale_vector.z * 100, 2)
        scale_z = round(scale_vector.y * 100, 2)

    except (IndexError, AttributeError):
        return None

    time = QDateTime.currentDateTime().toString("HH-mm")
    year =  QDateTime.currentDateTime().toString("yyyy")
    month = QDateTime.currentDateTime().toString("MM")
    day = QDateTime.currentDateTime().toString("dd")
    hour = QDateTime.currentDateTime().toString("HH")
    minute = QDateTime.currentDateTime().toString("mm")
    second = QDateTime.currentDateTime().toString("ss")
    date = year + '-' + month + '-' + day
    datetime = date + 'T' + hour + minute + second

    tokens = re.split(r'\W+', filename_format)      # TODO: split on brackets only

    for t in tokens:
        stack1 = first_extruder_stack.material.getMetaData().get(t, "")
        stack2 = global_stack.userChanges.getProperty(t, "value")
        stack3 = first_extruder_stack.getProperty(t, "value")

        if stack1 is not None and stack1 != "":
            if type(stack1) is float:
                print_settings[t] = round(stack1, 2)
            else:
                print_settings[t] = stack1
        elif stack2 is not None and stack2 != "":
            if type(stack2) is float:
                print_settings[t] = round(stack2, 2)
            else:
                print_settings[t] = stack2
        elif stack3 is not None and stack3 != "":
            if type(stack3) is float:
                print_settings[t] = round(stack3, 2)
            else:
                print_settings[t] = stack3
        else:
            print_settings[t] = None

        for a in active_extruder_stacks:
            extruder_position = a.position

            try:
                stack1 = a.material.getMetaData().get(t[:-1], "")
                stack2 = a.getProperty(t[:-1], "value")

                if stack1 is not None and stack1 != "" and stack1 != 0 and extruder_position + 1 == int(t[-1]):
                    if type(stack1) is float:
                        multi_extruder_settings[t] = round(stack1, 2)
                    else:
                        multi_extruder_settings[t] = stack1
                    multi_extruder_settings[t] = round(stack1, 2)
                elif stack2 is not None and stack2 != "" and stack2 != 0 and extruder_position + 1 == int(t[-1]):
                    if type(stack2) is float:
                        multi_extruder_settings[t] = round(stack2, 2)
                    else:
                        multi_extruder_settings[t] = stack2
                #else:
                #    #print_settings[t] = None
                #    multi_extruder_settings[t] = None
            except TypeError:
                pass
            except ValueError:
                pass

    print_settings["base_name"] = base_name
    print_settings["job_name"] = base_name
    print_settings["abbr_machine"] = abbr_machine
    print_settings["printer_name"] = printer_name
    print_settings["profile_name"] = profile_name
    print_settings["print_time"] = print_time
    print_settings["print_time_days"] = print_time_days
    print_settings["print_time_hours"] = print_time_hours
    print_settings["print_time_hours_all"] = print_time_hours_all
    print_settings["print_time_minutes"] = print_time_minutes
    print_settings["print_time_seconds"] = print_time_seconds
    print_settings["date"] = date
    print_settings["time"] = time
    print_settings["datetime"] = datetime
    print_settings["year"] = year
    print_settings["month"] = month
    print_settings["day"] = day
    print_settings["hour"] = hour
    print_settings["minute"] = minute
    print_settings["second"] = second
    print_settings["object_count"] = object_count
    print_settings["cura_version"] = cura_version
    print_settings["scale_x"] = f'{scale_x:g}'
    print_settings["scale_y"] = f'{scale_y:g}'
    print_settings["scale_z"] = f'{scale_z:g}'
    print_settings["scale"] = print_settings["scale_x"]

    try:
        if len(material_weight) > 1:
            print_settings["material_weight1"] = round(float(material_weight[0]))
            print_settings["material_weight2"] = round(float(material_weight[1]))
            print_settings["material_weight"] = round(float(material_weight[0] + material_weight[1]))
        else:
            print_settings["material_weight"] = round(float(material_weight[0]))
    except IndexError:
        pass

    try:
        if len(material_length) > 1:
            print_settings["material_length1"] = round(float(material_length[0]), 2)
            print_settings["material_length2"] = round(float(material_length[1]), 2)
            print_settings["material_length"] = round(float(material_length[0] + material_length[1]), 2)
        else:
            print_settings["material_length"] = round(float(material_length[0]), 2)
    except IndexError:
        pass

    try:
        if len(material_cost) > 1:
            print_settings["material_cost1"] = round(float(material_cost[0]), 2)
            print_settings["material_cost2"] = round(float(material_cost[1]), 2)
            print_settings["material_cost"] = round(float(material_cost[0] + material_cost[1]), 2)
        else:
            print_settings["material_cost"] = round(float(material_cost[0]), 2)
    except IndexError:
        pass

    print_settings.update(multi_extruder_settings)

    return print_settings