#=============================================================================== # Copyright (C) 2014 Ryan Holmes # # This file is part of pyfa. # # pyfa is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # pyfa is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with pyfa. If not, see . #=============================================================================== import eos.db import eos.types import copy class ImportError(Exception): pass class TargetResists(): instance = None @classmethod def getInstance(cls): if cls.instance is None: cls.instance = TargetResists() return cls.instance def getTargetResistsList(self): return eos.db.getTargetResistsList() def getTargetResists(self, name): return eos.db.getTargetResists(name) def newPattern(self): p = eos.types.TargetResists(0.0, 0.0, 0.0, 0.0) p.name = "" return p def renamePattern(self, p, newName): p.name = newName eos.db.save(p) def deletePattern(self, p): eos.db.remove(p) def copyPattern(self, p): newP = copy.deepcopy(p) eos.db.save(newP) return newP def saveChanges(self, p): eos.db.save(p) def importPatterns(self, text): lookup = {} current = self.getTargetResistsList() for pattern in current: lookup[pattern.name] = pattern imports, num = eos.types.TargetResists.importPatterns(text) for pattern in imports: if pattern.name in lookup: match = lookup[pattern.name] match.__dict__.update(pattern.__dict__) else: eos.db.save(pattern) eos.db.commit() lenImports = len(imports) if lenImports == 0: raise ImportError("No patterns found for import") if lenImports != num: raise ImportError("%d patterns imported from clipboard; %d had errors"%(num, num-lenImports)) def exportPatterns(self): patterns = self.getTargetResistsList() patterns.sort(key=lambda p: p.name) return eos.types.TargetResists.exportPatterns(*patterns)