mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 05:39:37 +08:00
73 lines
2.8 KiB
Python
73 lines
2.8 KiB
Python
# Copyright (c) 2023 UltiMaker
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import configparser
|
|
from typing import Tuple, List
|
|
import io
|
|
from UM.VersionUpgrade import VersionUpgrade
|
|
import re
|
|
|
|
|
|
|
|
class VersionUpgrade54to55(VersionUpgrade):
|
|
profile_regex = re.compile(
|
|
r"um\_(?P<machine>s(3|5|7))_(?P<core_type>aa|cc|bb)(?P<nozzle_size>0\.(6|4|8))_(?P<material>pla|petg|abs|tough_pla)_(?P<layer_height>0\.\d{1,2}mm)")
|
|
|
|
@staticmethod
|
|
def _isUpgradedUltimakerDefinitionId(definition_id: str) -> bool:
|
|
if definition_id.startswith("ultimaker_s5"):
|
|
return True
|
|
if definition_id.startswith("ultimaker_s3"):
|
|
return True
|
|
if definition_id.startswith("ultimaker_s7"):
|
|
return True
|
|
|
|
return False
|
|
|
|
@staticmethod
|
|
def _isBrandedMaterialID(material_id: str) -> bool:
|
|
return material_id.startswith("ultimaker_")
|
|
|
|
@staticmethod
|
|
def upgradeStack(serialized: str, filename: str) -> Tuple[List[str], List[str]]:
|
|
"""
|
|
Upgrades stacks to have the new version number.
|
|
|
|
:param serialized: The original contents of the stack.
|
|
:param filename: The original file name of the stack.
|
|
:return: A list of new file names, and a list of the new contents for
|
|
those files.
|
|
"""
|
|
parser = configparser.ConfigParser(interpolation = None)
|
|
parser.read_string(serialized)
|
|
|
|
# Update version number.
|
|
if "general" not in parser:
|
|
parser["general"] = {}
|
|
|
|
extruder_definition_id = parser["containers"]["7"]
|
|
if parser["metadata"]["type"] == "extruder_train" and VersionUpgrade54to55._isUpgradedUltimakerDefinitionId(extruder_definition_id):
|
|
# We only need to update certain Ultimaker extruder ID's
|
|
material_id = parser["containers"]["4"]
|
|
quality_id = parser["containers"]["3"]
|
|
intent_id = parser["containers"]["2"]
|
|
if VersionUpgrade54to55._isBrandedMaterialID(material_id):
|
|
# We have an Ultimaker branded material ID, so we should change the intent & quality!
|
|
|
|
quality_id = VersionUpgrade54to55.profile_regex.sub(
|
|
r"um_\g<machine>_\g<core_type>\g<nozzle_size>_um-\g<material>_\g<layer_height>", quality_id)
|
|
|
|
|
|
intent_id = VersionUpgrade54to55.profile_regex.sub(
|
|
r"um_\g<machine>_\g<core_type>\g<nozzle_size>_um-\g<material>_\g<layer_height>", intent_id)
|
|
|
|
parser["containers"]["3"] = quality_id
|
|
parser["containers"]["2"] = intent_id
|
|
|
|
# We're not changing any settings, but we are changing how certain stacks are handled.
|
|
parser["general"]["version"] = "6"
|
|
|
|
result = io.StringIO()
|
|
parser.write(result)
|
|
return [filename], [result.getvalue()]
|