Python3鸭子类型注解实践：dataclass参数的mypy类型检查问题

问题场景 #

你尝试给函数参数添加类型注解，要求只要传入的dataclass至少拥有a和b属性（允许有额外属性）就能通过检查，于是定义了CarInterface作为接口，用TypeVar绑定它，但运行mypy时遇到两个错误：

• 函数blah的mar注解提示：TypeVar 'mar'在泛型函数签名中仅出现一次
• 传入car_instance时提示：Car类型无法赋值给CarInterface类型

错误原因 #

1. TypeVar使用不当：泛型TypeVar的设计目的是关联函数签名中多个位置的类型（比如参数与返回值同类型），仅在参数中单独出现一次没有意义，因此mypy抛出错误。
2. 名义类型检查限制：Mypy默认采用名义类型系统，只有显式继承的类才会被视为子类。你的Car没有继承CarInterface，所以mypy不认可两者的类型兼容性。

解决方案：使用Protocol实现结构类型（鸭子类型） #

Python的typing.Protocol（Python 3.8+，旧版本可使用typing_extensions.Protocol）专门用于定义结构接口——只要类拥有Protocol中声明的所有属性/方法，就会被认为符合该接口，无需显式继承，完美适配鸭子类型的需求。

示例代码 #

from dataclasses import dataclass
from typing import Protocol

# 定义结构接口：只要类包含a和b属性即符合要求
class CarInterface(Protocol):
    a: int
    b: str

# 实际的dataclass，包含额外属性zar
@dataclass
class Car:
    a: int
    b: str
    zar: bool

# 函数直接用CarInterface作为参数类型
def blah(car: CarInterface) -> None:
    print(car.a, car.b)

# 使用示例
car_instance = Car(a=1, b="test", zar=True)
blah(car_instance)  # mypy无报错

方案优势 #

• 无需抽象类，新增类时只需保证包含a和b属性即可，定义简单。
• 完美实现鸭子类型检查，mypy会自动验证传入对象的结构是否符合要求。
• 避免了TypeVar使用不当带来的错误。

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