summaryrefslogtreecommitdiff
path: root/examples/statemachine/video_demo.py
blob: fadfb9d6c68ab9b2af1166e5dde0bc3fe5b21249 (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
#
# video_demo.py
#
# Simple statemachine demo, based on the state transitions given in videostate.pystate
#

import statemachine
import videostate


class Video(videostate.VideoStateMixin):
    def __init__(self, title):
        self.initialize_state(videostate.Stopped)
        self.title = title


# ==== main loop - a REPL ====

v = Video("Die Hard.mp4")

while True:
    print(v.state)
    cmd = input("Command ({})> ".format('/'.join(videostate.VideoState.transition_names))).lower().strip()
    if not cmd:
        continue

    if cmd in ('?', 'h', 'help'):
        print('enter a transition {!r}'.format(videostate.VideoState.transition_names))
        print(' q - quit')
        print(' ?, h, help - this message')
        continue

    # quitting out
    if cmd.startswith('q'):
        break

    # get transition function for given command
    state_transition_fn = getattr(v, cmd, None)

    if state_transition_fn is None:
        print('???')
        continue

    # invoke the input transition, handle invalid commands
    try:
        state_transition_fn()
    except videostate.VideoState.InvalidTransitionException as e:
        print(e)