From 2f71e1ff32f97715e5e073f1bf8f2a9caa3f6945 Mon Sep 17 00:00:00 2001
From: cncfanatics <diego.duclos@gmail.com>
Date: Fri, 20 Aug 2010 21:06:27 +0200
Subject: [PATCH] Make the delete button work

---
 controller/fit.py  | 6 ++++++
 gui/shipBrowser.py | 9 +++++++++
 2 files changed, 15 insertions(+)

diff --git a/controller/fit.py b/controller/fit.py
index 38b5c2da2..44264dce0 100644
--- a/controller/fit.py
+++ b/controller/fit.py
@@ -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()
+
diff --git a/gui/shipBrowser.py b/gui/shipBrowser.py
index a1a2377e2..2bf4ee763 100644
--- a/gui/shipBrowser.py
+++ b/gui/shipBrowser.py
@@ -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)