如何为装饰器注入的成员函数设置正确的类型提示

VSCode的Pylance类型检查器无法自动识别动态注入的方法，核心原因是它依赖静态代码分析和函数的__annotations__属性来推断类型。以下是几种可行的解决方法：

方案1：为注入方法手动添加类型注解 #

直接给动态生成的SET_*和GET_*方法添加__annotations__属性，让Pylance能读取到明确的类型信息：

from dataclasses import dataclass, fields
from typing import TypeVar

TimeType = int
T = TypeVar('T')

def add_field_methods(cls):
    for field in fields(cls):
        field_name = field.name
        field_type = field.type

        # 定义并配置SET方法
        def set_method(self, value: T, time: TimeType) -> None:
            setattr(self, field_name, value)
            setattr(self, f"_last_updated_{field_name}", time)
        
        # 绑定当前字段的类型到方法注解
        set_method.__annotations__["value"] = field_type
        setattr(cls, f"SET_{field_name}", set_method)

        # 定义并配置GET方法
        def get_method(self) -> TimeType:
            return getattr(self, f"_last_updated_{field_name}", 0)
        
        setattr(cls, f"GET_{field_name}", get_method)
        # 初始化更新时间字段
        setattr(cls, f"_last_updated_{field_name}", 0)
    return cls

@add_field_methods
@dataclass
class MyData:
    a: int
    b: str

方案2：用Protocol补充静态类型提示 #

通过typing.Protocol定义注入方法的签名，让类型检查器认为dataclass实现了该协议：

from dataclasses import dataclass, fields
from typing import Protocol, TYPE_CHECKING, TypeVar

TimeType = int
T = TypeVar('T')

# 定义字段访问的协议模板
class FieldAccessProtocol(Protocol, Generic[T]):
    def SET_a(self, value: T, time: TimeType) -> None: ...
    def GET_a(self) -> TimeType: ...
    def SET_b(self, value: T, time: TimeType) -> None: ...
    def GET_b(self) -> TimeType: ...

def add_field_methods(cls):
    # （这里保留你原有的方法注入逻辑）
    for field in fields(cls):
        field_name = field.name
        def set_method(self, value, time):
            setattr(self, field_name, value)
            setattr(self, f"_last_updated_{field_name}", time)
        
        def get_method(self):
            return getattr(self, f"_last_updated_{field_name}", 0)
        
        setattr(cls, f"SET_{field_name}", set_method)
        setattr(cls, f"GET_{field_name}", get_method)
        setattr(cls, f"_last_updated_{field_name}", 0)
    return cls

@add_field_methods
@dataclass
class MyData:
    a: int
    b: str

# 仅在类型检查时绑定协议
if TYPE_CHECKING:
    MyData = FieldAccessProtocol[int] & FieldAccessProtocol[str]

方案3：动态生成带完整类型的类 #

直接生成包含所有方法和类型注解的新类，替代动态修改原类的方式，类型检查器能直接识别类结构：

from dataclasses import make_dataclass
from typing import TypeVar

TimeType = int
T = TypeVar('T')

def make_timed_dataclass(cls_name, field_defs):
    # 构建基础字段和更新时间字段
    base_fields = [(name, type_) for name, type_ in field_defs]
    updated_fields = [(f"_last_updated_{name}", TimeType, 0) for name, _ in field_defs]
    base_cls = make_dataclass(cls_name, base_fields + updated_fields)

    # 为每个字段添加带类型注解的方法
    for name, type_ in field_defs:
        def set_method(self, value: type_, time: TimeType) -> None:
            setattr(self, name, value)
            setattr(self, f"_last_updated_{name}", time)
        
        def get_method(self) -> TimeType:
            return getattr(self, f"_last_updated_{name}")
        
        setattr(base_cls, f"SET_{name}", set_method)
        setattr(base_cls, f"GET_{name}", get_method)
    
    return base_cls

# 使用示例
MyData = make_timed_dataclass("MyData", [("a", int), ("b", str)])

内容的提问来源于stack exchange，提问作者G. C.