blob: ee04e86dd08fbfb5f40f929a03c594dc235f7083 (
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
|
import datetime
import sys, termios, tty
import inputs
frequency = 120 # 120Hz refresh
def countdown(length):
"""countdown timer; length in seconds. returns start and time left (>0 if quit early)"""
time_left = length
start = datetime.datetime.now()
end = start + datetime.timedelta(seconds=time_left) # change to mins in final
accum = 0
timeout = 1/frequency
old_settings = termios.tcgetattr(sys.stdin)
tty.setcbreak(sys.stdin.fileno())
while True:
now = datetime.datetime.now()
time_left = end - now
if time_left.total_seconds() <= 0:
break
ch = inputs.getch_nb(timeout)
if ch == 'q':
print("\nexited early!")
break
# see https://gist.github.com/fnky/458719343aabd01cfb17a3a4f7296797
print("\x1b[2K\r" + inputs.delta_to_HM(time_left), end="", flush=True)
accum += timeout
if accum >= 1:
accum = 0
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_settings)
return (time_left, start)
def stopwatch():
"""stopwatch timer; returns time elapsed"""
start = datetime.datetime.now()
accum = 0
timeout = 1/frequency
old_settings = termios.tcgetattr(sys.stdin)
tty.setcbreak(sys.stdin.fileno())
while True:
elapsed = datetime.datetime.now() - start
ch = inputs.getch_nb(timeout)
if ch == 'q':
break
# see https://gist.github.com/fnky/458719343aabd01cfb17a3a4f7296797
print("\x1b[2K\r" + inputs.delta_to_HM(elapsed), end="", flush=True)
accum += timeout
if accum >= 1:
accum = 0
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_settings)
return (elapsed, start)
|
