如何在prompt_toolkit

时间:2017-10-02 15:16:39

标签: python autocomplete prompt-toolkit

我正在为我的项目创建一个REPL工具(简化为了清晰)直接执行输入的命令或(如果输入命令" .x some / path / to / file")读取和执行他们来自档案。我的问题与自动完成用户输入有关(使用prompt_toolkit)。

我有类似的(最小可执行示例):

import prompt_toolkit
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.document import Document
from prompt_toolkit.contrib.completers import PathCompleter


class CommandCompleter(Completer):
    def __init__(self):
        self.path_completer = PathCompleter()
        self.commands = [".x", "command1", "command2"]

    def get_completions(self, document, complete_event):
        if document.text.startswith(".x "):
            sub_doc = Document(document.text[3:])
            yield from (Completion(cmd.text, -document.cursor_position)
#                                  ????????  ?????????????????????????
                        for cmd
                        in self.path_completer.get_completions(sub_doc, complete_event))
#                                                              ???????
        else:
            yield from (Completion(cmd, -document.cursor_position)
                        for cmd in self.commands
                        if cmd.startswith(document.text))

if __name__ == "__main__":
    while True:
        other_args = {}
        input = prompt_toolkit.prompt(">>> ", completer=CommandCompleter(), **other_args)
        # Do something with input (omitted)

第二个if-branch(用于命令)工作正常,但我不知道如何正确调用PathCompleter.get_completions()方法并从其结果重建Completion个对象(其中?? ?在第一个分支中。诀窍在于我只对输入的一部分使用完成,并且各种子串,位置计算等(尚未)导致令人满意的行为(即提供路径并构造正确的输入线)。 / p>

我肯定会继续搜索,但如果有人知道如何重写,那将非常有用。

注意:如果整个输入只是路径(并且这可以正常工作),则会使用yield from self.path_completer.get_completions(document, complete_event)

1 个答案:

答案 0 :(得分:1)

可能以下内容应该解决它:

sub_doc = Document(document.text[3:])
yield from (Completion(completion.text, completion.start_position, display=completion.display)
            for completion
            in self.path_completer.get_completions(sub_doc, complete_event))
  • completion.text包含要插入的文字;
  • completion.start_position包含将相对于光标位置插入文本的位置(在此特定示例中,我们可以从嵌套的完成符中获取值)。
  • completion.display是弹出菜单中显示的值。 (在这种情况下,整个文件名,而不仅仅是插入的字符串。

如果您有任何其他问题,请随意打开GitHub问题。