如何为泛型对象新增属性添加类型提示？Python类型标注问题

可行的类型标注方案 #

使用Protocol定义包含新增属性的协议，结合Python 3.10+支持的交集类型（&），可以同时保留原对象和新增属性的代码提示：

from typing import TypeVar, Protocol, List

T = TypeVar("T")

class WithAnnotationProtocol(Protocol):
    some_attribute: str

def annotate_attribute[T](objs: List[T]) -> List[T & WithAnnotationProtocol]:
    for obj in objs:
        obj.some_attribute = "something"
    return objs

这个方案的核心逻辑：

• WithAnnotationProtocol仅定义新增属性的类型约束，无需实际继承或实例化
• 交集类型T & WithAnnotationProtocol告知类型检查器：返回对象同时具备原类型T的所有属性，以及协议中定义的some_attribute
• PyCharm会正确识别并同时提供原对象和新增属性的代码提示

原方案无效的原因 #

你之前的代码中，WithAnnotation[T]继承自type[T]（元类），这会让类型检查器误判返回值为元类实例，而非原对象的扩展。这种写法混淆了类型本身和类型的实例，直接导致原对象的代码提示丢失。

无需“魔法”标注的替代方案 #

如果不想依赖高级类型特性，也可以通过包装类或子类的方式实现，同时保留完整类型提示：

方案1：属性代理包装类 #

创建包装类代理原对象的所有属性，同时添加新增属性：

from typing import TypeVar, Generic, List

T = TypeVar("T")

class AnnotatedWrapper(Generic[T]):
    def __init__(self, original: T):
        self._original = original
        self.some_attribute = "something"

    # 代理原对象的属性与方法访问
    def __getattr__(self, name):
        return getattr(self._original, name)

def annotate_attribute[T](objs: List[T]) -> List[AnnotatedWrapper[T]]:
    return [AnnotatedWrapper(obj) for obj in objs]

使用时，既可以直接访问wrapper.some_attribute，也能直接调用原对象的方法/属性（比如wrapper.original_method()）。

方案2：创建子类（适用于可继承的原类型） #

如果原类型是自定义类且支持继承，可直接创建子类并添加属性：

from typing import TypeVar, List

T = TypeVar("T", bound="BaseClass")

class BaseClass:
    def original_method(self):
        pass

class AnnotatedClass(BaseClass):
    some_attribute: str = "something"

def annotate_attribute[T](objs: List[T]) -> List[AnnotatedClass]:
    # 注意：需确保原对象可转换为子类实例，此处仅为示例
    return [AnnotatedClass(**obj.__dict__) for obj in objs]

这种方案更直观，但仅适用于你能控制原类型、或原类型可被继承/转换的场景。

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