Início / coding / Perguntas / 79291519
Accepted
Pierre-olivier Gendraud
Asked: 2024-12-18 22:51:22 +0800 CST

Como construir um tipo a partir de outro tipo e fazer a verificação mypy funcionar

  • 772

Estou trabalhando com o sistema de dicas de tipo do Python e estou tentando criar um alias de tipo para uma função que seja semelhante a um tipo de função existente, mas com um argumento adicional. Aqui está o que tentei:

from typing import Callable

# Original function type
RK_function = Callable[[float, int], int]

# Attempted new function type with an additional int argument
RK_functionBIS = Callable[[*RK_function.__args__[:-1], int], int]

Eu esperava que RK_functionBIS representasse Callable[[float, int, int], int], o que ele faz quando executado diretamente sem verificar com mypy. No entanto, quando executo mypy para verificação de tipo, recebo este erro: text

erro: Alias ​​de tipo inválido: expressão não é um tipo válido [tipo-válido] Q

python

2 respostas

  1. Best Answer

    Até hoje, o Python não suporta extração direta dos Callableargumentos de um alias para uso em outro tipo. No entanto, se seu trecho de código for representativo do problema que você está resolvendo, então seus RK_functions parecem ser completamente feitos de parâmetros somente posicionais , então você pode pelo menos refatorar a parte dos argumentos. Veja mypy Playground :

    from collections.abc import Callable
from typing_extensions import Unpack, TypeAliasType

RK_function_args = TypeAliasType("RK_function_args", tuple[float, int])

# Original function type
RK_function = TypeAliasType("RK_function", Callable[[Unpack[RK_function_args]], int])

# Attempted new function type with an additional int argument
RK_functionBIS = TypeAliasType("RK_functionBIS", Callable[[Unpack[RK_function_args], int], int])

    >>> bis: RK_functionBIS
>>> res: int = bis(1.0, 2, 3)  # OK
>>> bis("1.0", 2, 3)           # Error: first argument should be float
>>> bis(1.0, 2)                # Error: not enough arguments
    • 1

  2. exatamente o mesmo código que dROOOze, mas adicionei um valor concreto para bis

    from collections.abc import Callable
from typing import TypeAliasType,Unpack

RK_function_args = TypeAliasType("RK_function_args", tuple[float, int])

# Original function type
RK_function = TypeAliasType("RK_function", Callable[[Unpack[RK_function_args]], int])

# Attempted new function type with an additional int argument
RK_functionBIS = TypeAliasType("RK_functionBIS", Callable[[Unpack[RK_function_args], int], int])

def ff(a: float, b:int,c:int)-> int:
    return 2

bis: RK_functionBIS =ff
res: int = bis(1.0, 2, 3)  # OK
print(res)
    • 0

relate perguntas