如何让VS Code中Python的types.MethodType支持跳转到定义？

因为types.MethodType是运行时动态绑定方法到实例，VS Code的静态代码分析工具（比如Pylance）没法在编译期识别这个绑定关系，所以会提示找不到定义。可以用以下几种方式解决：

方案1：给实例或类添加类型注解 #

直接在绑定方法时给a.foo加上类型提示，或者在类里提前声明foo的类型，让静态分析工具识别到这个方法的存在：

import types
from typing import Callable


class A:
    def __init__(self) -> None:
        self.a = "a"
    # 提前在类里声明foo的类型，告诉分析工具实例会有这个方法
    foo: Callable[[], None]


def _foo(obj: A):  # 给_foo的参数指定类型，增强关联
    print(obj.a)


if __name__ == "__main__":
    a = A()
    a.foo = types.MethodType(_foo, a)
    a.foo()

配置后，Ctrl+点击a.foo()就能直接跳转到_foo的定义。

方案2：使用Protocol定义实例结构 #

如果不想修改原类，用Protocol定义一个包含foo方法的协议，然后把实例指定为这个协议类型：

import types
from typing import Protocol, Callable


class HasFoo(Protocol):
    a: str
    foo: Callable[[], None]


class A:
    def __init__(self) -> None:
        self.a = "a"


def _foo(obj: A):
    print(obj.a)


if __name__ == "__main__":
    a: HasFoo = A()  # 指定实例类型为HasFoo协议
    a.foo = types.MethodType(_foo, a)
    a.foo()

这种方式更灵活，不需要改动原类代码，同样能让静态分析工具识别到foo方法的关联。

方案3：直接替换为类方法（如果场景允许） #

如果不需要仅绑定到单个实例，而是所有实例都能用这个方法，直接把_foo定义为类的实例方法，天然支持跳转：

class A:
    def __init__(self) -> None:
        self.a = "a"
    
    def foo(self):
        print(self.a)


if __name__ == "__main__":
    a = A()
    a.foo()

这个方案只适用于不需要动态绑定到单个实例的场景。

内容的提问来源于stack exchange，提问作者J. Lo