如何根据运行时条件指定类型

Question

我正在使用 mypy 和协议，如果可能的话，我遇到了一个我想要类型提示的地方，但我无法弄清楚我应该如何设置它，这样 mypy 就不会出错。

考虑以下示例：

class TProtocol(Protocol):
    t: str

@attrs(auto_attribs=True)
class T:
    t: str
    t2: str

@attrs(auto_attribs=True)
class T2:
    t: str

def func(var: TProtocol) -> None:
    if some_condition:
        var: T
        reveal_type(var)
    else:
        reveal_type(var)

虽然非常做作，但它说明了我的目标，即我有一些运行时条件，如果满足，我会根据代码库的知识知道该变量的类型。然后我想将这些知识传递给 mypy，以便进一步的类型检查使用该类型。

同样的例子可以用 Union 代替。一些运行时检查会根据代码库的知识明确告诉我我拥有哪种类型。然后，我想根据外部知识明确告诉 mypy 哪种类型用于进一步类型检查。

上面的示例引发了一个错误，指出 var 已定义。我尝试了allow_redefinition 选项，但它并没有改变输出。

Answer 1

使用typing.cast 强制将变量声明为特定类型。这会忽略任何其他类型的信息，并且也适用于运行时分支。

def func(var: TProtocol) -> None:
    reveal_type(var)         # line 21
    if some_condition:
        var = cast(T, var)   # line 23
        reveal_type(var)
    else:
        var = cast(T2, var)  # line 26
        reveal_type(var)

这使得mypy 以不同的方式处理var 的每次转换，即在每个分支之前和内部：

type_tests.py:21: note: Revealed type is 'type_tests.TProtocol'
type_tests.py:24: note: Revealed type is 'type_tests.T'
type_tests.py:27: note: Revealed type is 'type_tests.T2'