如何在泛型子类化/继承的for循环中使用mypy注解？

我来帮你分析这个问题，其实核心原因出在父类Element的存根定义上，它没有被声明为泛型类，导致mypy无法正确推断子类HtmlElement调用getchildren()时的返回类型。

问题根源 #

看你给出的Element类存根：

T = TypeVar('T', bound='Element')
class Element():
    attrib: Dict[str,str]
    def getchildren(self)->List[T]:...

这里的T是一个绑定到Element的类型变量，但Element本身并不是泛型类（没有继承Generic[T]）。这就导致mypy无法将T与子类HtmlElement关联起来——当你调用HtmlElement实例的getchildren()时，mypy没法确定返回的列表元素是HtmlElement类型，只能模糊地认为是Element的某个子类。

当你提前注解b_element: HtmlElement然后在循环中赋值时，mypy会检查迭代元素的类型是否和变量注解匹配，但因为它没法确定getchildren()返回的元素是HtmlElement，就会抛出<nothing> has no attribute "drop"的错误（本质是类型推断失败）。

解决方案 #

方案1：修正父类存根为泛型类 #

这是最彻底的解决方法，让Element成为泛型类，这样子类继承时可以明确绑定类型变量：

修改后的stubs/xml/__init__.pyi：

from typing import Dict, List, Optional, TypeVar, Generic

T = TypeVar('T', bound='Element[T]')

class Element(Generic[T]):
    attrib: Dict[str, str]
    def getchildren(self) -> List[T]: ...
    def getparent(self) -> List[Element[T]]: ...
    def drop(self) -> None: ...

修改后的stubs/xml/html/__init__.pyi：

from xml import Element

class HtmlElement(Element['HtmlElement']): ...

这样定义后，HtmlElement的getchildren()会明确返回List[HtmlElement]，mypy就能正确推断循环中元素的类型，不管你是提前注解变量还是直接循环都不会报错。

方案2：使用Python 3.11+的Self类型（推荐） #

如果你使用的是Python 3.11及以上版本，mypy支持Self类型，可以更简洁地定义父类：

修改后的stubs/xml/__init__.pyi：

from typing import Dict, List, Self

class Element:
    attrib: Dict[str, str]
    def getchildren(self) -> List[Self]: ...
    def getparent(self) -> List[Self]: ...
    def drop(self) -> None: ...

子类HtmlElement无需额外修改，继承后getchildren()会自动返回List[HtmlElement]，mypy的类型推断会完全正常工作。

方案3：临时绕过（无法修改存根时） #

如果你没办法修改父类的存根，可以在循环中显式标注元素类型：

for b_element in b.getchildren():  # type: HtmlElement
    b_element.drop()

或者去掉提前的变量注解，让mypy通过上下文推断类型：

for b_element in b.getchildren():
    b_element.drop()  # mypy现在能正确推断b_element是HtmlElement

这不是mypy的bug，而是因为父类的泛型定义不完整，导致类型推断链断裂。修正泛型定义后，所有场景都能正常通过mypy检查。

内容的提问来源于stack exchange，提问作者Benoît Galy