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