228 lines
7.6 KiB
Python
228 lines
7.6 KiB
Python
# =============================================================================
|
|
# Copyright (C) 2016 Ryan Holmes
|
|
#
|
|
# This file is part of pyfa.
|
|
#
|
|
# pyfa is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# pyfa is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with pyfa. If not, see <http://www.gnu.org/licenses/>.
|
|
# =============================================================================
|
|
|
|
# noinspection PyPackageRequirements
|
|
import wx
|
|
from logbook import Logger
|
|
|
|
from gui.auxFrame import AuxiliaryFrame
|
|
from gui.builtinViews.entityEditor import BaseValidator, EntityEditor
|
|
from gui.builtinViews.implantEditor import BaseImplantEditorView
|
|
from gui.utils.clipboard import fromClipboard, toClipboard
|
|
from service.implantSet import ImplantSets
|
|
|
|
|
|
pyfalog = Logger(__name__)
|
|
|
|
|
|
class ImplantTextValidor(BaseValidator):
|
|
def __init__(self):
|
|
BaseValidator.__init__(self)
|
|
|
|
def Clone(self):
|
|
return ImplantTextValidor()
|
|
|
|
def Validate(self, win):
|
|
entityEditor = win.parent
|
|
textCtrl = self.GetWindow()
|
|
text = textCtrl.GetValue().strip()
|
|
|
|
try:
|
|
if len(text) == 0:
|
|
raise ValueError("You must supply a name for the Implant Set!")
|
|
elif text in [x.name for x in entityEditor.choices]:
|
|
raise ValueError("Implant Set name already in use, please choose another.")
|
|
|
|
return True
|
|
except ValueError as e:
|
|
pyfalog.error(e)
|
|
wx.MessageBox("{}".format(e), "Error")
|
|
textCtrl.SetFocus()
|
|
return False
|
|
|
|
|
|
class ImplantSetEntityEditor(EntityEditor):
|
|
def __init__(self, parent):
|
|
EntityEditor.__init__(self, parent, "Implant Set")
|
|
self.SetEditorValidator(ImplantTextValidor)
|
|
|
|
def getEntitiesFromContext(self):
|
|
sIS = ImplantSets.getInstance()
|
|
return sorted(sIS.getImplantSetList(), key=lambda c: c.name)
|
|
|
|
def DoNew(self, name):
|
|
sIS = ImplantSets.getInstance()
|
|
return sIS.newSet(name)
|
|
|
|
def DoRename(self, entity, name):
|
|
sIS = ImplantSets.getInstance()
|
|
sIS.renameSet(entity, name)
|
|
|
|
def DoCopy(self, entity, name):
|
|
sIS = ImplantSets.getInstance()
|
|
copy = sIS.copySet(entity)
|
|
sIS.renameSet(copy, name)
|
|
return copy
|
|
|
|
def DoDelete(self, entity):
|
|
sIS = ImplantSets.getInstance()
|
|
sIS.deleteSet(entity)
|
|
|
|
|
|
class ImplantSetEditorView(BaseImplantEditorView):
|
|
|
|
def __init__(self, parent):
|
|
BaseImplantEditorView.__init__(self, parent)
|
|
if 'wxMSW' in wx.PlatformInfo:
|
|
self.SetBackgroundColour(wx.SystemSettings.GetColour(wx.SYS_COLOUR_BTNFACE))
|
|
|
|
def bindContext(self):
|
|
self.Parent.entityEditor.Bind(wx.EVT_CHOICE, self.contextChanged)
|
|
|
|
def getImplantsFromContext(self):
|
|
sIS = ImplantSets.getInstance()
|
|
set_ = self.Parent.entityEditor.getActiveEntity()
|
|
if set_:
|
|
return sIS.getImplants(set_.ID)
|
|
return []
|
|
|
|
def addImplantToContext(self, item):
|
|
sIS = ImplantSets.getInstance()
|
|
set_ = self.Parent.entityEditor.getActiveEntity()
|
|
|
|
sIS.addImplants(set_.ID, item.ID)
|
|
|
|
def removeImplantFromContext(self, implant):
|
|
sIS = ImplantSets.getInstance()
|
|
set_ = self.Parent.entityEditor.getActiveEntity()
|
|
|
|
sIS.removeImplant(set_.ID, implant)
|
|
|
|
|
|
class ImplantSetEditor(AuxiliaryFrame):
|
|
|
|
def __init__(self, parent, dataToAdd=None):
|
|
super().__init__(
|
|
parent, id=wx.ID_ANY, title="Implant Set Editor", resizeable=True,
|
|
size=wx.Size(950, 500) if "wxGTK" in wx.PlatformInfo else wx.Size(850, 420))
|
|
|
|
self.block = False
|
|
self.SetSizeHints(wx.DefaultSize, wx.DefaultSize)
|
|
|
|
mainSizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
self.entityEditor = ImplantSetEntityEditor(self)
|
|
mainSizer.Add(self.entityEditor, 0, wx.ALL | wx.EXPAND, 2)
|
|
|
|
self.sl = wx.StaticLine(self)
|
|
mainSizer.Add(self.sl, 0, wx.EXPAND | wx.TOP | wx.BOTTOM, 5)
|
|
|
|
self.iview = ImplantSetEditorView(self)
|
|
mainSizer.Add(self.iview, 1, wx.ALL | wx.EXPAND, 5)
|
|
|
|
self.slfooter = wx.StaticLine(self)
|
|
mainSizer.Add(self.slfooter, 0, wx.EXPAND | wx.TOP, 5)
|
|
|
|
footerSizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
|
|
self.stNotice = wx.StaticText(self, wx.ID_ANY, "")
|
|
self.stNotice.Wrap(-1)
|
|
footerSizer.Add(self.stNotice, 1, wx.BOTTOM | wx.TOP | wx.LEFT, 5)
|
|
|
|
importExport = (("Import", wx.ART_FILE_OPEN, "from"),
|
|
("Export", wx.ART_FILE_SAVE_AS, "to"))
|
|
|
|
for name, art, direction in importExport:
|
|
bitmap = wx.ArtProvider.GetBitmap(art, wx.ART_BUTTON)
|
|
btn = wx.BitmapButton(self, wx.ID_ANY, bitmap)
|
|
|
|
btn.SetMinSize(btn.GetSize())
|
|
btn.SetMaxSize(btn.GetSize())
|
|
|
|
btn.Layout()
|
|
setattr(self, name, btn)
|
|
btn.Enable(True)
|
|
btn.SetToolTip("%s implant sets %s clipboard" % (name, direction))
|
|
footerSizer.Add(btn, 0, wx.ALIGN_CENTER_HORIZONTAL | wx.ALIGN_RIGHT)
|
|
|
|
mainSizer.Add(footerSizer, 0, wx.ALL | wx.EXPAND, 5)
|
|
|
|
self.SetSizer(mainSizer)
|
|
self.Layout()
|
|
|
|
if dataToAdd:
|
|
name, implants = dataToAdd
|
|
newSet = self.entityEditor.DoNew(name)
|
|
ImplantSets.getInstance().addImplants(newSet.ID, *[i.item.ID for i in implants])
|
|
self.entityEditor.refreshEntityList(newSet)
|
|
wx.PostEvent(self.entityEditor.entityChoices, wx.CommandEvent(wx.wxEVT_COMMAND_CHOICE_SELECTED))
|
|
elif not self.entityEditor.checkEntitiesExist():
|
|
self.Close()
|
|
return
|
|
|
|
self.Bind(wx.EVT_CHOICE, self.entityChanged)
|
|
self.Bind(wx.EVT_CHAR_HOOK, self.kbEvent)
|
|
|
|
self.Import.Bind(wx.EVT_BUTTON, self.importPatterns)
|
|
self.Export.Bind(wx.EVT_BUTTON, self.exportPatterns)
|
|
|
|
self.SetMinSize(self.GetSize())
|
|
self.CenterOnParent()
|
|
|
|
def entityChanged(self, event):
|
|
if not self.entityEditor.checkEntitiesExist():
|
|
self.Close()
|
|
return
|
|
|
|
def kbEvent(self, event):
|
|
if event.GetKeyCode() == wx.WXK_ESCAPE and event.GetModifiers() == wx.MOD_NONE:
|
|
self.Close()
|
|
return
|
|
event.Skip()
|
|
|
|
def __del__(self):
|
|
pass
|
|
|
|
def importPatterns(self, event):
|
|
"""Event fired when import from clipboard button is clicked"""
|
|
|
|
text = fromClipboard()
|
|
if text:
|
|
sIS = ImplantSets.getInstance()
|
|
try:
|
|
sIS.importSets(text)
|
|
self.stNotice.SetLabel("Patterns successfully imported from clipboard")
|
|
except ImportError as e:
|
|
pyfalog.error(e)
|
|
self.stNotice.SetLabel(str(e))
|
|
except Exception as e:
|
|
pyfalog.error(e)
|
|
self.stNotice.SetLabel("Could not import from clipboard: unknown errors")
|
|
finally:
|
|
self.entityEditor.refreshEntityList()
|
|
else:
|
|
self.stNotice.SetLabel("Could not import from clipboard")
|
|
|
|
def exportPatterns(self, event):
|
|
"""Event fired when export to clipboard button is clicked"""
|
|
|
|
sIS = ImplantSets.getInstance()
|
|
toClipboard(sIS.exportSets())
|
|
self.stNotice.SetLabel("Sets exported to clipboard")
|