|
|
@@ -1,13 +1,15 @@
|
|
|
# Copyright (c) 2018 Ultimaker B.V.
|
|
|
# Cura is released under the terms of the LGPLv3 or higher.
|
|
|
|
|
|
-from typing import List, Tuple, TYPE_CHECKING, Optional
|
|
|
+from typing import Any, cast, Dict, List, Set, Tuple, TYPE_CHECKING, Optional
|
|
|
|
|
|
-from cura.CuraApplication import CuraApplication #To find some resource types.
|
|
|
+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.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
|
|
|
@@ -17,6 +19,31 @@ if TYPE_CHECKING:
|
|
|
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)
|
|
|
@@ -47,3 +74,12 @@ class CuraPackageManager(PackageManager):
|
|
|
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
|
casper
