Added import/export support

Export works swimmingly, but import is slow and I can't figure out why.
This commit is contained in:
Corollax
2010-11-20 06:48:17 -06:00
parent 0b191623ad
commit 24b7c50b0a
2 changed files with 19 additions and 16 deletions

View File

@@ -31,17 +31,15 @@ class DamagePattern():
return cls.instance
def __init__(self):
self.getDamagePatternList()
def getDamagePatternList(self):
patterns = eos.db.getDamagePatternList()
if len(patterns) == 0:
try:
uniform = eos.db.getDamagePattern("Uniform")
except:
uniform = eos.types.DamagePattern(25, 25, 25, 25)
uniform.name = "Uniform"
eos.db.save(uniform)
patterns.append(uniform)
return patterns
def getDamagePatternList(self):
return eos.db.getDamagePatternList()
def getDamagePattern(self, name):
return eos.db.getDamagePattern(name)
@@ -67,8 +65,17 @@ class DamagePattern():
eos.db.save(p)
def importPatterns(self, text):
eos.types.DamagePattern.importPatterns(text)
lookup = {}
current = self.getDamagePatternList()
for pattern in current:
lookup[pattern.name] = pattern
imports = eos.types.DamagePattern.importPatterns(text)
for pattern in imports:
if pattern.name in lookup:
self.deletePattern( lookup[pattern.name] )
eos.db.save(pattern)
def exportPatterns(self):
patterns = self.getDamagePatternList()
return eos.types.DamagePattern.exportPatterns(*patterns)