27 lines
632 B
Python
27 lines
632 B
Python
|
"""
|
||
|
|
>>> basketball_team = ['Sue', 'Tina', 'Maya', 'Diana', 'Pat']
|
||
|
|
>>> bus = TwilightBus(basketball_team)
|
||
|
|
>>> bus.drop('Tina')
|
||
|
|
>>> bus.drop('Pat')
|
||
|
|
>>> basketball_team
|
||
|
|
['Sue', 'Maya', 'Diana']
|
||
|
|
"""
|
||
|
|
|
||
|
|
# tag::TWILIGHT_BUS_CLASS[]
|
||
|
|
class TwilightBus:
|
||
|
|
"""A bus model that makes passengers vanish"""
|
||
|
|
|
||
|
|
def __init__(self, passengers=None):
|
||
|
|
if passengers is None:
|
||
|
|
self.passengers = [] # <1>
|
||
|
|
else:
|
||
|
|
self.passengers = passengers #<2>
|
||
|
|
|
||
|
|
def pick(self, name):
|
||
|
|
self.passengers.append(name)
|
||
|
|
|
||
|
|
def drop(self, name):
|
||
|
|
self.passengers.remove(name) # <3>
|
||
|
|
# end::TWILIGHT_BUS_CLASS[]
|
||
|
|
|