Implement the + button in stage 2 and make the fit count work

gui/shipBrowser.py

@@ -72,11 +72,12 @@ class ShipBrowser(wx.Panel):
     def stage2(self, event):
         categoryID = event.categoryID
         sMarket = service.Market.getInstance()
+        sFit = service.Fit.getInstance()
         self.lpane.RemoveAllChildren()
         shipList = sMarket.getShipList(categoryID)
         shipList.sort(key=self.raceNameKey)
         for ID, name, race in shipList:
-            self.lpane.AddWidget(ShipItem(self.lpane, ID, (name, 0), race))
+            self.lpane.AddWidget(ShipItem(self.lpane, ID, (name, len(sFit.getFitsWithShip(ID))), race))
 
         self.lpane.RefreshList()
         self.Show()
@@ -379,7 +380,8 @@ class ShipItem(wx.Window):
         event.Skip()
 
     def createNewFit(self, event=None):
-        print "New :", self.tcFitName.GetValue(), "GTFO from stage2 to stage 3 (refresh stage 3)"
+        sFit = service.Fit.getInstance()
+        sFit.newFit(self.shipID, self.tcFitName.GetValue())
         self.tcFitName.Show(False)
         self.editWasShown = 0
         wx.PostEvent(self.shipBrowser,Stage3Selected(shipID=self.shipID))
@@ -586,8 +588,6 @@ class FitItem(wx.Window):
                 self.editWasShown = 0
                 self.Refresh()
 
-
-
         event.Skip()
 
     def renameFit(self, event=None):

service/fit.py

@@ -45,10 +45,10 @@ class Fit(object):
         fit = eos.db.getFit(fitID)
         return fit.modules[pos]
 
-    def newFit(self, shipID, name):
+    def newFit(self, shipID, name=None):
         fit = eos.types.Fit()
         fit.ship = eos.types.Ship(eos.db.getItem(shipID))
-        fit.name = name
+        fit.name = name if name is not None else "New %s" % fit.ship.item.name
         fit.damagePattern = DamagePattern.getInstance().getDamagePattern("Uniform")
         eos.db.save(fit)
         fit.calculateModifiedAttributes()