ASPN ActiveState Programmer Network  
ActiveState, a division of Sophos
/ Home / Perl / PHP / Python / Tcl / XSLT /
/ Safari / My ASPN /
Cookbooks | Documentation | Mailing Lists | Modules | News Feeds | Products | User Groups
Submit Recipe
My Recipes

All Recipes
All Cookbooks


View by Category

Title: getch()-like unbuffered character reading from stdin on both Windows and Unix
Submitter: Danny Yoo (other recipes)
Last Updated: 2003/01/07
Version no: 1.1
Category: System

 

3 stars 3 vote(s)


Description:

A small utility class to read single characters from standard input, on both Windows and UNIX systems. It provides a getch() function-like instance.

Source: Text Source

class _Getch:
    """Gets a single character from standard input.  Does not echo to the
screen."""
    def __init__(self):
        try:
            self.impl = _GetchWindows()
        except ImportError:
            self.impl = _GetchUnix()

    def __call__(self): return self.impl()


class _GetchUnix:
    def __init__(self):
        import tty, sys

    def __call__(self):
        import sys, tty, termios
        fd = sys.stdin.fileno()
        old_settings = termios.tcgetattr(fd)
        try:
            tty.setraw(sys.stdin.fileno())
            ch = sys.stdin.read(1)
        finally:
            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
        return ch


class _GetchWindows:
    def __init__(self):
        import msvcrt

    def __call__(self):
        import msvcrt
        return msvcrt.getch()


getch = _Getch()

Discussion:



Add comment

Number of comments: 6

old python versions have to include TERMIOS, Tobias Polzin, 2002/10/01
If you use an old python version (e.g. 1.5) you have to include "TERMIOS" additionally to "termios" and the "TCSADRAIN" is found in "TERMIOS".
Add comment

extended for MacOS, C Smith, 2003/05/18

A few modifications make this work for the mac with Carbon,
too: a modification to the _Getch to try one more import, a 
modification to the Unix init, and the inclusion of the instructions
for the MacOS with Carbon support.


class _Getch:
    """Gets a single character from standard input.  Does not echo to the
screen."""
    def __init__(self):
        try:
            self.impl = _GetchWindows()
        except ImportError:
            try:
	        self.impl = _GetchUnix()
            except ImportError:
                self.impl = _GetchMacCarbon()

    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 sys, tty, termios
        fd = sys.stdin.fileno()
        old_settings = termios.tcgetattr(fd)
        try:
            tty.setraw(sys.stdin.fileno())
            ch = sys.stdin.read(1)
        finally:
            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
        return ch

class _GetchWindows:
    def __init__(self):
        import msvcrt

    def __call__(self):
        import msvcrt
        return msvcrt.getch()


class _GetchMacCarbon:
	"""
	A function which returns the current ASCII key that is down;
	if no ASCII key is down, the null string is returned.  The
	page http://www.mactech.com/macintosh-c/chap02-1.html was
	very helpful in figuring out how to do this.  
	"""
	def __init__(self):
		import Carbon
		
	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 
A few modifications make this work for the mac with Carbon,
too: a modification to the _Getch to try one more import, a 
modification to the Unix init, and the inclusion of the instructions
for the MacOS with Carbon support.


class _Getch:
    """Gets a single character from standard input.  Does not echo to the
screen."""
    def __init__(self):
        try:
            self.impl = _GetchWindows()
        except ImportError:
            try:
	        self.impl = _GetchUnix()
            except ImportError:
                self.impl = _GetchMacCarbon()

    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 sys, tty, termios
        fd = sys.stdin.fileno()
        old_settings = termios.tcgetattr(fd)
        try:
            tty.setraw(sys.stdin.fileno())
            ch = sys.stdin.read(1)
        finally:
            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
        return ch

class _GetchWindows:
    def __init__(self):
        import msvcrt

    def __call__(self):
        import msvcrt
        return msvcrt.getch()


class _GetchMacCarbon:
	"""
	A function which returns the current ASCII key that is down;
	if no ASCII key is down, the null string is returned.  The
	page http://www.mactech.com/macintosh-c/chap02-1.html was
	very helpful in figuring out how to do this.  
	"""
	def __init__(self):
		import Carbon
		
	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 

Add comment

erratum for MacOS modification, C Smith, 2003/05/19

Sorry about the repeated text.  I see that the ampersand
inside the <pre> tag is what caused it...and it also caused
the last line to appear incorrectly.  The last line should
read

return chr(msg & 0x000000FF)

Add comment

updated for OS X, C Smith, 2005/05/04
The following code (along with the _GetUnix and _GetWindows above) gives the correct getch behavior whether imported in the pythonIDE or in a Terminal script on the Mac. The order of trial imports is changed in _Getch because when in the IDE, the Unix import was succeeding. A single line in the _GetMacCarbon was added to see if the Carbon module has the Evt method. When the import succeeds when in the Terminal, the Carbon library there does not have the Evt method and so the import fails and the _GetchUnix() line is executed.

I also found that the curses snippet at < http://www.pythonapocrypha.com/Chapter22/Chapter22.shtml > works in the Terminal but will cause the pythonIDE to quit without warning if you run it there.

###

class _Getch:
    """Gets a single character from standard input.  Does not echo to the
screen."""
    def __init__(self):
        try:
            self.impl = _GetchWindows()
        except ImportError:
            try:
                self.impl = _GetchMacCarbon()
            except AttributeError:
                self.impl = _GetchUnix()

    def __call__(self): return self.impl()

class _GetchMacCarbon:
    """
    A function which returns the current ASCII key that is down;
    if no ASCII key is down, the null string is returned.  The
    page http://www.mactech.com/macintosh-c/chap02-1.html was
    very helpful in figuring out how to do this.
    """
    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()
   import sys
   for i in xrange(sys.maxint):
      k=inkey()
      if k<>'':break
   print 'you pressed ',k
###

Add comment

getch with IDLE, Cullen Newsom, 2006/03/15
I was unable to make this work in IDLE on OSX. I guess because of what IDLE seems to do to stdio. I always got "AttributeError".
Add comment

Carbon less invasive?, qubodup qubodup, 2008/03/29
Hi, I tried the code on linux and got " ImportError: No module named Carbon" You should wrap the carbon loading in a "try:" I suppose (I'm not experienced enough to post an example, as I eg. don't know what to write into "except ImportError:" if I want nothing to happen.
Add comment



Highest rated recipes:

1. A simple XML-RPC server

2. Web service accessible ...

3. Wrapping template engine ...

4. Assignment in expression

5. SOLVING THE METACLASS ...

6. Povray for python

7. Calling Windows API ...

8. Generic filter logic ...

9. Function Decorators by ...

10. MS SQL Server log monitor




Privacy Policy | Email Opt-out | Feedback | Syndication
© 2006 ActiveState Software Inc. All rights reserved.