Cura/plugins/Marketplace/InstallMissingPackagesDialog.py

# Copyright (c) 2022 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.

import os

from PyQt6.QtCore import QObject, pyqtSignal, pyqtProperty
from typing import Optional, List, Dict, cast, Callable
from cura.CuraApplication import CuraApplication
from UM.PluginRegistry import PluginRegistry
from cura.CuraPackageManager import CuraPackageManager
from UM.i18n import i18nCatalog

from UM.FlameProfiler import pyqtSlot
from .MissingPackageList import MissingPackageList

i18n_catalog = i18nCatalog("cura")

class InstallMissingPackageDialog(QObject):
    """Dialog used to display packages that need to be installed to load 3mf file materials"""
    def __init__(self, packages_metadata: List[Dict[str, str]], show_missing_materials_warning: Callable[[], None]) -> None:
        """Initialize

        :param packages_metadata: List of dictionaries containing information about missing packages.
        """
        super().__init__()

        self._plugin_registry: PluginRegistry = CuraApplication.getInstance().getPluginRegistry()
        self._package_manager: CuraPackageManager = cast(CuraPackageManager, CuraApplication.getInstance().getPackageManager())
        self._package_manager.installedPackagesChanged.connect(self.checkIfRestartNeeded)

        self._dialog: Optional[QObject] = None
        self._restart_needed = False
        self._package_metadata: List[Dict[str, str]] = packages_metadata

        self._package_model: MissingPackageList = MissingPackageList(packages_metadata)
        self._show_missing_materials_warning = show_missing_materials_warning

    def show(self) -> None:
        plugin_path = self._plugin_registry.getPluginPath("Marketplace")
        if plugin_path is None:
            plugin_path = os.path.dirname(__file__)

        # create a QML component for the license dialog
        license_dialog_component_path = os.path.join(plugin_path, "resources", "qml", "InstallMissingPackagesDialog.qml")
        self._dialog = CuraApplication.getInstance().createQmlComponent(license_dialog_component_path, {"manager": self})
        self._dialog.show()

    def checkIfRestartNeeded(self) -> None:
        if self._dialog is None:
            return

        self._restart_needed = self._package_manager.hasPackagesToRemoveOrInstall
        self.showRestartChanged.emit()

    showRestartChanged = pyqtSignal()

    @pyqtProperty(bool, notify=showRestartChanged)
    def showRestartNotification(self) -> bool:
        return self._restart_needed

    @pyqtProperty(QObject)
    def model(self) -> MissingPackageList:
        return self._package_model

    @pyqtSlot()
    def showMissingMaterialsWarning(self) -> None:
        self._show_missing_materials_warning()