Python命令行界面中的选项卡完成 - 如何捕获Tab事件

时间:2010-01-12 01:11:58

标签: python mercurial tab-completion raw-input

我在Python中编写了一个CLI(作为Mercurial的扩展),并希望支持制表完成。具体来说,我想在提示中捕获标签并显示匹配选项列表(就像bash一样)。

示例:输入部分名称:

 ext*TAB*  
 extensions  
 extras

问题是我不确定如何捕捉Tab事件。我正在使用Mercurial的ui.prompt() API,它只是在引擎盖下调用raw_input()

据我所知,raw_input()仅返回'enter',如果用户输入标签,则返回的字符串只包含"\t"

3 个答案:

答案 0 :(得分:15)

为此您使用readline模块。

我能想到的最简单的代码:

import readline
COMMANDS = ['extra', 'extension', 'stuff', 'errors',
            'email', 'foobar', 'foo']

def complete(text, state):
    for cmd in COMMANDS:
        if cmd.startswith(text):
            if not state:
                return cmd
            else:
                state -= 1

readline.parse_and_bind("tab: complete")
readline.set_completer(complete)
raw_input('Enter section name: ')

使用示例:

Enter section name: <tab>
email      errors     extension  extra      foo        foobar    stuff
Enter section name: e<tab>
email      errors     extension  extra      
Enter section name: ext<tab>
extension  extra      

除完成外,readline为您提供:

  • 线路编辑
  • 键绑定配置(包括emacs和vi模式)
  • 历史记录(向上箭头以回忆以前的值)
  • 历史搜索,保存和加载

答案 1 :(得分:2)

标准库中作为readline模块提供了如何与rlcompleter合作完成制表完成的一个很好的示例 - 您不能按原样使用它(它完成基于当前在Python的main和builtin中定义的名称,但它显示了如何执行常规任务以及如何将其连接到readline

答案 2 :(得分:1)

你几乎肯定会使用cmd module,它已经实现了标签完成等等,可能还有你正在尝试做的其他部分,使用readline模块等等。重新发明轮子毫无意义。