fifth.py

# Create a SpaceX class.
# it should take 1 parameter in its constructor: the stored fuel
#
# it should have 4 methods:
#
# addRocket(rocket)
# it should add a new rocket to SpaceX that is given in its first parameter
# use the rockets from the fourth exercise.
#
# refill_all()
# it should refill all of its rockets with fuel and substract the needed fuel from the storage
#
# launch_all()
# it should launch all the rockets
#
# buy_fuel(amount)
# it should increase stored fuel by amount
#
# getStats()
# it should return its stats as a sting like: "rockets: 3, fuel: 100, launches: 1"

################################################


from fourth import Rocket

# class implementation goes here


class SpaceX:

    def __init__(self, storedFuel):
        self.storedFuel = storedFuel
        self.rocketList = []

    def addRocket(self, rocket):
        self.rocketList.append(rocket)

    def refill_all(self):
        for rocket in self.rocketList:
            self.storedFuel -= rocket.refill()

    def launch_all(self):
        for rocket in self.rocketList:
            rocket.launch()

    def buy_fuel(self, amount):
        self.storedFuel += amount

    def getStats(self):
        allLaunches = 0
        for rocket in self.rocketList:
            allLaunches += rocket.launches
        return 'rockets: {}, fuel: {}, launches: {}'.format(len(self.rocketList), self.storedFuel, allLaunches)


# The following code should work with the class:


space_x = SpaceX(100)
falcon1 = Rocket('falcon1', 0, 0)
falcon9 = Rocket('falcon9', 0, 0)
returned_falcon9 = Rocket('falcon9', 11, 1)

print(returned_falcon9.getStats()) # name: falcon9, fuel: 11

space_x.addRocket(falcon1)
space_x.addRocket(falcon9)
space_x.addRocket(returned_falcon9)

print(space_x.getStats()) # rockets: 3, fuel: 100, launches: 1
space_x.refill_all()
print(space_x.getStats()) # rockets: 3, fuel: 66, launches: 1
space_x.launch_all()
print(space_x.getStats()) # rockets: 3, fuel: 66, launches: 4
space_x.buy_fuel(50)
print(space_x.getStats()) # rockets: 3, fuel: 116, launches: 4