From 91f32634e49f1284e1498134a537b39e356f27d7 Mon Sep 17 00:00:00 2001
From: cncfanatics <diego.duclos@gmail.com>
Date: Thu, 21 Oct 2010 21:03:37 +0200
Subject: [PATCH] Fix some search related stuff in the ship browser

---
 gui/shipBrowser.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/gui/shipBrowser.py b/gui/shipBrowser.py
index 0df0a5a2a..36fd3a382 100644
--- a/gui/shipBrowser.py
+++ b/gui/shipBrowser.py
@@ -124,6 +124,7 @@ class ShipBrowser(wx.Panel):
                 return
 
             type, fitID = data
+            print type
             if type == "fit":
                 for btn in btns:
                     btn.Enable()
@@ -166,17 +167,25 @@ class ShipBrowser(wx.Panel):
         tree = self.getActiveTree()
         root = tree.GetSelection()
         type, shipID = tree.GetPyData(root)
-        if type == "fit":
-            root = tree.GetItemParent(root)
-            type, shipID = tree.GetPyData(root)
-
-        name = "%s fit" % tree.GetItemText(root)
         cFit = service.Fit.getInstance()
+        noChildren = False
+        if type == "fit":
+            fit = cFit.getFit(shipID)
+            type, shipID = "ship", fit.ship.item.ID
+            name = "%s fit" % fit.ship.item.name
+            if tree == self.searchView:
+                root = self.searchRoot
+                noChildren = True
+        else:
+            name = "%s fit" % tree.GetItemText(root)
+
         fitID = cFit.newFit(shipID, name)
         childId = tree.AppendItem(root, name, self.fitIconId, data=wx.TreeItemData(("fit", fitID)))
         tree.SetItemText(childId, name)
-        tree.SortChildren(root)
-        tree.Expand(root)
+        if not noChildren:
+            tree.SortChildren(root)
+            tree.Expand(root)
+
         tree.SelectItem(childId)
         tree.EditLabel(childId)