当参数默认值由模块级别变量确定时，构成超载函数

Question

我具有带有输入参数的Python函数，其值控制返回值的类型。可以省略参数（默认值 none ），在这种情况下，使用模块级别变量。这样，可以通过更改模块级别变量来更改默认行为。 最小示例：

from typing import overload, Union, Literal, Optional

option_default: Literal["str", "int"] = "int"


@overload
def test(option: Literal["str"]) -> str:
    ...


@overload
def test(option: Literal["int"]) -> int:
    ...


@overload
def test(option: Literal[None] = None) -> Union[str, int]:
    ...


def test(option: Optional[Literal["str", "int"]] = None) -> Union[str, int]:

    if option is None:
        option = option_default

    if option == "str":
        return "foo"
    else:
        return 1


foo: int = test() # Incompatible types in assignment (expression has type "Union[str, int]", variable has type "int")

option_default = "str"
baz: str = test() # Incompatible types in assignment (expression has type "Union[str, int]", variable has type "str")

通常，人们会使用typing.overload来处理这种情况。但是，对于test（），没有参数无法正常工作的参数，因此会出现mypy错误。这里有更好的方法在这里提供类型信息吗？

Answer 1

那是不可能的。有 github上的建议 添加typescript的typef ，但这是另一回事 - 它不会像您显示的那样跟踪可变变量。

我认为即使在理论上也无法实施这样的功能。它为类型检查带来了许多新的复杂性。我可以想到一些有问题的方案：

1. 通过回调

def foo(callback: Callable[[], int]):
    some_module.definitely_returns_an_int = callback

option_default = "int"
foo(test)  # ok?
option_default = "str"

现在foo保存了一个返回字符串的函数。

2. 跟踪在功能中发生的突变

def ints_are_the_bomb():
    global option_default
    option_default = "int"

option_default = "str"
some_other_module.baz()

value = test()

您可以确定value的类型？无法保证some_other_module.baz（）未调用your_module.ints_are_the_bomb（）。因此，现在您需要跟踪所有可能发生在your_module.option_default的更改，可能是跨模块的。如果客户端代码（如果是库）可以更改标志，那么这是不可能的。

要概括，当您突变某些东西时，值（包括函数）的类型（包括函数）不会改变。这可能会破坏遥远的代码，而遥远的代码也恰好引用了此对象。