Merge pull request #541 from aacn500/missing-charge-skills

Fix "Change charge skills" showing module skills #535
2016-03-20 19:55:16 -04:00
parent 397040549f f8d7b68289
commit dd15d52130
1 changed files with 4 additions and 2 deletions
--- a/gui/builtinContextMenus/changeAffectingSkills.py
+++ b/gui/builtinContextMenus/changeAffectingSkills.py
@@ -28,10 +28,12 @@ class ChangeAffectingSkills(ContextMenu):
            fitID = self.mainFrame.getActiveFit()
            sFit = service.Fit.getInstance()
            self.stuff = sFit.getFit(fitID).ship
+            cont = sFit.getFit(fitID).ship.itemModifiedAttributes
+        elif srcContext == "fittingCharge":
+            cont = selection[0].chargeModifiedAttributes
        else:
-            self.stuff = selection[0]
+            cont = selection[0].itemModifiedAttributes

-        cont = self.stuff.itemModifiedAttributes
        skills = set()

        for attrName in cont.iterAfflictions():