我正在为我的项目创建一个 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 (用于命令)正常工作,但我不知道如何正确调用该方法并从第一个分支中的结果(???'s 所在的位置)PathCompleter.get_completions()
重建对象。Completion
诀窍是我只对输入的一部分使用完成,各种子串、位置计算等(还)没有导致令人满意的行为(即提供路径和构建正确的输入行)。
我肯定会继续搜索,但如果有人知道如何重写它,那将非常有用。
注意:yield from self.path_completer.get_completions(document, complete_event)
如果整个输入只是路径(并且可以正常工作),则将使用。