具有任意数量的 int 参数键入提示的函数

如何解决具有任意数量的 int 参数键入提示的函数

def f() -> Callable[[ # how to show there can be any number of int?
                    ],float]:
    def g(*args):
        assert all(type(x) == int for x in args)
        return 0.1
    return g

我阅读了打字文档，但 Callable（即 Callable[…,ReturnType]）不是我需要的。

我知道 Tuple[int,…]，但 Callable[[int,…],float] 返回错误 "…" not allowed in this context Pylance。

解决方法

您可以通过定义带有 Protocol 的 __call__ 来实现此目的：

from typing import Protocol


class IntCallable(Protocol):
    def __call__(self,*args: int) -> float: ...


def f() -> IntCallable:
    def g(*args: int) -> float:
        assert all(type(x) == int for x in args)
        return 0.1
    return g

用 mypy 测试一下：

f()(1,2,3)  # fine
f()("foo")    # error: Argument 1 to "__call__" of "IntCallable" has incompatible type "str"; expected "int"

另一种选择是让您的函数采用单个 Iterable[int] 参数而不是任意数量的 int 参数，这样您就可以使用简单的 Callable 输入而不必去更复杂的 Protocol 路线。