Python命令行界面中的制表符完成-如何捕获制表符事件

问题描述

我正在用Python编写一个小CLI(作为Mercurial的扩展)，并希望支持制表符完成。具体地说，我希望在提示符中显示catch选项卡，并显示匹配选项的列表(就像bash一样)。

示例：输入节名：

 ext*TAB*  
 extensions  
 extras

问题是我不确定如何捕捉Tab事件。我使用的是Mercurial的ui.prompt()API，它只是在幕后调用raw_input()。

据我所知，raw_input()只在‘Enter’时返回，如果用户输入制表符，则返回的字符串只包含一个" "。

---

解决方案

为此，您可以使用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模式)
• 历史记录(向上箭头可调用以前的值)
• 历史记录搜索、保存和加载