0

使用类型存根，我想知道是否可以在 Python 中表达一种类型，允许您为任意数量的参数正确键入：

def test(*args):
  return args

乍一看，我带着：

T = TypeVar('T')
def test(*args: T) -> Tuple[T, ...]:
  return args

但这当然只会正确输入第一个 T。

是为所有 arities 编写覆盖的唯一可能方法吗？

T1 = TypeVar('T1')
T2 = TypeVar('T2')
T3 = TypeVar('T3')
T4 = TypeVar('T4')

@overload
def test(arg1: T1) -> Tuple[T1]: ...
@overload
def test(arg1: T1, arg2: T2) -> Tuple[T1, T2]: ...
@overload
def test(arg1: T1, arg2: T2, arg3: T3) -> Tuple[T1, T2, T3]: ...
@overload
def test(arg1: T1, arg2: T2, arg3: T3, arg4: T4) -> Tuple[T1, T2, T3, T4]: ...
# etc
def test(*args: Any) -> Tuple[Any, ...]:
  return args

这也不完整，因为它没有携带足够的类型信息来键入以下内容：

x: Tuple[int, int, str] = test(*[1, 2, "4"])
4

1 回答 1

1

TLDR：@overload目前是注释某种程度的方差的唯一可行方法。PEP 646 - Variadic Generics是一个提案草案，旨在启用对可变参数的正确注释。

正确的注释方法*args是确定某种程度的受支持的“方差长度”，并使用@overload. 值得注意的是，显式参数只能是位置参数——它们必须是__anonymousor positional, /。最后一个包罗万象的可变参数@overload处理更多参数的情况。

from typing import TypeVar, Tuple, Any, overload

T = TypeVar('T')
T1 = TypeVar('T1')
T2 = TypeVar('T2')
T3 = TypeVar('T3')

# positional parameters via `, /` – Python 3.8+ only
@overload
def test(arg1: T1, /) -> Tuple[T1]: ...
# positional parameters via double underscore prefix
@overload
def test(__arg1: T1, __arg2: T2) -> Tuple[T1, T2]: ...
@overload
def test(__arg1: T1, __arg2: T2, __arg3: T3) -> Tuple[T1, T2, T3]: ...
@overload
def test(*args: T) -> Tuple[T, ...]: ...
# etc
def test(*args: Any) -> Tuple[Any, ...]:
  return args

reveal_type(test(1, 2, "three"))     # note: Revealed type is 'Tuple[builtins.int*, builtins.int*, builtins.str*]'
reveal_type(test(1, 2, "three", 4))  # note: Revealed type is 'builtins.tuple[builtins.object*]'
reveal_type(test(1, 2, 3, 4))        # note: Revealed type is 'builtins.tuple[builtins.int*]'

值得注意的是，虽然打包成可变参数可以被类型化，但解包参数通常不能： 任何容器，但tuple长度是任意的——例如List[int] = [1, 2, 3]——因此它的元素没有确切的类型信息。

# unpack tuple of int, int
reveal_type(test(*(1, 2,)))  # note: Revealed type is 'Tuple[builtins.int*, builtins.int*]'
# unpack list of some ints
reveal_type(test(*[1, 2,]))  # note: Revealed type is 'builtins.tuple[builtins.int*]'
于 2021-06-10T11:48:07.760 回答