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