has_key.py 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. #
  2. # Emulation of has_key() function for platforms that don't use ncurses
  3. #
  4. import _curses
  5. # Table mapping curses keys to the terminfo capability name
  6. _capability_names = {
  7. _curses.KEY_A1: 'ka1',
  8. _curses.KEY_A3: 'ka3',
  9. _curses.KEY_B2: 'kb2',
  10. _curses.KEY_BACKSPACE: 'kbs',
  11. _curses.KEY_BEG: 'kbeg',
  12. _curses.KEY_BTAB: 'kcbt',
  13. _curses.KEY_C1: 'kc1',
  14. _curses.KEY_C3: 'kc3',
  15. _curses.KEY_CANCEL: 'kcan',
  16. _curses.KEY_CATAB: 'ktbc',
  17. _curses.KEY_CLEAR: 'kclr',
  18. _curses.KEY_CLOSE: 'kclo',
  19. _curses.KEY_COMMAND: 'kcmd',
  20. _curses.KEY_COPY: 'kcpy',
  21. _curses.KEY_CREATE: 'kcrt',
  22. _curses.KEY_CTAB: 'kctab',
  23. _curses.KEY_DC: 'kdch1',
  24. _curses.KEY_DL: 'kdl1',
  25. _curses.KEY_DOWN: 'kcud1',
  26. _curses.KEY_EIC: 'krmir',
  27. _curses.KEY_END: 'kend',
  28. _curses.KEY_ENTER: 'kent',
  29. _curses.KEY_EOL: 'kel',
  30. _curses.KEY_EOS: 'ked',
  31. _curses.KEY_EXIT: 'kext',
  32. _curses.KEY_F0: 'kf0',
  33. _curses.KEY_F1: 'kf1',
  34. _curses.KEY_F10: 'kf10',
  35. _curses.KEY_F11: 'kf11',
  36. _curses.KEY_F12: 'kf12',
  37. _curses.KEY_F13: 'kf13',
  38. _curses.KEY_F14: 'kf14',
  39. _curses.KEY_F15: 'kf15',
  40. _curses.KEY_F16: 'kf16',
  41. _curses.KEY_F17: 'kf17',
  42. _curses.KEY_F18: 'kf18',
  43. _curses.KEY_F19: 'kf19',
  44. _curses.KEY_F2: 'kf2',
  45. _curses.KEY_F20: 'kf20',
  46. _curses.KEY_F21: 'kf21',
  47. _curses.KEY_F22: 'kf22',
  48. _curses.KEY_F23: 'kf23',
  49. _curses.KEY_F24: 'kf24',
  50. _curses.KEY_F25: 'kf25',
  51. _curses.KEY_F26: 'kf26',
  52. _curses.KEY_F27: 'kf27',
  53. _curses.KEY_F28: 'kf28',
  54. _curses.KEY_F29: 'kf29',
  55. _curses.KEY_F3: 'kf3',
  56. _curses.KEY_F30: 'kf30',
  57. _curses.KEY_F31: 'kf31',
  58. _curses.KEY_F32: 'kf32',
  59. _curses.KEY_F33: 'kf33',
  60. _curses.KEY_F34: 'kf34',
  61. _curses.KEY_F35: 'kf35',
  62. _curses.KEY_F36: 'kf36',
  63. _curses.KEY_F37: 'kf37',
  64. _curses.KEY_F38: 'kf38',
  65. _curses.KEY_F39: 'kf39',
  66. _curses.KEY_F4: 'kf4',
  67. _curses.KEY_F40: 'kf40',
  68. _curses.KEY_F41: 'kf41',
  69. _curses.KEY_F42: 'kf42',
  70. _curses.KEY_F43: 'kf43',
  71. _curses.KEY_F44: 'kf44',
  72. _curses.KEY_F45: 'kf45',
  73. _curses.KEY_F46: 'kf46',
  74. _curses.KEY_F47: 'kf47',
  75. _curses.KEY_F48: 'kf48',
  76. _curses.KEY_F49: 'kf49',
  77. _curses.KEY_F5: 'kf5',
  78. _curses.KEY_F50: 'kf50',
  79. _curses.KEY_F51: 'kf51',
  80. _curses.KEY_F52: 'kf52',
  81. _curses.KEY_F53: 'kf53',
  82. _curses.KEY_F54: 'kf54',
  83. _curses.KEY_F55: 'kf55',
  84. _curses.KEY_F56: 'kf56',
  85. _curses.KEY_F57: 'kf57',
  86. _curses.KEY_F58: 'kf58',
  87. _curses.KEY_F59: 'kf59',
  88. _curses.KEY_F6: 'kf6',
  89. _curses.KEY_F60: 'kf60',
  90. _curses.KEY_F61: 'kf61',
  91. _curses.KEY_F62: 'kf62',
  92. _curses.KEY_F63: 'kf63',
  93. _curses.KEY_F7: 'kf7',
  94. _curses.KEY_F8: 'kf8',
  95. _curses.KEY_F9: 'kf9',
  96. _curses.KEY_FIND: 'kfnd',
  97. _curses.KEY_HELP: 'khlp',
  98. _curses.KEY_HOME: 'khome',
  99. _curses.KEY_IC: 'kich1',
  100. _curses.KEY_IL: 'kil1',
  101. _curses.KEY_LEFT: 'kcub1',
  102. _curses.KEY_LL: 'kll',
  103. _curses.KEY_MARK: 'kmrk',
  104. _curses.KEY_MESSAGE: 'kmsg',
  105. _curses.KEY_MOVE: 'kmov',
  106. _curses.KEY_NEXT: 'knxt',
  107. _curses.KEY_NPAGE: 'knp',
  108. _curses.KEY_OPEN: 'kopn',
  109. _curses.KEY_OPTIONS: 'kopt',
  110. _curses.KEY_PPAGE: 'kpp',
  111. _curses.KEY_PREVIOUS: 'kprv',
  112. _curses.KEY_PRINT: 'kprt',
  113. _curses.KEY_REDO: 'krdo',
  114. _curses.KEY_REFERENCE: 'kref',
  115. _curses.KEY_REFRESH: 'krfr',
  116. _curses.KEY_REPLACE: 'krpl',
  117. _curses.KEY_RESTART: 'krst',
  118. _curses.KEY_RESUME: 'kres',
  119. _curses.KEY_RIGHT: 'kcuf1',
  120. _curses.KEY_SAVE: 'ksav',
  121. _curses.KEY_SBEG: 'kBEG',
  122. _curses.KEY_SCANCEL: 'kCAN',
  123. _curses.KEY_SCOMMAND: 'kCMD',
  124. _curses.KEY_SCOPY: 'kCPY',
  125. _curses.KEY_SCREATE: 'kCRT',
  126. _curses.KEY_SDC: 'kDC',
  127. _curses.KEY_SDL: 'kDL',
  128. _curses.KEY_SELECT: 'kslt',
  129. _curses.KEY_SEND: 'kEND',
  130. _curses.KEY_SEOL: 'kEOL',
  131. _curses.KEY_SEXIT: 'kEXT',
  132. _curses.KEY_SF: 'kind',
  133. _curses.KEY_SFIND: 'kFND',
  134. _curses.KEY_SHELP: 'kHLP',
  135. _curses.KEY_SHOME: 'kHOM',
  136. _curses.KEY_SIC: 'kIC',
  137. _curses.KEY_SLEFT: 'kLFT',
  138. _curses.KEY_SMESSAGE: 'kMSG',
  139. _curses.KEY_SMOVE: 'kMOV',
  140. _curses.KEY_SNEXT: 'kNXT',
  141. _curses.KEY_SOPTIONS: 'kOPT',
  142. _curses.KEY_SPREVIOUS: 'kPRV',
  143. _curses.KEY_SPRINT: 'kPRT',
  144. _curses.KEY_SR: 'kri',
  145. _curses.KEY_SREDO: 'kRDO',
  146. _curses.KEY_SREPLACE: 'kRPL',
  147. _curses.KEY_SRIGHT: 'kRIT',
  148. _curses.KEY_SRSUME: 'kRES',
  149. _curses.KEY_SSAVE: 'kSAV',
  150. _curses.KEY_SSUSPEND: 'kSPD',
  151. _curses.KEY_STAB: 'khts',
  152. _curses.KEY_SUNDO: 'kUND',
  153. _curses.KEY_SUSPEND: 'kspd',
  154. _curses.KEY_UNDO: 'kund',
  155. _curses.KEY_UP: 'kcuu1'
  156. }
  157. def has_key(ch):
  158. if isinstance(ch, str):
  159. ch = ord(ch)
  160. # Figure out the correct capability name for the keycode.
  161. capability_name = _capability_names.get(ch)
  162. if capability_name is None:
  163. return False
  164. #Check the current terminal description for that capability;
  165. #if present, return true, else return false.
  166. if _curses.tigetstr( capability_name ):
  167. return True
  168. else:
  169. return False
  170. if __name__ == '__main__':
  171. # Compare the output of this implementation and the ncurses has_key,
  172. # on platforms where has_key is already available
  173. try:
  174. L = []
  175. _curses.initscr()
  176. for key in _capability_names.keys():
  177. system = _curses.has_key(key)
  178. python = has_key(key)
  179. if system != python:
  180. L.append( 'Mismatch for key %s, system=%i, Python=%i'
  181. % (_curses.keyname( key ), system, python) )
  182. finally:
  183. _curses.endwin()
  184. for i in L: print(i)