Python Prompt-Toolkit多重补全如何避免冗余嵌套字典实现动态补全

完全不需要预定义全量多层嵌套字典，核心方案是自定义实现Completer类，根据用户实时输入的上下文动态计算返回补全项，输入走到哪个分支才加载对应分支的补全逻辑，从根源上避免枚举所有输入组合的问题。

核心实现逻辑 #

Prompt-Toolkit 内置的Completer基类允许自定义补全规则，只要实现get_completions方法即可：该方法会在用户每次输入触发补全时实时执行，能拿到当前输入的完整文本、光标位置等上下文信息，解析完用户已经输入的命令片段后，仅返回当前输入位置合法的补全项即可。

基础实现示例代码：

from prompt_toolkit.completion import Completer, Completion

class DynamicCliCompleter(Completer):
    def get_completions(self, document, complete_event):
        # 拆分已输入的命令token，自动忽略多余空白字符
        input_tokens = document.text.strip().split()
        # 获取光标前的待补全文本，用于前缀匹配
        prefix = document.get_word_before_cursor()

        # 顶层命令补全：无输入时返回一级命令
        if len(input_tokens) == 0:
            for cmd in ["show", "exit"]:
                if cmd.startswith(prefix):
                    yield Completion(cmd, start_position=-len(prefix))
            return

        # 处理show命令分支
        if input_tokens[0] == "show":
            # show后第一层选项补全
            if len(input_tokens) == 1:
                for opt in ["-i", "-s"]:
                    if opt.startswith(prefix):
                        yield Completion(opt, start_position=-len(prefix))
                return
            # 处理-i选项分支
            if input_tokens[1] == "-i":
                if len(input_tokens) == 2:
                    # 此处支持动态取值：可从配置、接口、系统实时查询可选项，无需硬编码
                    for val in ["a", "b", "c", "clock"]:
                        if val.startswith(prefix):
                            yield Completion(val, start_position=-len(prefix))
                    return
                # 仅当-i后输入a时，才加载后续-s选项的补全逻辑
                if input_tokens[2] == "a":
                    if len(input_tokens) == 3:
                        for opt in ["-s"]:
                            if opt.startswith(prefix):
                                yield Completion(opt, start_position=-len(prefix))
                        return
                    # 处理-s选项的后续补全
                    if input_tokens[3] == "-s":
                        for val in ["interface"]:
                            if val.startswith(prefix):
                                yield Completion(val, start_position=-len(prefix))
                        return
            # 处理-s选项分支，逻辑按需扩展即可
            if input_tokens[1] == "-s":
                if len(input_tokens) == 2:
                    for val in ["interface", "other"]:
                        if val.startswith(prefix):
                            yield Completion(val, start_position=-len(prefix))
                    return

        # exit命令无后续补全，直接返回
        if input_tokens[0] == "exit":
            return

优化建议 #

• 分支逻辑拆分：命令层级较多时，可将不同命令、不同参数的补全逻辑拆分为独立函数，通过token路径路由到对应处理函数，避免单函数代码臃肿
• 混合使用静态补全：对于补全项固定的简单分支，可局部实例化小的NestedCompleter处理，无需全量手写匹配逻辑，灵活组合即可
• 动态值支持：涉及文件名、系统资源、实时配置等动态补全场景，可在对应分支的处理逻辑中实时查询返回，完全不受静态字典的限制

方案优势 #

• 无需提前枚举所有输入组合构造嵌套字典，代码体量更小，维护时仅需修改对应分支的逻辑，不会出现嵌套字典语法错误、漏写分支等问题
• 补全逻辑按需加载，启动速度和运行效率远高于全量定义的巨型嵌套字典
• 支持动态生成补全项，可根据用户权限、当前系统状态、实时查询结果返回差异化补全内容，适配复杂命令行场景

内容的提问来源于stack exchange，提问作者yankeefan11