如何正确标注LinkedList的迭代返回类型与add方法参数类型？

问题场景 #

现有一段Python泛型链表代码，已为__next__添加返回类型注解，但PyCharm仍将迭代时的elem识别为int（而非ListElem[int]）。需求如下：

• 迭代LinkedList时返回ListElem[T]类型节点
• add方法接收T类型的参数（如int）
• 避免重复标注（如LinkedList[ListElem[int], int]这类冗余写法）

问题根源 #

1. LinkedList的elem_factory参数类型注解未绑定泛型参数T，导致类型检查器无法关联节点与值的类型
2. __iter__方法未明确返回Iterator[ListElem[T]]，类型检查器无法正确推断迭代元素的类型

修正后的代码 #

from __future__ import annotations

from typing import TypeVar, Generic, Type, Iterator


T = TypeVar('T')


class ListElem(Generic[T]):
    def __init__(self, value: T, nxt: ListElem[T] | None = None):
        self.value = value
        self.nxt = nxt


class LinkedList(Generic[T]):
    def __init__(self, elem_factory: Type[ListElem[T]], head: ListElem[T] | None = None):
        self._elem_factory = elem_factory
        self._head = head

    def add(self, value: T):
        elem = self._elem_factory(value, self._head)
        self._head = elem

    def __iter__(self) -> Iterator[ListElem[T]]:
        self._next = self._head
        return self

    def __next__(self) -> ListElem[T]:
        if not self._next:
            raise StopIteration

        result = self._next
        self._next = self._next.nxt

        return result


def main():
    lst: LinkedList[int] = LinkedList(ListElem)
    lst.add(5)

    for elem in lst:
        print(elem.value)  # 现在PyCharm会正确识别elem为ListElem[int]


if __name__ == '__main__':
    main()

关键修正点 #

• 将elem_factory的类型从Type[ListElem]改为Type[ListElem[T]]，明确工厂类创建的是绑定了T的节点类型
• 给__iter__添加返回类型Iterator[ListElem[T]]，直接告知类型检查器迭代器的返回元素类型
• 用ListElem[T] | None替代原有的ListElem[T] = None，符合Python 3.10+的类型注解语法（也可保留原写法，视Python版本调整）

这样修改后，LinkedList[int]中的int既代表add方法接收的值类型，也关联到ListElem[int]节点类型，类型检查器能正确识别迭代时的elem为ListElem[int]，同时避免了冗余标注。

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