Rework conditions in some context menus to check mainItem when it's needed by the menu

This commit is contained in:
DarkPhoenix
2019-04-23 10:14:59 +03:00
parent 87ba6a9af0
commit ad0c7a7a9d
21 changed files with 128 additions and 47 deletions

View File

@@ -16,10 +16,16 @@ class DroneSplitStack(ContextMenu):
self.settings = ContextMenuSettings.getInstance()
def display(self, srcContext, mainItem, selection):
return srcContext == "droneItem" and mainItem.amount > 1
if srcContext != "droneItem":
return False
if mainItem is None:
return False
return mainItem.amount > 1
def getText(self, itmContext, mainItem, selection):
return "Split {0} Stack".format(itmContext)
return "Split {} Stack".format(itmContext)
def activate(self, fullContext, mainItem, selection, i):
dlg = DroneStackSplit(self.mainFrame, mainItem.amount)