Make the delete button work

This commit is contained in:
cncfanatics
2010-08-20 21:06:27 +02:00
parent 8dba43d8ea
commit 2f71e1ff32
2 changed files with 15 additions and 0 deletions

View File

@@ -49,3 +49,9 @@ class Fit(object):
fit = eos.db.getFit(fitID)
fit.name = newName
eos.db.saveddata_session.flush()
def deleteFit(self, fitID):
fit = eos.db.getFit(fitID)
eos.db.saveddata_session.delete(fit)
eos.db.saveddata_session.flush()

View File

@@ -43,6 +43,7 @@ class ShipBrowser(wx.Panel):
#Bind buttons
self.shipMenu.new.Bind(wx.EVT_BUTTON, self.newFit)
self.shipMenu.rename.Bind(wx.EVT_BUTTON, self.renameFit)
self.shipMenu.delete.Bind(wx.EVT_BUTTON, self.deleteFit)
def build(self):
if not self.built:
@@ -127,6 +128,14 @@ class ShipBrowser(wx.Panel):
cFit = controller.Fit.getInstance()
cFit.renameFit(fitID, newName)
def deleteFit(self, event):
root = self.shipView.GetSelection()
type, fitID = self.shipView.GetPyData(root)
if type == "fit":
cFit = controller.Fit.getInstance()
cFit.deleteFit(fitID)
self.shipView.Delete(root)
class ShipView(wx.TreeCtrl):
def __init__(self, parent):
wx.TreeCtrl.__init__(self, parent)