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

Question

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

示例:输入部分名称:

 ext*TAB*  
 extensions  
 extras

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

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

Answer 1

为此,您使用该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模式)
• 历史(向上箭头以回忆以前的值)
• 历史搜索,保存和加载