Merge pull request #99 from blitzmann/issue-63

Fixes #63
This commit is contained in:
Anton Vorobyov
2014-06-03 00:34:58 +04:00
2 changed files with 26 additions and 16 deletions

View File

@@ -220,7 +220,11 @@ class Fit(object):
eos.db.commit()
self.recalc(fit, withBoosters=True)
def getFit(self, fitID):
def getFit(self, fitID, projected = False):
''' Gets fit from database, and populates fleet data.
Projected is a recursion flag that is set to reduce recursions into projected fits
'''
if fitID is None:
return None
fit = eos.db.getFit(fitID)
@@ -233,8 +237,13 @@ class Fit(object):
fit.fleet = None
else:
fit.fleet = f
self.recalc(fit, withBoosters=True)
fit.fill()
if not projected:
for fitP in fit.projectedFits:
self.getFit(fitP.ID, projected = True)
self.recalc(fit, withBoosters=True)
fit.fill()
eos.db.commit()
fit.inited = True
return fit