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()
