Fixed bug where the bandwidth per drone was always taken from the fir…

…st drone stack
pyfa-org · Jul 22, 2020 · 0750124 · 0750124
1 parent 0fc260e
commit 0750124
Showing 1 changed file with 16 additions and 15 deletions.
diff --git a/gui/builtinContextMenus/droneSplitStack.py b/gui/builtinContextMenus/droneSplitStack.py
@@ -54,22 +54,23 @@ def getText(self, callingWindow, itmContext, mainItem):
     def activate(self, callingWindow, fullContext, mainItem, i):
         fitID = self.mainFrame.getActiveFit()
         fit = Fit.getInstance().getFit(fitID)
-        bandwidth_per_drone = fit.drones[0].item. \
-            attributes['droneBandwidthUsed'].value
-        ship_bandwidth = fit.ship.item.attributes['droneBandwidth'].value
-        max_active_drones = int(ship_bandwidth/bandwidth_per_drone)
-        if max_active_drones == 0:
-            wx.MessageDialog(
-                None, "Cannot split drone stack to fit bandwidth. Each drone "
-                "uses {0} mbit/s and this ship only has {1} mbit/s."
-                .format(int(bandwidth_per_drone), int(ship_bandwidth)),
-                "Ship drone bandwidth exceeded", wx.OK | wx.ICON_ERROR
-            ).ShowModal()
-        else:
-            if max_active_drones > 5:
-                max_active_drones = 5
+        if mainItem in fit.drones:
+            bandwidth_per_drone = mainItem.item.\
+                                  attributes['droneBandwidthUsed'].value
+            ship_bandwidth = fit.ship.item.attributes['droneBandwidth'].value
+            max_active_drones = int(ship_bandwidth/bandwidth_per_drone)
+            if max_active_drones == 0:
+                wx.MessageDialog(
+                    None, "Cannot split drone stack to fit bandwidth. This "
+                    "drone type uses {0} mbit/s and this ship only has {1} "
+                    "mbit/s.".format(int(bandwidth_per_drone),
+                                     int(ship_bandwidth)),
+                    "Ship drone bandwidth exceeded", wx.OK | wx.ICON_ERROR
+                ).ShowModal()
+            else:
+                if max_active_drones > 5:
+                    max_active_drones = 5
 
-            if mainItem in fit.drones:
                 position = fit.drones.index(mainItem)
                 self.mainFrame.command.Submit(
                     cmd.GuiSplitLocalDroneStackCommand(fitID=fitID,