93 lines
2.0 KiB
Python
93 lines
2.0 KiB
Python
#!/usr/bin/env python
|
|
import os
|
|
import sys
|
|
import termios
|
|
import atexit
|
|
from select import select
|
|
|
|
class KBHit:
|
|
|
|
def __init__(self):
|
|
'''Creates a KBHit object that you can call to do various keyboard things.
|
|
'''
|
|
|
|
self.set_kbhit_terminal()
|
|
|
|
def set_kbhit_terminal(self):
|
|
# Save the terminal settings
|
|
self.fd = sys.stdin.fileno()
|
|
self.new_term = termios.tcgetattr(self.fd)
|
|
self.old_term = termios.tcgetattr(self.fd)
|
|
|
|
# New terminal setting unbuffered
|
|
self.new_term[3] = (self.new_term[3] & ~termios.ICANON & ~termios.ECHO)
|
|
termios.tcsetattr(self.fd, termios.TCSAFLUSH, self.new_term)
|
|
|
|
# Support normal-terminal reset at exit
|
|
atexit.register(self.set_normal_term)
|
|
|
|
def set_normal_term(self):
|
|
''' Resets to normal terminal. On Windows this is a no-op.
|
|
'''
|
|
|
|
if os.name == 'nt':
|
|
pass
|
|
|
|
else:
|
|
termios.tcsetattr(self.fd, termios.TCSAFLUSH, self.old_term)
|
|
|
|
|
|
def getch(self):
|
|
''' Returns a keyboard character after kbhit() has been called.
|
|
Should not be called in the same program as getarrow().
|
|
'''
|
|
return sys.stdin.read(1)
|
|
|
|
|
|
def getarrow(self):
|
|
''' Returns an arrow-key code after kbhit() has been called. Codes are
|
|
0 : up
|
|
1 : right
|
|
2 : down
|
|
3 : left
|
|
Should not be called in the same program as getch().
|
|
'''
|
|
|
|
if os.name == 'nt':
|
|
msvcrt.getch() # skip 0xE0
|
|
c = msvcrt.getch()
|
|
vals = [72, 77, 80, 75]
|
|
|
|
else:
|
|
c = sys.stdin.read(3)[2]
|
|
vals = [65, 67, 66, 68]
|
|
|
|
return vals.index(ord(c.decode('utf-8')))
|
|
|
|
|
|
def kbhit(self):
|
|
''' Returns True if keyboard character was hit, False otherwise.
|
|
'''
|
|
dr,dw,de = select([sys.stdin], [], [], 0)
|
|
return dr != []
|
|
|
|
|
|
# Test
|
|
if __name__ == "__main__":
|
|
|
|
kb = KBHit()
|
|
|
|
print('Hit any key, or ESC to exit')
|
|
|
|
while True:
|
|
|
|
if kb.kbhit():
|
|
c = kb.getch()
|
|
if ord(c) == 27: # ESC
|
|
break
|
|
print(c)
|
|
|
|
kb.set_normal_term()
|
|
|
|
|