damagePattern.py

# =============================================================================
# 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 <http://www.gnu.org/licenses/>.
# =============================================================================


import copy

import eos.db
from eos.saveddata.damagePattern import DamagePattern as es_DamagePattern


class ImportError(Exception):
    pass


class DamagePattern:
    instance = None

    @classmethod
    def getInstance(cls):
        if cls.instance is None:
            cls.instance = DamagePattern()

        return cls.instance

    @staticmethod
    def getUserDamagePatternList():
        return eos.db.getDamagePatternList()

    @staticmethod
    def getBuiltinDamagePatternList():
        return es_DamagePattern.getBuiltinList()

    @staticmethod
    def getDamagePattern(name):
        return eos.db.getDamagePattern(name)

    @staticmethod
    def newPattern(name):
        p = es_DamagePattern(0, 0, 0, 0)
        p.rawName = name
        eos.db.save(p)
        return p

    @staticmethod
    def renamePattern(p, newName):
        p.rawName = 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.getUserDamagePatternList()
        for i in range(len(patterns) - 1, -1, -1):
            if patterns[i].name in ("Uniform", "Selected Ammo"):
                del patterns[i]

        patterns.sort(key=lambda p: p.fullName)
        return es_DamagePattern.exportPatterns(*patterns)