Question

使用mypy输入以下代码进行类型检查：

def foo(a: str, b: float, c: int):
    print(a, b, c + 1)

foo('ok', 2.2, 'bad')

使用以下内容显示无效调用foo：

error: Argument 3 to "foo" has incompatible type "str"; expected "int"

现在假设我们有一个如下所示的包装函数：

from typing import Callable, Any

def say_hi_and_call(func: Callable[..., Any], *args):
    print('Hi.')
    func(*args)

并使用它进行无效通话

say_hi_and_call(foo, 'ok', 2.2, 'bad')

mypy不会报告任何错误，相反，我们只会在运行时了解此错误：

TypeError: must be str, not int

我想早点发现这个错误。是否有可能以mypy能够报告问题的方式优化类型注释？

Answer 1

好的，我想出的唯一解决方案是使函数的含义明确化，即

from typing import Any, Callable, TypeVar

A = TypeVar('A')
B = TypeVar('B')
C = TypeVar('C')

def say_hi_and_call_ternary(func: Callable[[A, B, C], Any], a: A, b: B, c: C):
    print('Hi.')
    func(a, b, c)

def foo(a: str, b: float, c: int):
    print(a, b, c + 1)

say_hi_and_call_ternary(foo, 'ok', 2.2, 'bad')

当然，人们也需要类似的say_hi_and_call_unary和say_hi_and_call_binary等。

但是由于我认为我的应用程序不会在PROD中爆炸而不是保存一些LOC，因此我很高兴mypy能够报告错误，现在确实是这种情况：

error: Argument 1 to "say_hi_and_call_ternary" has incompatible type "Callable[[str, float, int], Any]"; expected "Callable[[str, float, str], Any]"

使用* args时输入函数参数的安全性（mypy）

1 个答案: