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) |
---|