#!/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()