如何在Python类方法中实现参数值的自动补全功能？

要给Friends类的get_friend_last_name方法添加参数自动补全（避免用__getitem__导致的调用混淆），可以利用IPython的自定义补全机制实现，具体方案如下：

完整实现代码 #

class Friends:
    def __init__(self):
        self._friends_last_names = {
            "Rachel": "Green",
            "Ross": "Geller",
            "Chandler": "Bing"
        }

    def get_friend_last_name(self, first_name):
        last_name = self._friends_last_names.get(first_name, None)
        if last_name is None:
            print(f"Error, {first_name} not found")
            return
        return last_name

# 配置IPython自动补全
try:
    from IPython.core.interactiveshell import InteractiveShell

    def friend_name_completer(shell, obj, prev_completions):
        # 仅针对Friends实例的get_friend_last_name方法触发补全
        if isinstance(obj, Friends) and shell._last_expr.endswith('.get_friend_last_name('):
            # 提取用户已输入的前缀
            line = shell._last_parsed.line
            if 'first_name="' in line:
                prefix = line.split('first_name="')[1].rstrip('"')
                return [name for name in obj._friends_last_names if name.startswith(prefix)]
        return prev_completions

    # 注册补全函数
    shell = InteractiveShell.instance()
    shell.completer.custom_completers.append(friend_name_completer)
except ImportError:
    print("自动补全功能仅支持IPython环境")

使用效果 #

在IPython中运行代码后：

• 输入friends_obj.get_friend_last_name(first_name="Ch并按TAB，会自动补全为"Chandler"
• 输入friends_obj.get_friend_last_name(first_name="R并按TAB，会显示可选的"Rachel"和"Ross"

说明 #

通过IPython的custom_completers注册自定义补全逻辑，仅当调用Friends实例的get_friend_last_name方法时，才从实例的名字集合中匹配前缀生成补全选项，既实现了易用性，又不会混淆类的其他方法调用方式。

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