Введите аннотации для *args и **kwargs. ⇐ Python

[Anonymous]

Я пробую аннотации типов Python с абстрактными базовыми классами для написания некоторых интерфейсов. Есть ли способ аннотировать возможные типы *args и **kwargs?

Например, как бы выразить, что разумное аргументами функции являются либо int, либо два int? type(args) дает Tuple, поэтому я предполагал аннотировать тип как Union[Tuple[int, int], Tuple[int]], но это не работает.< /p>

Код: Выделить всё

from typing import Union, Tuple

def foo(*args: Union[Tuple[int, int], Tuple[int]]):
try:
i, j = args
return i + j
except ValueError:
assert len(args) == 1
i = args[0]
return i

# ok
print(foo((1,)))
print(foo((1, 2)))
# mypy does not like this
print(foo(1))
print(foo(1, 2))

Сообщения об ошибках от mypy:

Код: Выделить всё

t.py: note: In function "foo":
t.py:6: error: Unsupported operand types for + ("tuple" and "Union[Tuple[int, int], Tuple[int]]")
t.py: note: At top level:
t.py:12: error: Argument 1 to "foo" has incompatible type "int"; expected "Union[Tuple[int, int], Tuple[int]]"
t.py:14: error: Argument 1 to "foo" has incompatible type "int"; expected "Union[Tuple[int, int], Tuple[int]]"
t.py:15: error: Argument 1 to "foo" has incompatible type "int"; expected "Union[Tuple[int, int], Tuple[int]]"
t.py:15: error: Argument 2 to "foo" has incompatible type "int"; expected "Union[Tuple[int, int], Tuple[int]]"

Возможно, mypy не нравится это для вызова функции, поскольку он ожидает наличия кортежа в самом вызове. Добавление после распаковки также приводит к ошибке ввода, которую я не понимаю.

Как аннотировать разумные типы для *args и **kwargs< /код>?

Подробнее здесь: https://stackoverflow.com/questions/370 ... and-kwargs