Скопировать подпись типа из другой функции

Представьте, что у меня есть набор функций, как показано ниже. foo имеет множество аргументов различных типов, а bar передает все свои аргументы этой другой функции. Есть ли способ заставить mypy понять, что bar имеет тот же тип, что и foo, без явного копирования всего списка аргументов?

def foo(a: int, b: float, c: str, d: bool, *e: str, f: str = "a", g: str = "b") -> str:
    ...

def bar(*args, **kwargs):
    val = foo(*args, **kwargs)
    ...
    return val

python python-typing mypy
person Caagr98    schedule 13.01.2020    source источник
comment
Это описано в PEP-612 (ParamSpec). Поддержка mypy все еще не реализована (см. mypy#8645).   -  person Danielle Madeley    schedule 24.12.2020

Ответы (1)


arrow_upward
4
arrow_downward

Было много обсуждений добавления этой функции здесь. Для простого случая передачи всех аргументов вы можете использовать рецепт из этого комментария:

F = TypeVar('F', bound=Callable[..., Any])

class copy_signature(Generic[F]):
    def __init__(self, target: F) -> None: ...
    def __call__(self, wrapped: Callable[..., Any]) -> F: ...

def f(x: bool, *extra: int) -> str: ...

@copy_signature(f)
def test(*args, **kwargs):
    return f(*args, **kwargs)

reveal_type(test)  # Revealed type is 'def (x: bool, *extra: int) -> str'
person Alex Hall    schedule 13.01.2020
comment
Это работает, но с некоторыми недостатками: --strict по-прежнему сообщает, что параметры нетипизированы, а возвращаемый тип __init__ — None, поэтому копирование не совсем работает. - person Caagr98; 13.01.2020
comment
Если бы вы могли добавить комментарий об этом в вопрос, это было бы здорово. - person Alex Hall; 13.01.2020