python - 如何让Mypy意识到在某些情况下不会使用默认值

Question

我有以下功能:

#!/usr/bin/env python3
from typing import Union

def foo(b: Union[str, int]) -> int:
    def bar(a: int = b) -> int: # Incompatible default for argument "a" (default has type "Union[str, int]", argument has type "int")
        return a + 1

    if isinstance(b, str):
        return bar(0)
    else:
        return bar()

print(foo(3)) # 4
print(foo("hello")) # 1

在我定义 bar 的行上，Mypy 表示将 b 设置为默认值将不起作用。

但是，由于程序的工作方式，使用默认 b 的唯一方法是如果 b 是整数。所以这应该可以正常工作。

但 Mypy 没有意识到这一点。

我怎样才能

1. 让 Mypy 认识到 int 是 a 的正确类型或
2. 修复此问题的方法不会导致太多代码重复。

(例如，我知道我可以编写两个具有不同签名的 foo 函数，但这会导致太多代码重复。)

下面的 TL;DR 只是我的现实用例，因为至少有一个答案依赖于我上面的 MCVE 的简单程度。

这是一个需要字典的函数。该函数返回一个装饰器，使用该装饰器时，会将装饰函数(装饰函数是 TypeChecker)添加到字典中。装饰器允许使用一个参数来指定装饰函数(TypeChecker)在字典中放置的名称/键。如果未指定名称，则它将使用不同的函数 (StringHelper.str_function) 从函数本身的属性中找出名称。

由于装饰器参数的工作方式，装饰器创建者需要接受名称(或什么也不接受)或函数。如果它只接受函数，则没有指定名称，并且它应该从函数中获取名称。如果它只需要名称，那么它将在函数上再次调用，并且应该使用该名称。如果它什么都不做，那么它将在该函数上再次调用，并且它应该从该函数中获取一个名称。

def get_type_checker_decorator(type_checkers: Dict[str, TypeChecker]) -> Callable[[Union[Optional[str], TypeChecker]], Union[Callable[[TypeChecker], TypeChecker], TypeChecker]]:
    @overload
    def type_checker_decorator(name: Optional[str]) -> Callable[[TypeChecker], TypeChecker]:
        pass
    @overload
    def type_checker_decorator(name: TypeChecker) -> TypeChecker:
        pass
    def type_checker_decorator(name: Union[Optional[str], TypeChecker] = None) -> Union[Callable[[TypeChecker], TypeChecker], TypeChecker]:
        # if name is a function, then the default will never be used
        def inner_decorator(function: TypeChecker, name: Optional[str] = name) -> TypeChecker: # this gives the Mypy error
            if name is None:
                name = StringHelper.str_function(function)
            type_checkers[name] = function
            def wrapper(string: str) -> bool:
                return function(string)
            return wrapper

        if callable(name):
            # they just gave us the function right away without a name
            function = name
            name = None
            return inner_decorator(function, name)
        else:
            assert isinstance(name, str) or name is None
            # should be called with just the function as a parameter
            # the name will default to the given name (which may be None)
            return inner_decorator

    return type_checker_decorator

Answer 1

如果这不是函数真正所期望的，那么强制类型签名会感觉很尴尬。您的 bar 函数显然需要一个 int，并在类型提示上强制使用 Union，以便稍后断言您实际上只接受 int s 不应该是使 mypy 沉默的必要条件。

由于您接受 b 作为 bar 中的默认值，因此您应该注意 bar 内部的 str 情况，因为类型b 的签名已在 foo 中指定。我认为更适合当前问题的两种替代解决方案:

def foo(b: Union[str, int]) -> int:

    # bar takes care of the str case. Type of b already documented
    def bar(a=b) -> int:
        if isinstance(b, str):
            return bar(0)
        return a + 1

    return bar()

在定义bar之前定义默认值:

def foo(b: Union[str, int]) -> int:

    x: int = 0 if isinstance(b, str) else b

    # bar does not take a default type it won't use.
    def bar(a: int = x) -> int:
        return a + 1

    return bar()