Added Python function to emulate the ncurses has_key() function.

2 files changed, 197 insertions, 0 deletions

diff --git a/Lib/curses/__init__.py b/Lib/curses/__init__.py
index 1808ee6..6bcd566 100644
--- a/Lib/curses/__init__.py
+++ b/Lib/curses/__init__.py
@@ -32,3 +32,10 @@ def initscr():
     
     return stdscr
 
+# Import Python has_key() implementation if _curses doesn't contain has_key()
+
+try:
+    has_key
+except NameError:
+    from has_key import has_key
+
diff --git a/Lib/curses/has_key.py b/Lib/curses/has_key.py
new file mode 100644
index 0000000..b104d85
--- /dev/null
+++ b/Lib/curses/has_key.py
@@ -0,0 +1,190 @@
+
+#
+# Emulation of has_key() function for platforms that don't use ncurses
+#
+
+import _curses
+
+# Table mapping curses keys to the terminfo capability name
+
+_capability_names = {
+    _curses.KEY_A1: 'ka1', 
+    _curses.KEY_A3: 'ka3', 
+    _curses.KEY_B2: 'kb2', 
+    _curses.KEY_BACKSPACE: 'kbs', 
+    _curses.KEY_BEG: 'kbeg', 
+    _curses.KEY_BTAB: 'kcbt', 
+    _curses.KEY_C1: 'kc1', 
+    _curses.KEY_C3: 'kc3', 
+    _curses.KEY_CANCEL: 'kcan', 
+    _curses.KEY_CATAB: 'ktbc', 
+    _curses.KEY_CLEAR: 'kclr', 
+    _curses.KEY_CLOSE: 'kclo', 
+    _curses.KEY_COMMAND: 'kcmd', 
+    _curses.KEY_COPY: 'kcpy', 
+    _curses.KEY_CREATE: 'kcrt', 
+    _curses.KEY_CTAB: 'kctab', 
+    _curses.KEY_DC: 'kdch1', 
+    _curses.KEY_DL: 'kdl1', 
+    _curses.KEY_DOWN: 'kcud1', 
+    _curses.KEY_EIC: 'krmir', 
+    _curses.KEY_END: 'kend', 
+    _curses.KEY_ENTER: 'kent', 
+    _curses.KEY_EOL: 'kel', 
+    _curses.KEY_EOS: 'ked', 
+    _curses.KEY_EXIT: 'kext', 
+    _curses.KEY_F0: 'kf0', 
+    _curses.KEY_F1: 'kf1', 
+    _curses.KEY_F10: 'kf10', 
+    _curses.KEY_F11: 'kf11', 
+    _curses.KEY_F12: 'kf12', 
+    _curses.KEY_F13: 'kf13', 
+    _curses.KEY_F14: 'kf14', 
+    _curses.KEY_F15: 'kf15', 
+    _curses.KEY_F16: 'kf16', 
+    _curses.KEY_F17: 'kf17', 
+    _curses.KEY_F18: 'kf18', 
+    _curses.KEY_F19: 'kf19', 
+    _curses.KEY_F2: 'kf2', 
+    _curses.KEY_F20: 'kf20', 
+    _curses.KEY_F21: 'kf21', 
+    _curses.KEY_F22: 'kf22', 
+    _curses.KEY_F23: 'kf23', 
+    _curses.KEY_F24: 'kf24', 
+    _curses.KEY_F25: 'kf25', 
+    _curses.KEY_F26: 'kf26', 
+    _curses.KEY_F27: 'kf27', 
+    _curses.KEY_F28: 'kf28', 
+    _curses.KEY_F29: 'kf29', 
+    _curses.KEY_F3: 'kf3', 
+    _curses.KEY_F30: 'kf30', 
+    _curses.KEY_F31: 'kf31', 
+    _curses.KEY_F32: 'kf32', 
+    _curses.KEY_F33: 'kf33', 
+    _curses.KEY_F34: 'kf34', 
+    _curses.KEY_F35: 'kf35', 
+    _curses.KEY_F36: 'kf36', 
+    _curses.KEY_F37: 'kf37', 
+    _curses.KEY_F38: 'kf38', 
+    _curses.KEY_F39: 'kf39', 
+    _curses.KEY_F4: 'kf4', 
+    _curses.KEY_F40: 'kf40', 
+    _curses.KEY_F41: 'kf41', 
+    _curses.KEY_F42: 'kf42', 
+    _curses.KEY_F43: 'kf43', 
+    _curses.KEY_F44: 'kf44', 
+    _curses.KEY_F45: 'kf45', 
+    _curses.KEY_F46: 'kf46', 
+    _curses.KEY_F47: 'kf47', 
+    _curses.KEY_F48: 'kf48', 
+    _curses.KEY_F49: 'kf49', 
+    _curses.KEY_F5: 'kf5', 
+    _curses.KEY_F50: 'kf50', 
+    _curses.KEY_F51: 'kf51', 
+    _curses.KEY_F52: 'kf52', 
+    _curses.KEY_F53: 'kf53', 
+    _curses.KEY_F54: 'kf54', 
+    _curses.KEY_F55: 'kf55', 
+    _curses.KEY_F56: 'kf56', 
+    _curses.KEY_F57: 'kf57', 
+    _curses.KEY_F58: 'kf58', 
+    _curses.KEY_F59: 'kf59', 
+    _curses.KEY_F6: 'kf6', 
+    _curses.KEY_F60: 'kf60', 
+    _curses.KEY_F61: 'kf61', 
+    _curses.KEY_F62: 'kf62', 
+    _curses.KEY_F63: 'kf63', 
+    _curses.KEY_F7: 'kf7', 
+    _curses.KEY_F8: 'kf8', 
+    _curses.KEY_F9: 'kf9', 
+    _curses.KEY_FIND: 'kfnd', 
+    _curses.KEY_HELP: 'khlp', 
+    _curses.KEY_HOME: 'khome', 
+    _curses.KEY_IC: 'kich1', 
+    _curses.KEY_IL: 'kil1', 
+    _curses.KEY_LEFT: 'kcub1', 
+    _curses.KEY_LL: 'kll', 
+    _curses.KEY_MARK: 'kmrk', 
+    _curses.KEY_MESSAGE: 'kmsg', 
+    _curses.KEY_MOVE: 'kmov', 
+    _curses.KEY_NEXT: 'knxt', 
+    _curses.KEY_NPAGE: 'knp', 
+    _curses.KEY_OPEN: 'kopn', 
+    _curses.KEY_OPTIONS: 'kopt', 
+    _curses.KEY_PPAGE: 'kpp', 
+    _curses.KEY_PREVIOUS: 'kprv', 
+    _curses.KEY_PRINT: 'kprt', 
+    _curses.KEY_REDO: 'krdo', 
+    _curses.KEY_REFERENCE: 'kref', 
+    _curses.KEY_REFRESH: 'krfr', 
+    _curses.KEY_REPLACE: 'krpl', 
+    _curses.KEY_RESTART: 'krst', 
+    _curses.KEY_RESUME: 'kres', 
+    _curses.KEY_RIGHT: 'kcuf1', 
+    _curses.KEY_SAVE: 'ksav', 
+    _curses.KEY_SBEG: 'kBEG', 
+    _curses.KEY_SCANCEL: 'kCAN', 
+    _curses.KEY_SCOMMAND: 'kCMD', 
+    _curses.KEY_SCOPY: 'kCPY', 
+    _curses.KEY_SCREATE: 'kCRT', 
+    _curses.KEY_SDC: 'kDC', 
+    _curses.KEY_SDL: 'kDL', 
+    _curses.KEY_SELECT: 'kslt', 
+    _curses.KEY_SEND: 'kEND', 
+    _curses.KEY_SEOL: 'kEOL', 
+    _curses.KEY_SEXIT: 'kEXT', 
+    _curses.KEY_SF: 'kind', 
+    _curses.KEY_SFIND: 'kFND', 
+    _curses.KEY_SHELP: 'kHLP', 
+    _curses.KEY_SHOME: 'kHOM', 
+    _curses.KEY_SIC: 'kIC', 
+    _curses.KEY_SLEFT: 'kLFT', 
+    _curses.KEY_SMESSAGE: 'kMSG', 
+    _curses.KEY_SMOVE: 'kMOV', 
+    _curses.KEY_SNEXT: 'kNXT', 
+    _curses.KEY_SOPTIONS: 'kOPT', 
+    _curses.KEY_SPREVIOUS: 'kPRV', 
+    _curses.KEY_SPRINT: 'kPRT', 
+    _curses.KEY_SR: 'kri', 
+    _curses.KEY_SREDO: 'kRDO', 
+    _curses.KEY_SREPLACE: 'kRPL', 
+    _curses.KEY_SRIGHT: 'kRIT', 
+    _curses.KEY_SRSUME: 'kRES', 
+    _curses.KEY_SSAVE: 'kSAV', 
+    _curses.KEY_SSUSPEND: 'kSPD', 
+    _curses.KEY_STAB: 'khts', 
+    _curses.KEY_SUNDO: 'kUND', 
+    _curses.KEY_SUSPEND: 'kspd', 
+    _curses.KEY_UNDO: 'kund', 
+    _curses.KEY_UP: 'kcuu1'
+    }
+
+def has_key(ch):
+    if type(ch) == type( '' ): ch = ord(ch)
+
+    # Figure out the correct capability name for the keycode.
+    capability_name = _capability_names[ch]
+
+    #Check the current terminal description for that capability;
+    #if present, return true, else return false.
+    if _curses.tigetstr( capability_name ): return 1
+    else: return 0
+
+if __name__ == '__main__':
+    # Compare the output of this implementation and the ncurses has_key,
+    # on platforms where has_key is already available
+    import _curses
+    try:
+        L = []
+        _curses.initscr()
+        for key in _capability_names.keys():
+            system = _curses.has_key(key)
+            python = has_key(key)
+            if system != python:
+                L.append( 'Mismatch for key %s, system=%i, Python=%i'
+                          % (_curses.keyname( key ), system, python) )
+    finally:
+        _curses.endwin()
+        for i in L: print i
+        
+