Get a working UI skeleton going for assigning boost fits

eos/db/saveddata/fit.py

@@ -53,6 +53,12 @@ projectedFits_table = Table("projectedFits", saveddata_meta,
                             Column("active", Boolean, nullable = False, default = 1),
 )
 
+commandFits_table = Table("commandFits", saveddata_meta,
+                            Column("boosterID", ForeignKey("fits.ID"), primary_key = True),
+                            Column("boostedID", ForeignKey("fits.ID"), primary_key = True),
+                            Column("active", Boolean, nullable = False, default = 1)
+)
+
 class ProjectedFit(object):
     def __init__(self, sourceID, source_fit, amount=1, active=True):
         self.sourceID = sourceID
@@ -83,12 +89,38 @@ class ProjectedFit(object):
             self.sourceID, self.victimID, self.amount, self.active, hex(id(self))
         )
 
+class CommandFit(object):
+    def __init__(self, boosterID, booster_fit, active=True):
+        self.boosterID = boosterID
+        self.booster_fit = booster_fit
+        self.active = active
+
+    @reconstructor
+    def init(self):
+        if self.booster_fit.isInvalid:
+            # Very rare for this to happen, but be prepared for it
+            eos.db.saveddata_session.delete(self.booster_fit)
+            eos.db.saveddata_session.flush()
+            eos.db.saveddata_session.refresh(self.boosted_fit)
+
+    def __repr__(self):
+        return "CommandFit(boosterID={}, boostedID={}, active={}) at {}".format(
+            self.boosterID, self.boostedID, self.active, hex(id(self))
+        )
+
+
 Fit._Fit__projectedFits = association_proxy(
     "victimOf",  # look at the victimOf association...
     "source_fit",  # .. and return the source fits
     creator=lambda sourceID, source_fit: ProjectedFit(sourceID, source_fit)
 )
 
+Fit._Fit__commandFits = association_proxy(
+    "boostedOf",  # look at the boostedOf association...
+    "booster_fit",  # .. and return the booster fit
+    creator=lambda boosterID, booster_fit: CommandFit(boosterID, booster_fit)
+)
+
 mapper(Fit, fits_table,
         properties = {
             "_Fit__modules": relation(
@@ -169,6 +201,18 @@ mapper(Fit, fits_table,
                 backref='victim_fit',
                 collection_class=attribute_mapped_collection('sourceID'),
                 cascade='all, delete, delete-orphan'),
+            "boostedOnto": relationship(
+                CommandFit,
+                primaryjoin=commandFits_table.c.boosterID == fits_table.c.ID,
+                backref='booster_fit',
+                collection_class=attribute_mapped_collection('boostedID'),
+                cascade='all, delete, delete-orphan'),
+            "boostedOf": relationship(
+                CommandFit,
+                primaryjoin=fits_table.c.ID == commandFits_table.c.boostedID,
+                backref='boosted_fit',
+                collection_class=attribute_mapped_collection('boosterID'),
+                cascade='all, delete, delete-orphan'),
        }
 )
 
@@ -177,3 +221,5 @@ mapper(ProjectedFit, projectedFits_table,
             "_ProjectedFit__amount": projectedFits_table.c.amount,
        }
 )
+
+mapper(CommandFit, commandFits_table)

eos/saveddata/fit.py

@@ -232,9 +232,16 @@ class Fit(object):
         # not return them.
         return [fit for fit in self.__projectedFits.values() if not fit.isInvalid]
 
+    @property
+    def commandFits(self):
+        return [fit for fit in self.__commandFits.values() if not fit.isInvalid]
+
     def getProjectionInfo(self, fitID):
         return self.projectedOnto.get(fitID, None)
 
+    def getCommandInfo(self, fitID):
+        return self.boostedOnto.get(fitID, None)
+
     @property
     def projectedDrones(self):
         return self.__projectedDrones