如何为Python类动态赋值的属性添加正确的类型注解

Python的静态类型检查工具（比如mypy、pyright）默认不会识别运行时动态给实例挂载的额外属性，你可以根据自己的使用场景选以下方案，都能在TestCase上下文里让检查器正确识别animal.type的类型为Type：

方案1：声明仅类型检查生效的专用子类（最推荐，无运行时开销、类型安全） #

借助typing.TYPE_CHECKING常量，在类型检查阶段声明一个继承自Animal的专用子类型，提前定义好type属性的类型即可，这段代码运行时不会执行，完全不影响原有业务逻辑：

from typing import TYPE_CHECKING

# TYPE_CHECKING为True的场景仅在类型检查工具运行时成立，实际运行时不会进入这个代码块
if TYPE_CHECKING:
    class _TypedAnimal(Animal):
        # 仅补充type属性的类型声明，不需要写额外实现
        type: Type

class TestCase:
    # 把animal属性标注为我们声明的专用子类型
    animal: "_TypedAnimal"

    def __init__(self, name: str) -> None:
        self.animal = Animal(name)
        self.animal.type = Type()
        # 这里访问self.animal.type时，类型检查器会自动识别为Type类型

注意：不要直接修改原有Animal类的定义添加type属性，除非你所有业务里的Animal实例都必然携带这个属性，否则会导致其他使用Animal类的代码位置出现错误的类型提示。

方案2：用cast做类型窄化（适合单次使用的轻量场景） #

如果不想额外定义子类，可以用typing.cast配合结构类型协议，在使用前把实例转换为带type属性的目标类型：

from typing import Protocol, cast

# 定义协议类，描述带type属性的Animal实例结构
class _AnimalWithType(Protocol):
    name: str
    type: Type
    def get_name(self) -> str: ...

class TestCase:
    animal: Animal

    def __init__(self, name: str) -> None:
        self.animal = Animal(name)
        self.animal.type = Type()
        # 做类型转换后，typed_animal的type属性会被识别为Type类型
        typed_animal = cast(_AnimalWithType, self.animal)

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