支持__getitem__的类的Python类型提示

Question

我想向函数添加类型提示，该函数将接受带有__getitem__方法的任何对象。例如，在

def my_function(hasitems, locator):
    hasitems[locator]

我不想限制hasitems为特定类型，例如list或dict。只要它支持__getitem__，它就是的适当参数my_function。如何在没有不必要限制的情况下注释其类型？

编辑：显然，PyCharm可以在许多常见情况下得出适当的提示，但在我的实际用例中却不能。我无法发布代码，因为它是用于工作的，而且我找不到PyCharm失败的非专有最小示例。无论如何，原始问题不会引用PyCharm，它仍然是类型提示的有效用例。

Answer 1

如果你愿意安装一个不很-官方扩展typing，打字的扩展，您可以使用Protocol，这应该是一个实现PEP-0544：

from typing_extensions import Protocol
from typing import Any

class GetItem(Protocol):
    def __getitem__(self: 'Getitem', key: Any) -> Any: pass

class BadGetItem:
    def __getitem__(self, a: int, b: int) -> Any: pass

def do_thing(arg: GetItem):
    pass

do_thing(dict())  # OK
do_thing(BadGetItem())  # Fails with explanation of correct signature
do_thing(1)  # Fails