Added booster stuff view.. tho, it needs a close look since it yields some tracebacks

This commit is contained in:
HomeWorld
2010-09-19 20:02:21 +03:00
parent 87c6c42013
commit 5c609bdbc5
3 changed files with 70 additions and 5 deletions

View File

@@ -100,6 +100,35 @@ class Fit(object):
fit.calculateModifiedAttributes()
return True
def addBooster(self, fitID, itemID):
if fitID is None:
return False
fit = eos.db.getFit(fitID)
item = eos.db.getItem(itemID, eager=("attributes", "group.category"))
try:
booster = eos.types.Booster(item)
except ValueError:
return False
fit.boosters.freeSlot(booster)
fit.boosters.append(booster)
fit.clear()
fit.calculateModifiedAttributes()
return True
def removeBooster(self, fitID, position):
if fitID is None:
return False
fit = eos.db.getFit(fitID)
booster = fit.boosters[position]
fit.implants.remove(booster)
fit.clear()
fit.calculateModifiedAttributes()
return True
def appendModule(self, fitID, itemID):
fit = eos.db.getFit(fitID)
item = eos.db.getItem(itemID, eager=("attributes", "group.category"))
@@ -191,4 +220,4 @@ class Fit(object):
fit = eos.db.getFit(fitID)
fit.character = eos.db.getCharacter(charID)
fit.clear()
fit.calculateModifiedAttributes()
fit.calculateModifiedAttributes()