From 741aa3b5d5b637fff79a5beb3750bc3f913f9d78 Mon Sep 17 00:00:00 2001
From: Allanis <allanis@saracraft.net>
Date: Tue, 18 Jun 2013 21:37:41 +0100
Subject: [PATCH] [Change] Imroved sanity somewhat.

---
 utils/resedit/space.glade |  4 ++--
 utils/resedit/space.py    | 24 +++++++++++++++++++++++-
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/utils/resedit/space.glade b/utils/resedit/space.glade
index 7801c51..c27ed3d 100644
--- a/utils/resedit/space.glade
+++ b/utils/resedit/space.glade
@@ -441,9 +441,9 @@
                   </packing>
                 </child>
                 <child>
-                  <widget class="GtkComboBox" id="comFleet">
+                  <widget class="GtkComboBox" id="comFleets">
                     <property name="width_request">189</property>
-                    <property name="height_request">20</property>
+                    <property name="height_request">25</property>
                     <property name="visible">True</property>
                   </widget>
                   <packing>
diff --git a/utils/resedit/space.py b/utils/resedit/space.py
index 651e80c..f9c227c 100644
--- a/utils/resedit/space.py
+++ b/utils/resedit/space.py
@@ -21,6 +21,7 @@ class Space:
        self.factions = factions
      if fleets==None:
        self.fleets = {}
+     else:
        self.fleets = fleets
 
   def loadSystems(self, xmlfile):
@@ -92,6 +93,18 @@ class Space:
     area.connect("button-press-event", self.__space_down)
     area.connect("motion-notify-event", self.__space_drag)
 
+    # Factions.
+    wgt = self.__swidget("comFleets")
+    combo = gtk.ListStore(str)
+    combo.append(["None"])
+    for f in self.fleets.keys():
+      node = combo.append([f])
+    cell = gtk.CellRendererText()
+    wgt.pack_start(cell, True)
+    wgt.add_attribute(cell, 'text', 0)
+    wgt.set_model(combo)
+    wgt.set_active(0)
+
     # Display the window and such.
     self.__swidget("winSystems").show_all()
     self.cur_system = ""
@@ -129,6 +142,7 @@ class Space:
     wgt.pack_start(cell, True)
     wgt.add_attribute(cell, 'text', 0)
     wgt.set_model(combo)
+    wgt.set_active(0)
 
     # Factions.
     wgt = self.__pwidget("comFaction")
@@ -140,6 +154,7 @@ class Space:
     wgt.pack_start(cell, True)
     wgt.add_attribute(cell, "text", 0)
     wgt.set_model(combo)
+    wgt.set_active(0)
 
     # ---------------------------------------------
 
@@ -626,7 +641,11 @@ class Space:
     return model.get_value(iter, 0)
 
   def __fleet_add(self, wgt=None, event=None):
-    return
+    fleet = self.__swidget("comFleets").get_active_text()
+    value = self.__swidget("spiFleets").get_value_as_int()
+    if fleet != "None" and value > 0:
+      self.systems[self.cur_system]["fleets"][fleet] = value
+      self.__supdate()
 
   def __fleet_rm(self, wgt=None, event=None):
     sel = self.__fleet_sel()
@@ -700,6 +719,9 @@ class Space:
     self.__pupdate()
 
   def __pnewFact(self, wgt=None, event=None):
+    if self.cur_planet == "":
+      return
+
     combo = self.__pwidget("comFaction")
     fact = combo.get_active_text()
     planet = self.planets[self.cur_planet]