运算符模块的 Python3 类型提示

Question

如何为应该是运算符函数的参数编写类型提示？例如，看这个函数：import operatorfrom typing import Dict, Tupledef func(thresholds: Dict[str, Tuple[operator, float]] = None):    if thresholds is None:        thresholds = {"height": (operator.lt, 0.7), "width": (operator.gt, 0.1)}    pass当我尝试运行我的代码时，我得到：TypeError: Tuple[t0, t1, ...]: each t must be a type. Got <module 'operator' from ...>.编辑：我意识到运算符是一个模块，而不是类型，并且我正在寻找的参数是该模块中的函数。我想看看是否有任何方法可以让用户知道我期望一个运算符函数作为参数。是在文档字符串中解释的唯一方法吗？

Answer 1

operator是一个模块，而不是一个类或类型。您想要做的是创建一个可以满足传递给的函数的类型提示func。如果我们从头开始编写operator.lt，它可能看起来像这样：

from numbers import Number

def le(a: Number, b: Number) -> bool:

    ...

使用Callable类型我们可以将其写为：

Callable[[Number, Number], bool]

总而言之，你会得到这样的结果：

import operator

from typing import Dict, Tuple, Callable, Optional

from numbers import Number

Operator = Callable[[Number, Number], bool]

OperatorDict = Dict[str, Tuple[Operator, float]]

def func(thresholds: Optional[OperatorDict] = None) -> None:

    if thresholds is None:

        thresholds = {"height": (operator.lt, 0.7), "width": (operator.gt, 0.1)}

    pass