Also create fleet automatically on booster assignments

This commit is contained in:
DarkPhoenix
2011-11-21 21:22:20 +04:00
parent 69ba6c7792
commit 5c039579bb

View File

@@ -80,6 +80,9 @@ class Fleet(object):
eos.db.save(f)
def setLinearFleetCom(self, boostee, booster):
if self.isInLinearFleet(boostee) is False:
self.removeAssociatedFleetData(boostee)
self.makeLinearFleet(boostee)
squadIDs = set(eos.db.getSquadsIDsWithFitID(boostee.ID))
squad = eos.db.getSquad(squadIDs.pop())
squad.wing.gang.leader = booster
@@ -89,6 +92,9 @@ class Fleet(object):
self.removeAssociatedFleetData(boostee)
def setLinearWingCom(self, boostee, booster):
if self.isInLinearFleet(boostee) is False:
self.removeAssociatedFleetData(boostee)
self.makeLinearFleet(boostee)
squadIDs = set(eos.db.getSquadsIDsWithFitID(boostee.ID))
squad = eos.db.getSquad(squadIDs.pop())
squad.wing.leader = booster
@@ -98,6 +104,9 @@ class Fleet(object):
self.removeAssociatedFleetData(boostee)
def setLinearSquadCom(self, boostee, booster):
if self.isInLinearFleet(boostee) is False:
self.removeAssociatedFleetData(boostee)
self.makeLinearFleet(boostee)
squadIDs = set(eos.db.getSquadsIDsWithFitID(boostee.ID))
squad = eos.db.getSquad(squadIDs.pop())
squad.leader = booster