1

我正在尝试对主列表的子列表进行选项卡补全，例如

commands = ['help', 'set', 'info']
sub_command = ['module', 'level'] 
modules = ['pr', 'tls', 'tld']
levels = ['high', 'mid', 'low']

任何想法如何做这样的事情：

>>> se<tab> mo<tab> tl<tab>

这是我的主列表代码：

from core.libs.interpreter import interpreter

import re
try:
    import readline
except ImportError:
    print '\n[!] The "readline" module is required to provide elaborate line editing and history features'
else:
    pass

COMMANDS = interpreter.commands

RE_SPACE = re.compile('.*\s+$', re.M)


class Completer(object):
    '''
    internal readline buffer to determine the state of the overall completion,
    which makes the state logic a bit simpler
        '''
    def complete(self, text, state):
        "Generic readline completion entry point."
        buffer = readline.get_line_buffer()
        line = readline.get_line_buffer().split()
        # show all commands
        if not line:
            return [c + ' ' for c in COMMANDS][state]
        # account for last argument ending in a space
        if RE_SPACE.match(buffer):
            line.append('')
        # resolve command to the implementation function
        cmd = line[0].strip()
        if cmd in COMMANDS:
            #impl = getattr(self, 'complete_%s' % cmd)
            args = line[1:]
            if args:
                return (args + [None])[state]
            return [cmd + ' '][state]
        results = [c + ' ' for c in COMMANDS if c.startswith(cmd)] + [None]
        return results[state]

    def tab(self):
        # to work with non nix systems
        try:
            readline.set_completer_delims(' \t\n;')
            readline.parse_and_bind("tab: complete")
            readline.set_completer(self.complete)
        except:
            pass

complete = Completer()

我试图按照本教程进行操作，但没有运气。

提前致谢。

4

0 回答 0