Add automatic correction of dependencies

[bertos.git] / wizard / BModulePage.py

diff --git a/wizard/BModulePage.py b/wizard/BModulePage.py
index 7fcfbcc13d4e954b344b780355f42fbdd6ca73d2..48ad0892db6ec34f19b21532fa764905de1a8033 100644 (file)
--- a/wizard/BModulePage.py
+++ b/wizard/BModulePage.py
@@ -152,46 +152,53 @@ class BModulePage(BWizardPage):
         modules[selectedModule]["enabled"] = True
         self._projectInfoStore("MODULES", modules)
         depends = self._projectInfoRetrieve("MODULES")[selectedModule]["depends"]
-        unsatisfied = []
-        for element in depends:
-            if not self._projectInfoRetrieve("MODULES")[element]["enabled"]:
-                unsatisfied.append(element)
+        unsatisfied = self.selectDependencyCheck(selectedModule)
         if len(unsatisfied) > 0:
-            self._selectionDependencyFail(selectedModule, unsatisfied)
-    
-    def _selectionDependencyFail(self, selectedModule, unsatisfiedModules):
-        messageString = "The module " + selectedModule + " need the following modules:\n" + \
-                        ", ".join(unsatisfiedModules) + ".\nDo you want to resolve autmatically the prolem?"
-        messageBox = QMessageBox()
-        messageBox.setIcon(QMessageBox.Warning)
-        messageBox.setText(self.tr("Dependency fail"))
-        messageBox.setInformativeText(self.tr(messageString))
-        resolveButton = QPushButton("Resolve")
-        nothingButton = QPushButton("Do nothing")
-        messageBox.addButton(resolveButton, QMessageBox.YesRole)
-        messageBox.addButton(nothingButton, QMessageBox.NoRole)
-        messageBox.exec_()
+            message = self.tr("The module %1 needs the following modules:\n%2.\n\nDo you want to resolve automatically the problem?")
+            message = message.arg(selectedModule).arg(", ".join(unsatisfied))
+            choice = QMessageBox.warning(self, self.tr("Dependency error"), message, QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
+            if choice == QMessageBox.Yes:
+                for module in unsatisfied:
+                    modules = self._projectInfoRetrieve("MODULES")
+                    modules[module]["enabled"] = True
+                for index in range(self.pageContent.moduleTable.rowCount()):
+                    if unicode(self.pageContent.moduleTable.item(index, 1).text()) in unsatisfied:
+                        self._buttonGroup.button(index).setChecked(True)
     
     def _moduleUnselected(self, unselectedModule):
         modules = self._projectInfoRetrieve("MODULES")
         modules[unselectedModule]["enabled"] = False
         self._projectInfoStore("MODULES", modules)
-        unsatisfied = []
-        for module, infos in self._projectInfoRetrieve("MODULES").items():
-            if unselectedModule in infos["depends"] and infos["enabled"]:
-                unsatisfied.append(module)
+        unsatisfied = self.unselectDependencyCheck(unselectedModule)
         if len(unsatisfied) > 0:
-            self._unselectionDependencyFail(unselectedModule, unsatisfied)
-    
-    def _unselectionDependencyFail(self, unselectedModule, unsatisfiedModules):
-        messageString = "The module " + unselectedModule + " is needed by the following modules:\n" + \
-                        ", ".join(unsatisfiedModules) + ".\nDo you want to resolve autmatically the prolem?"
-        messageBox = QMessageBox()
-        messageBox.setIcon(QMessageBox.Warning)
-        messageBox.setText(self.tr("Dependency fail"))
-        messageBox.setInformativeText(self.tr(messageString))
-        resolveButton = QPushButton("Resolve")
-        nothingButton = QPushButton("Do nothing")
-        messageBox.addButton(resolveButton, QMessageBox.YesRole)
-        messageBox.addButton(nothingButton, QMessageBox.NoRole)
-        messageBox.exec_()
\ No newline at end of file
+            message = self.tr("The module %1 is needed by the following modules:\n%2.\n\nDo you want to resolve automatically the problem?")
+            message = message.arg(unselectedModule).arg(", ".join(unsatisfied))
+            choice = QMessageBox.warning(self, self.tr("Dependency error"), message, QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
+            if choice == QMessageBox.Yes:
+                for module in unsatisfied:
+                    modules = self._projectInfoRetrieve("MODULES")
+                    modules[module]["enabled"] = False
+                for index in range(self.pageContent.moduleTable.rowCount()):
+                    if unicode(self.pageContent.moduleTable.item(index, 1).text()) in unsatisfied:
+                        self._buttonGroup.button(index).setChecked(False)
+    
+    
+    def selectDependencyCheck(self, module):
+        unsatisfied = set()
+        modules = self._projectInfoRetrieve("MODULES")
+        for dependency in modules[module]["depends"]:
+            if not modules[dependency]["enabled"]:
+                unsatisfied |= set([dependency])
+                if dependency not in unsatisfied:
+                    unsatisfied |= self.selectDependencyCheck(dependency)
+        return unsatisfied
+    
+    def unselectDependencyCheck(self, dependency):
+        unsatisfied = set()
+        modules = self._projectInfoRetrieve("MODULES")
+        for module, informations in modules.items():
+            if dependency in informations["depends"] and informations["enabled"]:
+                unsatisfied |= set([module])
+                if dependency not in unsatisfied:
+                    unsatisfied |= self.unselectDependencyCheck(module)
+        return unsatisfied
\ No newline at end of file