不幸的是,目前这不是可以使用PEP 484类型提示键入的内容.
你可以做的最好的事情是使用重载来近似签名:基本上,我们硬编码签名应该达到一定数量然后再回到推断’任何’:
from typing import TypeVar, overload, Any, Callable
T1 = TypeVar('T1')
T2 = TypeVar('T2')
T3 = TypeVar('T3')
T4 = TypeVar('T4')
@overload
def chain_call(input_value: T1,
*f_rest: Callable[[T1], T1]) -> T1: ...
@overload
def chain_call(input_value: T1,
f1: Callable[[T1], T2],
f2: Callable[[T2], T3],
f3: Callable[[T3], T4],
f4: Callable[[T4], Any],
*f_rest: Callable[[Any], Any]) -> Any: ...
@overload
def chain_call(input_value: T1,
f1: Callable[[T1], T2],
f2: Callable[[T2], T3],
f3: Callable[[T3], T4]) -> T4: ...
@overload
def chain_call(input_value: T1,
f1: Callable[[T1], T2],
f2: Callable[[T2], T3]) -> T3: ...
@overload
def chain_call(input_value: T1,
f1: Callable[[T1], T2]) -> T2: ...
def chain_call(input_value, *f_rest):
for function in f_rest:
input_value = function(input_value)
return input_value
在这里,我硬编码了最多3个输入函数应该发生的事情(并且在所有callables碰巧具有相同输入和输出类型的特殊情况下以重载开始).
这种技术是类型化的当前类型,如zip函数,它可以接受任意数量的iterables.
注意:您可能需要使用master中最新版本的mypy才能使此代码逐字工作.