# ============================================================================= # Copyright (C) 2010 Diego Duclos # # 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 copy import eos.db from eos.saveddata.damagePattern import DamagePattern as es_DamagePattern class ImportError(Exception): pass class DamagePattern(object): instance = None @classmethod def getInstance(cls): if cls.instance is None: cls.instance = DamagePattern() return cls.instance @staticmethod def getDamagePatternList(): return eos.db.getDamagePatternList() @staticmethod def getDamagePattern(name): return eos.db.getDamagePattern(name) @staticmethod def newPattern(name): p = es_DamagePattern(0, 0, 0, 0) p.name = name eos.db.save(p) return p @staticmethod def renamePattern(p, newName): p.name = newName eos.db.save(p) @staticmethod def deletePattern(p): eos.db.remove(p) @staticmethod def copyPattern(p): newP = copy.deepcopy(p) eos.db.save(newP) return newP @staticmethod def saveChanges(p): eos.db.save(p) def importPatterns(self, text): imports, num = es_DamagePattern.importPatterns(text) 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.getDamagePatternList() for i in xrange(len(patterns) - 1, -1, -1): if patterns[i].name in ("Uniform", "Selected Ammo"): del patterns[i] patterns.sort(key=lambda p: p.name) return es_DamagePattern.exportPatterns(*patterns)