diff --git a/gui/patternEditor.py b/gui/patternEditor.py index 96bc8f97d..7d96b19be 100644 --- a/gui/patternEditor.py +++ b/gui/patternEditor.py @@ -67,7 +67,8 @@ class DmgPatternEntityEditor(EntityEditor): def getEntitiesFromContext(self): sDP = DamagePattern.getInstance() choices = sorted(sDP.getUserDamagePatternList(), key=lambda p: p.rawName) - return [c for c in choices if c.rawName != "Selected Ammo"] + choices = [c for c in choices if c.rawName != "Selected Ammo"] + return choices def DoNew(self, name): sDP = DamagePattern.getInstance() @@ -183,6 +184,10 @@ class DmgPatternEditor(AuxiliaryFrame): footerSizer.Add(btn, 0, wx.ALIGN_CENTER_HORIZONTAL | wx.ALIGN_RIGHT) btn.Bind(wx.EVT_BUTTON, getattr(self, "{}Patterns".format(name.lower()))) + if not self.entityEditor.checkEntitiesExist(): + self.Close() + return + self.Layout() bsize = self.GetBestSize() self.SetSize((-1, bsize.height)) @@ -232,6 +237,11 @@ class DmgPatternEditor(AuxiliaryFrame): self.entityEditor.btnDelete.Enable() def patternChanged(self, event=None): + + if not self.entityEditor.checkEntitiesExist(): + self.Close() + return + p = self.entityEditor.getActiveEntity() if p is None: diff --git a/gui/targetProfileEditor.py b/gui/targetProfileEditor.py index 94701ad1c..ce626019f 100644 --- a/gui/targetProfileEditor.py +++ b/gui/targetProfileEditor.py @@ -25,12 +25,12 @@ from collections import OrderedDict import wx from logbook import Logger -import gui.mainFrame import gui.globalEvents as GE +import gui.mainFrame from gui.auxFrame import AuxiliaryFrame from gui.bitmap_loader import BitmapLoader -from gui.builtinViews.entityEditor import EntityEditor, BaseValidator -from gui.utils.clipboard import toClipboard, fromClipboard +from gui.builtinViews.entityEditor import BaseValidator, EntityEditor +from gui.utils.clipboard import fromClipboard, toClipboard from gui.utils.inputs import FloatBox, InputValidator, strToFloat from service.fit import Fit from service.targetProfile import TargetProfile