主页 / coding / 问题 / 79291519
Accepted
Pierre-olivier Gendraud
Asked: 2024-12-18 22:51:22 +0800 CST

如何从另一个类型构建一个类型并使 mypy 检查工作

  • 772

我正在使用 Python 的类型提示系统，并尝试为与现有函数类型类似的函数创建类型别名，但多了一个参数。以下是我尝试过的方法：

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]

我预期 RK_functionBIS 表示 Callable[[float, int, int], int]，直接运行时确实如此，无需使用 mypy 检查。但是，当我运行 mypy 进行类型检查时，出现此错误：text

错误：无效的类型别名：表达式不是有效类型 [valid-type] Q

python

2 个回答

  1. Best Answer

    截至目前，Python 不支持直接提取Callable别名的参数以用于其他类型。但是，如果您的代码片段代表了您正在解决的问题，那么您的s 看起来就像完全由位置专用参数RK_function组成，因此您至少可以重构出参数部分。请参阅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. 与 dROOOze 的代码完全相同，但我为 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

相关问题