ECM2429-event-loop/code/example-5/curses-countdown.py at main · ExeterBScDTS/ECM2429-event-loop · GitHub

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
"""
    See https://theailearner.com/2019/03/10/snake-game-using-python-curses/
"""
import curses
from datetime import timedelta, datetime

def show_count(win, deadline: datetime):
    remaining = deadline - datetime.now()
    if remaining < timedelta(0):
        remaining = "     Launch     "
        deadline = None
    win.addstr(9, 7, f"{remaining}")
    return deadline


def start_countdown(seconds):
    return datetime.now() + timedelta(seconds=seconds)


def setup(stdsrc):
    stdsrc.clear()
    sc = curses.initscr()
    h, w = sc.getmaxyx()
    win = curses.newwin(h, w, 0, 0)
    curses.curs_set(0)
    win.border(0)

    win.addstr(3, 5, "s - start countdown")
    win.addstr(5, 5, "r - reset")
    win.addstr(7, 5, "c - cancel")
    win.addstr(7, 25, "x - exit")
    return win


def mainloop(stdscr):
    win = setup(stdscr)

    countdown_seconds = 10
    deadline = None

    while True:

        win.timeout(10)
        key = win.getch()

        if deadline:
            deadline = show_count(win, deadline)

        if key == -1:
            continue

        if key == ord('s'):
            # Start the countdown if it isn't already running
            if not deadline:
                deadline = start_countdown(countdown_seconds)
            else:
                curses.beep()
        elif key == ord('r'):
            # Restart the countdown if it is already running
            if deadline:
                deadline = start_countdown(countdown_seconds)
            else:
                curses.beep()
        elif key == ord('c'):
            # Cancel the countdown
            if deadline:
                deadline = None
            else:
                curses.beep()
        elif key == ord('x'):
            # Exit.  Only available if the countdown has completed.
            if not deadline:
                break
            curses.beep()
        else:
            curses.beep()


curses.wrapper(mainloop)
print("Goodbye")