blob: 89920d4ac6e8057bd0891964742cf01e431e44ac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
import readline
import sys
class Completer(object):
@classmethod
def install_new_completer(cls, options):
completer = cls(options)
readline.set_completer(completer.complete)
readline.set_completer_delims('') # allow options with whitespace
readline.parse_and_bind('tab: complete')
def __encode_completion_string(self, s):
try:
return s.encode(sys.stdin.encoding) if isinstance(s, unicode) else s
except NameError: # Python 3.x does not have a 'unicode' type
pass
return s
def __init__(self, options):
self.options = [self.__encode_completion_string(s) for s in options]
def complete(self, text, state):
text = text.lower()
if state == 0: # on first trigger, build possible matches
if text: # cache matches (entries that start with entered text)
self.matches = [s for s in self.options
if s and s.lower().startswith(text)]
else: # no text entered, all matches possible
self.matches = self.options[:]
# return match indexed by state
try:
return self.matches[state]
except IndexError:
return None
|
