使用prompt_toolkit.completionCompleter()实现命令行中文自动补全

下面是一个使用 prompt_toolkit 中的 completionCompleter 实现命令行中文自动补全的示例代码。

from prompt_toolkit import prompt
from prompt_toolkit.completion import Completion, Completer

class ChineseCompleter(Completer):
    def __init__(self, words):
        self.words = words

    def get_completions(self, document, complete_event):
        word_before_cursor = document.get_word_before_cursor()
        matches = [word for word in self.words if word.startswith(word_before_cursor)]
        
        for match in matches:
            yield Completion(match, start_position=-len(word_before_cursor))

# 示例用于测试的词汇列表
words = ['你好', '你好吗', '我很好', '谢谢', '再见']

completer = ChineseCompleter(words)

def main():
    while True:
        user_input = prompt('请输入：', completer=completer)
        print(f'您输入了：{user_input}')

if __name__ == '__main__':
    main()

在上面的例子中，我们首先实现了一个名为 ChineseCompleter 的自动补全器，它继承自 Completer 类，并重写了其中的 get_completions 方法。在这个方法中，我们首先获取光标前的单词 word_before_cursor，然后找出与之匹配的词汇。最后，我们使用 yield 将匹配到的词汇作为 Completion 对象返回，其中 start_position 参数用于指定光标的起始位置。

然后，我们创建了一个示例用于测试的词汇列表 words，并使用这个词汇列表创建了一个 ChineseCompleter 对象 completer。

最后，我们在 main 函数中使用 prompt 函数获取用户的输入，并将 completer 作为参数传递给它，从而实现了中文自动补全。同时，我们打印了用户输入的内容。运行这段代码后，当用户输入一些中文字符时，可以通过按下 Tab 键来触发自动补全功能。

请注意，由于 prompt_toolkit 默认使用 Readline 库来处理输入，而中文字符的输入可能会导致一些显示上的问题。如果遇到这种情况，可以尝试使用 prompt_toolkit 提供的其他输入处理库。