dave/pyfa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

Remove fleet structure automatically, when last booster is removed

Browse Source
This commit is contained in:
DarkPhoenix
2011-11-21 21:19:04 +04:00
parent 7990aa07e3
commit 69ba6c7792
1 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
service/fleet.py
View File

@@ -83,19 +83,28 @@ class Fleet(object):
squadIDs = set(eos.db.getSquadsIDsWithFitID(boostee.ID))
squad = eos.db.getSquad(squadIDs.pop())
squad.wing.gang.leader = booster
squad.wing.gang.calculateModifiedAttributes()
if self.anyBoosters(squad):
squad.wing.gang.calculateModifiedAttributes()
else:
self.removeAssociatedFleetData(boostee)
def setLinearWingCom(self, boostee, booster):
squadIDs = set(eos.db.getSquadsIDsWithFitID(boostee.ID))
squad = eos.db.getSquad(squadIDs.pop())
squad.wing.leader = booster
squad.wing.gang.calculateModifiedAttributes()
if self.anyBoosters(squad):
squad.wing.gang.calculateModifiedAttributes()
else:
self.removeAssociatedFleetData(boostee)
def setLinearSquadCom(self, boostee, booster):
squadIDs = set(eos.db.getSquadsIDsWithFitID(boostee.ID))
squad = eos.db.getSquad(squadIDs.pop())
squad.leader = booster
squad.wing.gang.calculateModifiedAttributes()
if self.anyBoosters(squad):
squad.wing.gang.calculateModifiedAttributes()
else:
self.removeAssociatedFleetData(boostee)
def isInLinearFleet(self, fit):
sqIDs = eos.db.getSquadsIDsWithFitID(fit.ID)
@@ -137,3 +146,10 @@ class Fleet(object):
fleet = eos.db.getFleet(fleetID)
eos.db.remove(fleet)
return
def anyBoosters(self, squad):
wing = squad.wing
fleet = wing.gang
if squad.leader is None and wing.leader is None and fleet.leader is None:
return False
return True