Browse Source

Add typing

CURA-6990
c.lamboo 2 years ago
parent
commit
eccef62dc7

+ 4 - 4
plugins/3MFReader/WorkspaceDialog.py

@@ -27,7 +27,7 @@ i18n_catalog = i18nCatalog("cura")
 class WorkspaceDialog(QObject):
     showDialogSignal = pyqtSignal()
 
-    def __init__(self, parent = None):
+    def __init__(self, parent = None) -> None:
         super().__init__(parent)
         self._component = None
         self._context = None
@@ -287,11 +287,11 @@ class WorkspaceDialog(QObject):
         self.missingPackagesChanged.emit()
 
     @pyqtProperty("QVariantList", notify=missingPackagesChanged)
-    def missingPackages(self):
+    def missingPackages(self) -> List[Dict[str, str]]:
         return self._missing_package_metadata
 
     @pyqtSlot()
-    def installMissingPackages(self):
+    def installMissingPackages(self) -> None:
         self._install_missing_package_dialog = InstallMissingPackageDialog(self._missing_package_metadata)
         self._install_missing_package_dialog.show()
 
@@ -408,7 +408,7 @@ class WorkspaceDialog(QObject):
         result_message.actionTriggered.connect(self._onMessageActionTriggered)
         result_message.show()
 
-    def _onMessageActionTriggered(self, message: Message, sync_message_action: str):
+    def _onMessageActionTriggered(self, message: Message, sync_message_action: str) -> None:
         if sync_message_action == "install_materials":
             self.installMissingPackages()
             message.hide()

+ 4 - 4
plugins/Marketplace/InstallMissingPackagesDialog.py

@@ -11,7 +11,7 @@ from plugins.Marketplace.MissingPackageList import MissingPackageList
 
 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]]):
+    def __init__(self, packages_metadata: List[Dict[str, str]]) -> None:
         """Initialize
 
         :param packages_metadata: List of dictionaries containing information about missing packages.
@@ -26,9 +26,9 @@ class InstallMissingPackageDialog(QObject):
         self._restart_needed = False
         self._package_metadata: List[Dict[str, str]] = packages_metadata
 
-        self._package_model = MissingPackageList(packages_metadata)
+        self._package_model: MissingPackageList = MissingPackageList(packages_metadata)
 
-    def show(self):
+    def show(self) -> None:
         plugin_path = self._plugin_registry.getPluginPath("Marketplace")
         if plugin_path is None:
             plugin_path = os.path.dirname(__file__)
@@ -52,5 +52,5 @@ class InstallMissingPackageDialog(QObject):
         return self._restart_needed
 
     @pyqtProperty(QObject)
-    def model(self):
+    def model(self) -> MissingPackageList:
         return self._package_model

+ 1 - 1
plugins/Marketplace/MissingPackageList.py

@@ -32,7 +32,7 @@ class MissingPackageList(RemotePackageList):
         if not self.hasMore:
             self._addPackagesMissingFromRequest()
 
-    def _addPackagesMissingFromRequest(self):
+    def _addPackagesMissingFromRequest(self) -> None:
         """Create cards for packages the user needs to install that could not be found"""
         returned_packages_ids = [item["package"].packageId for item in self._items]
 

+ 2 - 2
plugins/Marketplace/PackageModel.py

@@ -87,7 +87,7 @@ class PackageModel(QObject):
         self._is_missing_package_information = False
 
     @classmethod
-    def fromIncompletePackageInformation(cls, display_name: str, package_version: str, package_type: str):
+    def fromIncompletePackageInformation(cls, display_name: str, package_version: str, package_type: str) -> PackageModel:
         package_data = {
             "display_name": display_name,
             "package_version": package_version,
@@ -402,7 +402,7 @@ class PackageModel(QObject):
 
     isMissingPackageInformationChanged = pyqtSignal()
 
-    def setIsMissingPackageInformation(self, isMissingPackageInformation: bool):
+    def setIsMissingPackageInformation(self, isMissingPackageInformation: bool) -> None:
         self._is_missing_package_information = isMissingPackageInformation
         self.isMissingPackageInformationChanged.emit()