Python类型中的可下标类型

我想检查函数的参数是否可下标。如何使用Python的

键入

模块实现这一点

---

我查过文件，但什么也没找到。但也许可以创建一个自定义的

类型

。我该怎么做？

要提示标准的

\uuu getitem\uuu

行为，请使用

collections.abc

的通用版本，例如、或

要键入任何支持

\uuuu getitem\uuuu

的类型，请定义具有所需行为的自定义

typing.Protocol

from typing import Protocol, Any

class Lookup(Protocol):
      def __getitem__(self, key) -> Any: ...

def get(container: Lookup, key):
    return container[key]

get(['zero', 'one', 'two'], 2)

---

请注意，序列和映射类型是通用的，协议也可以定义为通用的

from typing import Protocol, TypeVar

K = TypeVar('K', contravariant=True)
V = TypeVar('V', covariant=True)


class Lookup(Protocol[K, V]):
    def __getitem__(self, key: K) -> V: ...


def get(container: Lookup[K, V], key: K) -> V:
    return container[key]


get({1: 'one', 2: 'two'}, 2)    # succeeds type checking
get({1: 'one', 2: 'two'}, '2')  # fails type checking

---

按照这些思路（尽管完全未经测试）应该可以做到：

从输入导入协议，TypeVar
K=类型变量（“K”）
V=类型变量（“V”）
类可下标（协议[K，V]）：
def_uugetItem_uu（self，k:k）->V:
...

我对Python或MyPy一无所知，但是包括

\uuu getitem\uuuu

的duck类型可能会有用吗？这是否回答了您的问题？啊，伙计，你赢了。好的一面是，我实际上学习MyPy只是为了回答这个问题，所以至少我从中得到了一些东西：-d这比上提供的解决方案要好，因为它还显示了如何访问键和值的类型

from typing import Protocol, TypeVar

K = TypeVar('K', contravariant=True)
V = TypeVar('V', covariant=True)


class Lookup(Protocol[K, V]):
    def __getitem__(self, key: K) -> V: ...


def get(container: Lookup[K, V], key: K) -> V:
    return container[key]


get({1: 'one', 2: 'two'}, 2)    # succeeds type checking
get({1: 'one', 2: 'two'}, '2')  # fails type checking