123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134 |
- # Copyright (c) 2018 Ultimaker B.V.
- # Cura is released under the terms of the LGPLv3 or higher.
- import glob
- import os
- from pathlib import Path
- from typing import Any, cast, Dict, List, Set, Tuple, TYPE_CHECKING, Optional
- from UM.Logger import Logger
- from UM.PluginRegistry import PluginRegistry
- from cura.CuraApplication import CuraApplication # To find some resource types.
- from cura.Settings.GlobalStack import GlobalStack
- from UM.PackageManager import PackageManager # The class we're extending.
- from UM.Resources import Resources # To find storage paths for some resource types.
- from UM.i18n import i18nCatalog
- catalog = i18nCatalog("cura")
- if TYPE_CHECKING:
- from UM.Qt.QtApplication import QtApplication
- from PyQt6.QtCore import QObject
- class CuraPackageManager(PackageManager):
- def __init__(self, application: "QtApplication", parent: Optional["QObject"] = None) -> None:
- super().__init__(application, parent)
- self._local_packages: Optional[List[Dict[str, Any]]] = None
- self._local_packages_ids: Optional[Set[str]] = None
- self.installedPackagesChanged.connect(self._updateLocalPackages)
- def _updateLocalPackages(self) -> None:
- self._local_packages = self.getAllLocalPackages()
- self._local_packages_ids = set(pkg["package_id"] for pkg in self._local_packages)
- @property
- def local_packages(self) -> List[Dict[str, Any]]:
- """locally installed packages, lazy execution"""
- if self._local_packages is None:
- self._updateLocalPackages()
- # _updateLocalPackages always results in a list of packages, not None.
- # It's guaranteed to be a list now.
- return cast(List[Dict[str, Any]], self._local_packages)
- @property
- def local_packages_ids(self) -> Set[str]:
- """locally installed packages, lazy execution"""
- if self._local_packages_ids is None:
- self._updateLocalPackages()
- # _updateLocalPackages always results in a list of packages, not None.
- # It's guaranteed to be a list now.
- return cast(Set[str], self._local_packages_ids)
- def initialize(self) -> None:
- self._installation_dirs_dict["materials"] = Resources.getStoragePath(CuraApplication.ResourceTypes.MaterialInstanceContainer)
- self._installation_dirs_dict["qualities"] = Resources.getStoragePath(CuraApplication.ResourceTypes.QualityInstanceContainer)
- super().initialize()
- def isMaterialBundled(self, file_name: str, guid: str):
- """ Check if there is a bundled material name with file_name and guid """
- for path in Resources.getSecureSearchPaths():
- # Secure search paths are install directory paths, if a material is in here it must be bundled.
- paths = [Path(p) for p in glob.glob(path + '/**/*.xml.fdm_material', recursive=True)]
- for material in paths:
- if material.name == file_name:
- Logger.info(f"Found bundled material: {material.name}. Located in path: {str(material)}")
- with open(material, encoding="utf-8") as f:
- # Make sure the file we found has the same guid as our material
- # Parsing this xml would be better but the namespace is needed to search it.
- parsed_guid = PluginRegistry.getInstance().getPluginObject(
- "XmlMaterialProfile").getMetadataFromSerialized(
- f.read(), "GUID")
- if guid == parsed_guid:
- # The material we found matches both filename and GUID
- return True
- return False
- def getMaterialFilePackageId(self, file_name: str, guid: str) -> str:
- """Get the id of the installed material package that contains file_name"""
- for material_package in [f for f in os.scandir(self._installation_dirs_dict["materials"]) if f.is_dir()]:
- package_id = material_package.name
- for root, _, file_names in os.walk(material_package.path):
- if file_name not in file_names:
- # File with the name we are looking for is not in this directory
- continue
- with open(os.path.join(root, file_name), encoding="utf-8") as f:
- # Make sure the file we found has the same guid as our material
- # Parsing this xml would be better but the namespace is needed to search it.
- parsed_guid = PluginRegistry.getInstance().getPluginObject("XmlMaterialProfile").getMetadataFromSerialized(
- f.read(), "GUID")
- if guid == parsed_guid:
- return package_id
- Logger.error("Could not find package_id for file: {} with GUID: {} ".format(file_name, guid))
- return ""
- def getMachinesUsingPackage(self, package_id: str) -> Tuple[List[Tuple[GlobalStack, str, str]], List[Tuple[GlobalStack, str, str]]]:
- """Returns a list of where the package is used
- It loops through all the package contents and see if some of the ids are used.
- :param package_id: package id to search for
- :return: empty if it is never used, otherwise a list consisting of 3-tuples
- """
- ids = self.getPackageContainerIds(package_id)
- container_stacks = self._application.getContainerRegistry().findContainerStacks()
- global_stacks = [container_stack for container_stack in container_stacks if isinstance(container_stack, GlobalStack)]
- machine_with_materials = []
- machine_with_qualities = []
- for container_id in ids:
- for global_stack in global_stacks:
- for extruder_nr, extruder_stack in enumerate(global_stack.extruderList):
- if container_id in (extruder_stack.material.getId(), extruder_stack.material.getMetaData().get("base_file")):
- machine_with_materials.append((global_stack, str(extruder_nr), container_id))
- if container_id == extruder_stack.quality.getId():
- machine_with_qualities.append((global_stack, str(extruder_nr), container_id))
- return machine_with_materials, machine_with_qualities
- def getAllLocalPackages(self) -> List[Dict[str, Any]]:
- """ Returns an unordered list of all the package_info of installed, to be installed, or bundled packages"""
- packages: List[Dict[str, Any]] = []
- for packages_to_add in self.getAllInstalledPackagesInfo().values():
- packages.extend(packages_to_add)
- return packages
|