-
Notifications
You must be signed in to change notification settings - Fork 12
/
fifth.py
76 lines (59 loc) · 2.06 KB
/
fifth.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
# 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"
################################################
# The following code should work with the class:
from fourth import Rocket
# class implementation goes here
class SpaceX:
def __init__(self, stored_fuel):
self.stored_fuel = stored_fuel
self.spacex_rockets = []
self.all_launch = 0
def addRocket(self, rocket):
self.spacex_rockets.append(rocket)
def refill_all(self):
for rocket in self.spacex_rockets:
rocket.refill()
used_fuel = rocket.refill()
self.stored_fuel -= used_fuel
return used_fuel
def launch_all(self):
for _ in self.spacex_rockets:
self.all_launch += 1
def buy_fuel(self, amount):
self.stored_fuel += amount
def getStats(self):
return "rockets: {}, fuel: {}, launches: {}".format(len(self.spacex_rockets), self.stored_fuel, self.all_launch+1)
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