Recalc fit when changing implant source

2016-03-19 15:42:49 -04:00
parent dc8aff6b6c
commit 7c787cd13b
2 changed files with 11 additions and 7 deletions
--- a/service/fit.py
+++ b/service/fit.py
@@ -721,6 +721,14 @@ class Fit(object):
        self.recalc(fit)
        return True

+    def toggleImplantSource(self, fitID, source):
+        fit = eos.db.getFit(fitID)
+        fit.implantSource = source
+        
+        eos.db.commit()
+        self.recalc(fit)
+        return True
+
    def toggleBooster(self, fitID, i):
        fit = eos.db.getFit(fitID)
        booster = fit.boosters[i]