输入通用（协议）类参数的注释

Question

我正在尝试计算如何为应该是实现通用协议的类函数参数添加类型注释。

例如，假设我有一个可以看起来像这样的集合的协议：

from typing import (
    Protocol, TypeVar, Iterable
)

T = TypeVar('T', contravariant=True)


class Set(Protocol[T]):
    """A set of elements of type T."""

    def __init__(self, init: Iterable[T]) -> None:
        """Initialise set with init."""
        ...

    def __contains__(self, x: T) -> bool:
        """Test if x is in set."""
        ...

    def add(self, x: T) -> None:
        """Add x to the set."""
        ...

    def remove(self, x: T) -> None:
        """Remove x from the set."""
        ...

并且我有一种使用各种类型集的算法，我想通过设置实现来参数。为简单起见，我将在此功能中创建一个列表以用作示例：

from typing import Type

def foo(set_type: Type[Set]) -> None:
    """Do clever stuff."""
    x = list(range(10))
    s = set_type(x)
    ...

mypy告诉我set缺少一个类型参数，我想是正确的，但我不想给它一个，因为我打算使用不同类型的set_type。

如果我给出set a typevar而是

def foo(set_type: Type[Set[T]]) -> None:
    """Do clever stuff"""
    x = list(range(10))
    s = set_type(x)
    ...

得到我的警告：i set_type（）获得不兼容的类型，list> list [int]而不是itoble [t]，这又是正确的，但对我没有太大帮助。

有没有办法指定我的函数参数可以用作不同类型集的通用构造函数？

Answer 1

协议什么也没说__ Init __的签名，即使它是在协议上定义的。 type做类似的事情 - 即使set不是协议，type [set]什么也没说如何调用类型。

我最初建议使用callable [[iTable [t]]，set [t]]。但是，这是有问题的，并且仅是因为我省略了通用参数，从本质上讲，它使其成为>任何，如这个github问题。您可以使用（而不是冗长）协议。

class MkSet(Protocol):
    def __call__(self, it: Iterable[T]) -> Set[T]:
        ...

def foo(set_type: MkSet) -> None:
    ...