classState(object):"""Base state. This is to share functionality"""defscan(self):"""Scan the dial to the next station"""
self.pos += 1if self.pos == len(self.stations):
self.pos = 0
print(u"Scanning... Station is %s %s" %
(self.stations[self.pos], self.name))
classAmState(State):def__init__(self, radio):
self.radio = radio
self.stations = ["1250", "1380", "1510"]
self.pos = 0
self.name = "AM"deftoggle_amfm(self):
print(u"Switching to FM")
self.radio.state = self.radio.fmstate
classFmState(State):def__init__(self, radio):
self.radio = radio
self.stations = ["81.3", "89.1", "103.9"]
self.pos = 0
self.name = "FM"deftoggle_amfm(self):
print(u"Switching to AM")
self.radio.state = self.radio.amstate
classRadio(object):"""A radio. It has a scan button, and an AM/FM toggle switch."""def__init__(self):"""We have an AM state and an FM state"""
self.amstate = AmState(self)
self.fmstate = FmState(self)
self.state = self.amstate
deftoggle_amfm(self):
self.state.toggle_amfm()
defscan(self):
self.state.scan()
# Test our radio outif __name__ == '__main__':
radio = Radio()
actions = [radio.scan] * 2 + [radio.toggle_amfm] + [radio.scan] * 2
actions *= 2for action in actions:
action()
### OUTPUT #### Scanning... Station is 1380 AM# Scanning... Station is 1510 AM# Switching to FM# Scanning... Station is 89.1 FM# Scanning... Station is 103.9 FM# Scanning... Station is 81.3 FM# Scanning... Station is 89.1 FM# Switching to AM# Scanning... Station is 1250 AM# Scanning... Station is 1380 AM