blob: b5fe5248564b06437321394ae2aa80d4c520bfb1 (
plain)
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
77
78
79
80
81
|
#
# vending_machine.py
#
# Example of using the statemachine parser without importing a .pystate module.
#
# A vending machine that dispenses candy and chips in a 4x4 grid, A1 thru D4.
# To dispense a product, you must press an alpha button, then a digit button.
#
import statemachine
# Vending machine buttons:
# A, B, C, D
# 1, 2, 3, 4
#
vending_machine_state_description = """\
statemachine VendingMachineState:
Idle-(press_alpha_button)->WaitingOnDigit
WaitingOnDigit-(press_alpha_button)->WaitingOnDigit
WaitingOnDigit-(press_digit_button)->DispenseProduct
DispenseProduct-(dispense)->Idle
"""
# convert state machine text to state classes
generated = statemachine.namedStateMachine.transformString(
vending_machine_state_description
)
# print(generated)
# exec generated code to define state classes and state mixin
exec(generated)
class VendingMachine(VendingMachineStateMixin):
def __init__(self):
self.initialize_state(Idle)
self._pressed = None
self._alpha_pressed = None
self._digit_pressed = None
def press_button(self, button):
if button in "ABCD":
self._pressed = button
self.press_alpha_button()
elif button in "1234":
self._pressed = button
self.press_digit_button()
else:
print("Did not recognize button {!r}".format(str(button)))
def press_alpha_button(self):
try:
super().press_alpha_button()
except VendingMachineState.InvalidTransitionException as ite:
print(ite)
else:
self._alpha_pressed = self._pressed
def press_digit_button(self):
try:
super().press_digit_button()
except VendingMachineState.InvalidTransitionException as ite:
print(ite)
else:
self._digit_pressed = self._pressed
self.dispense()
def dispense(self):
try:
super().dispense()
except VendingMachineState.InvalidTransitionException as ite:
print(ite)
else:
print("Dispensing at {}{}".format(self._alpha_pressed, self._digit_pressed))
self._alpha_pressed = self._digit_pressed = None
vm = VendingMachine()
for button in "1 A B 1".split():
print(">> pressing {!r}".format(button))
vm.press_button(button)
print("Vending machine is now in {} state".format(vm.state))
|
