mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-23 06:09:38 +08:00
120 lines
5.1 KiB
Python
120 lines
5.1 KiB
Python
# Copyright (c) 2020 Ultimaker B.V.
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
import configparser
|
|
from typing import Tuple, List
|
|
import io
|
|
import os # To get the path to check for hidden stacks to delete.
|
|
import re # To filter directories to search for hidden stacks to delete.
|
|
from UM.Resources import Resources # To get the path to check for hidden stacks to delete.
|
|
from UM.Version import Version # To sort folders by version number.
|
|
from UM.VersionUpgrade import VersionUpgrade
|
|
|
|
# Settings that were merged into one. Each one is a pair of settings. If both
|
|
# are overwritten, the key wins. If only the key or the value is overwritten,
|
|
# that value is used in the key.
|
|
_merged_settings = {
|
|
"machine_head_with_fans_polygon": "machine_head_polygon",
|
|
"support_wall_count": "support_tree_wall_count"
|
|
}
|
|
|
|
_removed_settings = {
|
|
"support_tree_wall_thickness"
|
|
}
|
|
|
|
class VersionUpgrade44to45(VersionUpgrade):
|
|
def __init__(self) -> None:
|
|
"""
|
|
Creates the version upgrade plug-in from 4.4 to 4.5.
|
|
|
|
In this case the plug-in will also check for stacks that need to be
|
|
deleted.
|
|
"""
|
|
data_storage_root = os.path.dirname(Resources.getDataStoragePath())
|
|
folders = os.listdir(data_storage_root) # All version folders.
|
|
folders = filter(lambda p: re.fullmatch(r"\d+\.\d+", p), folders) # Only folders with a correct version number as name.
|
|
latest_version = max(list(folders), key = Version) # Sort them by semantic version numbering.
|
|
if latest_version == "4.4":
|
|
self.removeHiddenStacks()
|
|
|
|
def removeHiddenStacks(self) -> None:
|
|
"""
|
|
If starting the upgrade from 4.4, this will remove any hidden printer
|
|
stacks from the configuration folder as well as all of the user profiles
|
|
and definition changes profiles.
|
|
|
|
This will ONLY run when upgrading from 4.4, not when e.g. upgrading from
|
|
4.3 to 4.6 (through 4.4). This is because it's to fix a bug
|
|
(https://github.com/Ultimaker/Cura/issues/6731) that occurred in 4.4
|
|
only, so only there will it have hidden stacks that need to be deleted.
|
|
If people upgrade from 4.3 they don't need to be deleted. If people
|
|
upgrade from 4.5 they have already been deleted previously or never got
|
|
the broken hidden stacks.
|
|
"""
|
|
pass
|
|
|
|
def getCfgVersion(self, serialised: str) -> int:
|
|
parser = configparser.ConfigParser(interpolation = None)
|
|
parser.read_string(serialised)
|
|
format_version = int(parser.get("general", "version")) # Explicitly give an exception when this fails. That means that the file format is not recognised.
|
|
setting_version = int(parser.get("metadata", "setting_version", fallback = "0"))
|
|
return format_version * 1000000 + setting_version
|
|
|
|
## Upgrades Preferences to have the new version number.
|
|
#
|
|
# This renames the renamed settings in the list of visible settings.
|
|
def upgradePreferences(self, serialized: str, filename: str) -> Tuple[List[str], List[str]]:
|
|
parser = configparser.ConfigParser(interpolation = None)
|
|
parser.read_string(serialized)
|
|
|
|
# Update version number.
|
|
parser["metadata"]["setting_version"] = "11"
|
|
|
|
result = io.StringIO()
|
|
parser.write(result)
|
|
return [filename], [result.getvalue()]
|
|
|
|
## Upgrades instance containers to have the new version
|
|
# number.
|
|
#
|
|
# This renames the renamed settings in the containers.
|
|
def upgradeInstanceContainer(self, serialized: str, filename: str) -> Tuple[List[str], List[str]]:
|
|
parser = configparser.ConfigParser(interpolation = None, comment_prefixes = ())
|
|
parser.read_string(serialized)
|
|
|
|
# Update version number.
|
|
parser["metadata"]["setting_version"] = "11"
|
|
|
|
if "values" in parser:
|
|
# Merged settings: When two settings are merged, one is preferred.
|
|
# If the preferred one is available, that value is taken regardless
|
|
# of the other one. If only the non-preferred one is available, that
|
|
# value is moved to the preferred setting value.
|
|
for preferred, removed in _merged_settings.items():
|
|
if removed in parser["values"]:
|
|
if preferred not in parser["values"]:
|
|
parser["values"][preferred] = parser["values"][removed]
|
|
del parser["values"][removed]
|
|
|
|
for removed in _removed_settings:
|
|
if removed in parser["values"]:
|
|
del parser["values"][removed]
|
|
|
|
result = io.StringIO()
|
|
parser.write(result)
|
|
return [filename], [result.getvalue()]
|
|
|
|
## Upgrades stacks to have the new version number.
|
|
def upgradeStack(self, serialized: str, filename: str) -> Tuple[List[str], List[str]]:
|
|
parser = configparser.ConfigParser(interpolation = None)
|
|
parser.read_string(serialized)
|
|
|
|
# Update version number.
|
|
if "metadata" not in parser:
|
|
parser["metadata"] = {}
|
|
parser["metadata"]["setting_version"] = "11"
|
|
|
|
result = io.StringIO()
|
|
parser.write(result)
|
|
return [filename], [result.getvalue()]
|