def updateObject(self, selectedObjt, dialogClassParameters): dialogClass = dialogClassParameters.dclass() dialog = dialogClass(self, dialogClassParameters) dialog.load(selectedObjt) if (dialog.ShowModal() == dialogClassParameters.createButtonId()): dialogOutParameters = dialog.parameters() updateFn = dialogClassParameters.setter() updateFn(dialogOutParameters) if (self.__class__.__name__ == 'VulnerabilitiesDialog' and self.__class__.__name__ == 'ThreatsDialog'): if (self.__class__.__name__ == 'VulnerabilitiesDialog'): dimName = 'vulnerability' else: dimName = 'threat' expCMs = self.dbProxy.exposedCountermeasures( dialogOutParameters) if (len(expCMs) > 0): ecDlg = ExposedCountermeasureDialog(self, expCMs) ecDlg.ShowModal() revisedExpCms = ecDlg.countermeasureEffectiveness() ecDlg.Destroy() self.dbProxy.updateCountermeasuresEffectiveness( dialogOutParameters.id(), dimName, revisedExpCms) self.updateDialogObject(dialogOutParameters.id(), self.listId, dialogOutParameters) dialog.Destroy()
def addObject(self,dialogClassParameters): dimName = 'dimension' try: dialogClass = dialogClassParameters.dclass() dialog = dialogClass(self,dialogClassParameters) if (dialog.ShowModal() == dialogClassParameters.createButtonId()): dialogOutParameters = dialog.parameters() addFn = dialogClassParameters.setter() objtId = addFn(dialogOutParameters) if (self.__class__.__name__ == 'VulnerabilitiesDialog' or self.__class__.__name__ == 'ThreatsDialog'): if (self.__class__.__name__ == 'VulnerabilitiesDialog'): dimName = 'vulnerability' else: dimName = 'threat' expCMs = self.dbProxy.exposedCountermeasures(dialogOutParameters) if (len(expCMs) > 0): ecDlg = ExposedCountermeasureDialog(self,expCMs) ecDlg.ShowModal() revisedExpCms = ecDlg.countermeasureEffectiveness() ecDlg.Destroy() self.dbProxy.updateCountermeasuresEffectiveness(objtId,dimName,revisedExpCms) self.addObjectToDialog(objtId,self.listId,dialogOutParameters) dialog.Destroy() except ARM.ARMException,errorText: dlg = wx.MessageDialog(self,str(errorText),'Add ' + dimName,wx.OK | wx.ICON_ERROR) dlg.ShowModal() dlg.Destroy() return
def addObject(self, dialogClassParameters): dimName = 'dimension' try: dialogClass = dialogClassParameters.dclass() dialog = dialogClass(self, dialogClassParameters) if (dialog.ShowModal() == dialogClassParameters.createButtonId()): dialogOutParameters = dialog.parameters() addFn = dialogClassParameters.setter() objtId = addFn(dialogOutParameters) if (self.__class__.__name__ == 'VulnerabilitiesDialog' or self.__class__.__name__ == 'ThreatsDialog'): if (self.__class__.__name__ == 'VulnerabilitiesDialog'): dimName = 'vulnerability' else: dimName = 'threat' expCMs = self.dbProxy.exposedCountermeasures( dialogOutParameters) if (len(expCMs) > 0): ecDlg = ExposedCountermeasureDialog(self, expCMs) ecDlg.ShowModal() revisedExpCms = ecDlg.countermeasureEffectiveness() ecDlg.Destroy() self.dbProxy.updateCountermeasuresEffectiveness( objtId, dimName, revisedExpCms) self.addObjectToDialog(objtId, self.listId, dialogOutParameters) dialog.Destroy() except ARMException, errorText: dlg = wx.MessageDialog(self, str(errorText), 'Add ' + dimName, wx.OK | wx.ICON_ERROR) dlg.ShowModal() dlg.Destroy() return
def updateObject(self,selectedObjt,dialogClassParameters): dialogClass = dialogClassParameters.dclass() dialog = dialogClass(self,dialogClassParameters) dialog.load(selectedObjt) if (dialog.ShowModal() == dialogClassParameters.createButtonId()): dialogOutParameters = dialog.parameters() updateFn = dialogClassParameters.setter() updateFn(dialogOutParameters) if (self.__class__.__name__ == 'VulnerabilitiesDialog' and self.__class__.__name__ == 'ThreatsDialog'): if (self.__class__.__name__ == 'VulnerabilitiesDialog'): dimName = 'vulnerability' else: dimName = 'threat' expCMs = self.dbProxy.exposedCountermeasures(dialogOutParameters) if (len(expCMs) > 0): ecDlg = ExposedCountermeasureDialog(self,expCMs) ecDlg.ShowModal() revisedExpCms = ecDlg.countermeasureEffectiveness() ecDlg.Destroy() self.dbProxy.updateCountermeasuresEffectiveness(dialogOutParameters.id(),dimName,revisedExpCms) self.updateDialogObject(dialogOutParameters.id(),self.listId,dialogOutParameters) dialog.Destroy()