| 8 | |
| 9 | |
| 10 | class CalcChangeFitSystemSecurityCommand(wx.Command): |
| 11 | |
| 12 | def __init__(self, fitID, secStatus): |
| 13 | wx.Command.__init__(self, True, 'Change Fit System Security') |
| 14 | self.fitID = fitID |
| 15 | self.secStatus = secStatus |
| 16 | self.savedSecStatus = None |
| 17 | |
| 18 | def Do(self): |
| 19 | pyfalog.debug('Doing changing system security status of fit {} to {}'.format(self.fitID, self.secStatus)) |
| 20 | fit = Fit.getInstance().getFit(self.fitID, basic=True) |
| 21 | # Fetching status via getter and then saving 'raw' security status |
| 22 | # is intentional, to restore pre-change state properly |
| 23 | if fit.getSystemSecurity() == self.secStatus: |
| 24 | return False |
| 25 | self.savedSecStatus = fit.systemSecurity |
| 26 | fit.systemSecurity = self.secStatus |
| 27 | return True |
| 28 | |
| 29 | def Undo(self): |
| 30 | pyfalog.debug('Undoing changing system security status of fit {} to {}'.format(self.fitID, self.secStatus)) |
| 31 | cmd = CalcChangeFitSystemSecurityCommand(fitID=self.fitID, secStatus=self.savedSecStatus) |
| 32 | return cmd.Do() |