Python typing中如何为TypeVar指定额外的子类边界

from typing import Generic, TypeVar, Protocol, overload, Literal, cast, runtime_checkable

T = TypeVar("T")

@runtime_checkable
class HasAttr(Protocol):
    attr: str

class AppendListWithAttrs(Generic[T]):
    def __init__(self):
        self._list: list[T] = []
        self._attrs: list[str] = []

    @overload
    def append(self, obj: HasAttr, record_attr: Literal[True]) -> None: ...
    
    @overload
    def append(self, obj: T, record_attr: Literal[False] = False) -> None: ...

    def append(self, obj: T, record_attr: bool = False) -> None:
        self._list.append(obj)
        if record_attr:
            # 方式1：运行时检查类型，mypy会自动窄化类型，安全性更高
            assert isinstance(obj, HasAttr)
            self._attrs.append(obj.attr)
            # 方式2：无运行时开销，需要开发者自行保证调用符合规则
            # self._attrs.append(cast(HasAttr, obj).attr)

核心修改说明 #

• 给HasAttr协议添加@runtime_checkable装饰器，支持运行时用isinstance判断对象是否符合协议，mypy可识别该判断自动窄化类型
• 修正重载签名：第一个重载对应record_attr=True场景，强制要求传入的obj必须满足HasAttr协议；第二个重载对应不传record_attr或传False的场景，仅要求对象符合泛型T的类型约束即可
• 在访问obj.attr前添加类型窄化逻辑，明确告知mypy进入该分支的对象一定具备attr属性，消除类型报错

你提供的测试用例可完全通过mypy校验，且如果错误传入无attr属性的对象+record_attr=True参数，mypy会自动抛出类型错误，符合预期需求。

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