Fix initial mode assignment

This commit is contained in:
blitzmann
2014-12-02 12:31:10 -05:00
parent b64238634a
commit 022a929399
2 changed files with 4 additions and 3 deletions

View File

@@ -100,10 +100,10 @@ class Fit(object):
self.extraAttributes = ModifiedAttributeDict(self)
self.extraAttributes.original = self.EXTRA_ATTRIBUTES
self.ship = Ship(db.getItem(self.shipID)) if self.shipID is not None else None
if self.ship is not None and self.modeID is not None:
self._mode = self.ship.checkModeItem(db.getItem(self.modeID))
if self.ship is not None:
self.mode = self.ship.checkModeItem(db.getItem(self.modeID) if self.modeID else None)
else:
self._mode = None
self.mode = None
@property
def targetResists(self):

View File

@@ -147,6 +147,7 @@ class Fit(object):
def newFit(self, shipID, name=None):
fit = eos.types.Fit()
fit.ship = eos.types.Ship(eos.db.getItem(shipID))
fit.mode = fit.ship.checkModeItem(None)
fit.name = name if name is not None else "New %s" % fit.ship.item.name
fit.damagePattern = self.pattern
fit.targetResists = self.targetResists