如何为类添加函数并确保Intellisense能识别这些动态添加的函数？

问题根源 #

你当前的装饰器是在类实例化（调用MyClass()）时才给类对象添加someFunc方法，但Intellisense这类静态分析工具是在代码编写阶段做类型检查，此时类还未被实例化，自然捕获不到动态添加的方法。而@dataclass是在装饰类的当下就直接修改类的结构，所以静态分析能识别到它添加的成员。

直接解决方案（与dataclass逻辑对齐） #

把添加方法的逻辑从实例化包装函数中移到装饰器顶层，直接在装饰阶段修改类对象，让静态分析工具能直接读取到类的结构变化：

import functools

def someFunc(self):
    print("YAY")

def mydec(cls):
    # 装饰类时直接给类添加方法
    setattr(cls, "someFunc", someFunc)
    # 保留类的元信息，辅助静态分析识别
    functools.wraps(cls)(cls)
    return cls

@mydec
class MyClass:
    ...

myInstance = MyClass()
myInstance.someFunc()  # Intellisense可正常识别

延迟添加场景的妥协方案 #

如果必须在实例化阶段才添加方法（比如依赖实例化参数），可以通过类型协议（Protocol）给静态分析工具提供类型提示：

import functools
from typing import Protocol, runtime_checkable

@runtime_checkable
class HasSomeFunc(Protocol):
    def someFunc(self) -> None: ...

def someFunc(self):
    print("YAY")

def mydec(func):
    @functools.wraps(func)
    def dec_inner(*args,**kwargs):
        setattr(func,"someFunc", someFunc)
        re = func(*args,**kwargs)
        return re
    
    return dec_inner

@mydec
class MyClass:
    ...

# 显式标注实例类型，让Intellisense识别方法
myInstance: HasSomeFunc = MyClass()
myInstance.someFunc()  # Intellisense可正常识别

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