116

我知道如何在 python 解释器(在 unix 上)中设置 python 对象的自动完成。

  • 谷歌显示了许多关于如何做到这一点的解释。
  • 不幸的是,参考资料太多,很难找到我需要做的事情,这略有不同。

我需要知道如何在用 python 编写的命令行程序中启用、制表符/自动完成任意项。

我的具体用例是一个需要发送电子邮件的命令行 python 程序。我希望能够在用户键入部分电子邮件地址(并且可以选择按 TAB 键)时自动完成电子邮件地址(我在磁盘上有地址)。

我不需要它在 windows 或 mac 上工作,只需要在 linux 上工作。

4

9 回答 9

77

使用 Python 的readline绑定。例如,

import readline

def completer(text, state):
    options = [i for i in commands if i.startswith(text)]
    if state < len(options):
        return options[state]
    else:
        return None

readline.parse_and_bind("tab: complete")
readline.set_completer(completer)

官方模块文档并没有更详细,请参阅readline 文档了解更多信息。

于 2008-10-09T15:01:38.597 回答
64

按照cmd 文档,你会没事的

import cmd

addresses = [
    'here@blubb.com',
    'foo@bar.com',
    'whatever@wherever.org',
]

class MyCmd(cmd.Cmd):
    def do_send(self, line):
        pass

    def complete_send(self, text, line, start_index, end_index):
        if text:
            return [
                address for address in addresses
                if address.startswith(text)
            ]
        else:
            return addresses


if __name__ == '__main__':
    my_cmd = MyCmd()
    my_cmd.cmdloop()

tab -> tab -> send -> tab -> tab -> f -> tab 的输出

(Cmd)
help  send
(Cmd) send
foo@bar.com            here@blubb.com         whatever@wherever.org
(Cmd) send foo@bar.com
(Cmd)
于 2008-10-09T15:08:18.663 回答
45

既然你在你的问题中说“不是解释器”,我想你不想要涉及 python readline 等的答案。(编辑:事后看来,显然不是这样。哼哼。无论如何,我认为此信息很有趣,所以我将其留在这里。

我想你可能会在之后。

它是关于向任意命令添加 shell 级补全,扩展 bash 自己的制表符补全。

简而言之,您将创建一个包含 shell 函数的文件,该函数将生成可能的完成,将其保存到/etc/bash_completion.d/并使用 command 进行注册complete。这是链接页面的一个片段:

_foo() 
{
    local cur prev opts
    COMPREPLY=()
    cur="${COMP_WORDS[COMP_CWORD]}"
    prev="${COMP_WORDS[COMP_CWORD-1]}"
    opts="--help --verbose --version"

    if [[ ${cur} == -* ]] ; then
        COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
        return 0
    fi
}
complete -F _foo foo

在这种情况下,输入foo --[TAB]将为您提供变量 中的值,opts即和。出于您的目的,您基本上需要自定义放入.--help--verbose--versionopts

请查看链接页面上的示例,这一切都非常简单。

于 2008-10-13T09:59:23.950 回答
34

我很惊讶没有人提到 argcomplete,这是文档中的一个示例:

from argcomplete.completers import ChoicesCompleter

parser.add_argument("--protocol", choices=('http', 'https', 'ssh', 'rsync', 'wss'))
parser.add_argument("--proto").completer=ChoicesCompleter(('http', 'https', 'ssh', 'rsync', 'wss'))
于 2014-05-30T16:59:54.013 回答
15

这是由 ehemient here提供的代码的完整版本(谢谢)。

import readline

addrs = ['angela@domain.com', 'michael@domain.com', 'david@test.com']

def completer(text, state):
    options = [x for x in addrs if x.startswith(text)]
    try:
        return options[state]
    except IndexError:
        return None

readline.set_completer(completer)
readline.parse_and_bind("tab: complete")

while 1:
    a = raw_input("> ")
    print "You entered", a
于 2008-10-16T19:26:31.450 回答
13
# ~/.pythonrc
import rlcompleter, readline
readline.parse_and_bind('tab:complete')

# ~/.bashrc
export PYTHONSTARTUP=~/.pythonrc
于 2013-10-24T00:48:58.013 回答
11

您可以尝试使用Python Prompt Toolkit,这是一个用于在 Python 中构建交互式命令行应用程序的库。

该库可以轻松添加交互式自动完成功能,允许用户使用该Tab键以可视方式循环浏览可用选项。该库是跨平台的(Linux、OS X、FreeBSD、OpenBSD、Windows)。例子:

pgcli - Python 提示工具包

(图片来源:pcgli

于 2019-05-29T15:10:57.097 回答
1

发布的答案工作正常,但我已经开源了一个我在工作中编写的自动完成库。我们已经在生产中使用了一段时间,它快速、稳定且易于使用。它甚至具有演示模式,因此您可以在键入单词时快速测试您会得到什么。

要安装它,只需运行:pip install fast-autocomplete

这是一个例子:

>>> from fast_autocomplete import AutoComplete
>>> words = {'book': {}, 'burrito': {}, 'pizza': {}, 'pasta':{}}
>>> autocomplete = AutoComplete(words=words)
>>> autocomplete.search(word='b', max_cost=3, size=3)
[['book'], ['burrito']]
>>> autocomplete.search(word='bu', max_cost=3, size=3)
[['burrito']]
>>> autocomplete.search(word='barrito', max_cost=3, size=3)  # mis-spelling
[['burrito']]

结帐:https ://github.com/seperman/fast-autocomplete获取源代码。

这里是它如何工作的解释:http: //zepworks.com/posts/you-autocomplete-me/

它处理拼写错误并可选择按单词的权重排序。(假设burrito比 更重要book,那么您给出burrito更高的“计数”,它将首先出现book在结果中。

Words 是一本字典,每个单词都可以有一个上下文。例如“计数”、如何显示单词、单词周围的一些其他上下文等。在这个例子中,单词没有任何上下文。

于 2019-03-29T22:36:14.160 回答
1

这很好用。

#!/usr/bin/python3

import readline
readline.parse_and_bind("tab: complete")

def complete(text,state):
    volcab = ['dog','cat','rabbit','bird','slug','snail']
    results = [x for x in volcab if x.startswith(text)] + [None]
    return results[state]

readline.set_completer(complete)

line = input('prompt> ')
于 2021-07-18T03:11:16.657 回答