Python 对于任何也暴露 len() 的可迭代对象，正确的类型是什么？ [复制]

Question

我有一个函数适用于任何可以返回其长度的迭代。 所以它适用于 list、dict 和 dict.tems()，即ItemsView。

我应该使用什么正确的打字方式？

编辑： 这是一个示例函数：

def print_iter(some_iterable: xxx):
    for idx, val in enumerate(some_iterable):
        print(val, idx, len(some_iterable))

我不确定要使用什么类型来代替 xxx

Answer 1

您需要导入typing_extensions 模块并定义一个新的Protocol 子类。

from typing_extensions import Protocol


class SupportsLen(Protocol):
    def __len__(self) -> int:
        return 0  # Exact value unimportant; this is for the type checker only.


def print_iter(some_iterable: SupportsLen):
    for idx, val in enumerate(some_iterable):
        print(val, idx, len(some_iterable))

（为了完整起见，SupportsLen 可能还应该定义您需要的任何内容，以使其真正可迭代。）