-
Notifications
You must be signed in to change notification settings - Fork 0
/
getkey.py
70 lines (65 loc) · 2.27 KB
/
getkey.py
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
class Getch:
def __init__(self):
import platform
os = platform.system()
if os == "Windows":
self.impl = _GetchWindows()
elif os == "Linux":
self.impl = _GetchUnix()
elif os == "Darwin":
try:
self.impl = _GetchMacCarbon()
except:
self.impl = _GetchUnix()
def __call__(self): return self.impl()
class _GetchUnix:
def __init__(self):
import tty, sys, termios # import termios now or else you'll get the Unix version on the Mac
def __call__(self):
import os
import sys, tty, termios
#fd = sys.stdin.fileno()
#old_settings = termios.tcgetattr(fd)
try:
#tty.setraw(sys.stdin.fileno())
#tty.setraw(fd)
os.system("stty cbreak -echo")
ch = sys.stdin.read(1)
finally:
#termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
os.system("stty -cbreak echo")
return ch
class _GetchWindows:
def __init__(self):
import msvcrt
def __call__(self):
import msvcrt
return msvcrt.getch()
class _GetchMacCarbon:
def __init__(self):
import Carbon
Carbon.Evt #see if it has this (in Unix, it doesn't)
def __call__(self):
import Carbon
if Carbon.Evt.EventAvail(0x0008)[0]==0: # 0x0008 is the keyDownMask
return ''
else:
#
# The event contains the following info:
# (what,msg,when,where,mod)=Carbon.Evt.GetNextEvent(0x0008)[1]
#
# The message (msg) contains the ASCII char which is
# extracted with the 0x000000FF charCodeMask; this
# number is converted to an ASCII character with chr() and
# returned
#
(what,msg,when,where,mod)=Carbon.Evt.GetNextEvent(0x0008)[1]
return chr(msg & 0x000000FF)
if __name__ == '__main__': # a little test
print 'Press a key'
inkey = Getch()
while True:
k = inkey()
if k=='x':
break
print 'you pressed ',k,ord(k)