Python中无需类定义添加属性实现自定义类型静态属性高亮

要让VSCode等IDE识别MyDict实例的动态属性（比如x.bar）并提供静态高亮、自动补全和错误提示，核心是给类型检查器传递足够的类型信息，无需显式声明属性。以下是几种可行方案：

方案一：泛型类绑定具体NamedTuple类型 #

借助TypeVar和Generic，让MyDict成为泛型类，明确它包装的是某个特定的NamedTuple子类，同时为__getattr__添加精准的类型注解：

from typing import NamedTuple, TypeVar, Generic, cast

# 定义绑定到NamedTuple子类的类型变量
T = TypeVar("T", bound=NamedTuple)

class Foo(NamedTuple):
    bar: int
    baz: str

class MyDict(Generic[T]):
    def __init__(self, n: T) -> None:
        self._n = n

    def __getattr__(self, key: str) -> T.__getitem__(key):
        return cast(T.__getitem__(key), getattr(self._n, key))

foo = Foo(3, "test")
x = MyDict(foo)
y = x.bar  # 此时VSCode会高亮bar，提供自动补全，错误提示正常
z = x.baz  # 同样支持属性识别

方案二：简化泛型+Any类型注解 #

如果不需要极致的类型精准度，用Any作为__getattr__的返回类型，配合泛型绑定也能满足基本的高亮和补全需求：

from typing import NamedTuple, TypeVar, Generic, Any

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

class Foo(NamedTuple):
    bar: int

class MyDict(Generic[T]):
    def __init__(self, n: T) -> None:
        self._n = n

    def __getattr__(self, key: str) -> Any:
        return getattr(self._n, key)

foo = Foo(3)
x = MyDict(foo)
y = x.bar  # 支持静态高亮与自动补全

方案三：Protocol适配更通用场景 #

如果你的包装类不止处理NamedTuple，可以用Protocol定义动态属性的协议，让MyDict实现该协议：

from typing import NamedTuple, Protocol, TypeVar, Generic, Any

class DynamicAttrs(Protocol):
    def __getattr__(self, key: str) -> Any: ...

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

class Foo(NamedTuple):
    bar: int

class MyDict(Generic[T]):
    def __init__(self, n: T) -> None:
        self._n = n

    def __getattr__(self, key: str) -> Any:
        return getattr(self._n, key)

foo = Foo(3)
x = MyDict(foo)
y = x.bar  # 正常识别属性

核心原理 #

VSCode的Python插件（基于Pylance/Mypy）依赖静态类型推断提供代码辅助功能。通过泛型绑定具体类型、给__getattr__添加精确注解，能让类型检查器推断出MyDict实例的动态属性类型，从而实现高亮、补全和错误提示。

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