Added booster stuff view.. tho, it needs a close look since it yields some tracebacks
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user