如何让编辑器对make_dataclass生成的运行时数据类提供类型提示？

用make_dataclass在运行时生成的数据类，由于Pyright是静态分析工具，无法追踪动态生成的类结构，所以不会触发自动补全。可以通过以下几种方式解决：

方式一：定义匹配结构的Protocol并添加类型注解 #

创建一个和动态类字段完全匹配的Protocol，给生成的类变量加上类型注解，让Pyright识别其结构：

from dataclasses import make_dataclass
from typing import Protocol

class PersonProtocol(Protocol):
    name: str
    age: int

# 给Person变量标注类型，让静态分析工具识别字段
Person: type[PersonProtocol] = make_dataclass('Person', [('name', str), ('age', int)])

方式二：使用typing.cast强制类型转换 #

如果不想定义Protocol，可以先写一个仅用于静态分析的类，再用cast把动态生成的类转换成该类型：

from dataclasses import make_dataclass
from typing import cast, Type

# 仅用于静态分析的类，运行时不会实例化
class _PersonStub:
    name: str
    age: int

# 强制转换类型，让Pyright识别字段
Person = cast(Type[_PersonStub], make_dataclass('Person', [('name', str), ('age', int)]))

方式三：改用静态定义的数据类（优先选择） #

如果业务场景不需要动态生成类，直接用@dataclass装饰器静态定义，Pyright天然支持完整的自动补全和类型检查：

from dataclasses import dataclass

@dataclass
class Person:
    name: str
    age: int

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