VS Code中Python的__sub__魔法方法无法自动补全如何解决？

问题描述 #

在VS Code中输入add时，自动补全会给出__add__选项，多数魔法方法（dunder-methods）都支持该补全功能，但输入sub时无法触发__sub__的自动补全，该如何修复？是否有相关插件可用？

用户提供的代码示例：

class ListMath:
    def __init__ (self, ls=None):
        self.lst_math = [] if not ls else [x for x in ls if type(x) in (int, float)]

    def __add__ (self, other):
        if type(other) in (int, float):
            return ListMath([x + other for x in self.lst_math])

    def __radd__ (self, other):
        return self + other

    def __iadd__ (self, other):
        if type(other) in (int, float):
            self.lst_math = [x + other for x in self.lst_math]
            return self

    def __sub__ (self, other):
        if type(other) in (int, float):
            return ListMath([x - other for x in self.lst_math])

解决办法 #

• 确认并启用Pylance语言服务：VS Code默认依赖Pylance提供Python代码补全能力。打开设置（快捷键Ctrl+,），搜索Python: Language Server，确保选项设置为Pylance。
• 添加类型注解增强识别：给__sub__方法的参数和返回值补充类型注解，帮助语言服务更准确识别方法。修改后的__sub__方法示例：

def __sub__(self, other: int | float) -> "ListMath":
    if isinstance(other, (int, float)):
        return ListMath([x - other for x in self.lst_math])

注：使用isinstance替代type()判断类型是Python更规范的写法，同时类型注解能提升语言服务的分析精度

• 重启Python语言服务：按Ctrl+Shift+P打开命令面板，输入Python: Restart Language Server执行重启操作，之后再尝试输入sub触发补全。
• 更新核心插件：确保VS Code中的Python（微软官方）和Pylance插件为最新版本。打开扩展面板（Ctrl+Shift+X），找到对应插件检查更新并安装。

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