在 Union[...] 类型上询问 isinstance

Question

我正在尝试就用户定义的类型提出isinstance 问题：ConstData = Union[int, str]：

from typing import Union, Optional
ConstData = Union[int, str]
def foo(x) -> Optional[ConstData]:
    if isinstance(x, ConstData):  # <--- this doesn't work
    # if isinstance(x, (int, str)): <--- this DOES work ...
        return x
    return None

很遗憾，它不起作用：

$ mypy main.py
main.py:4: error: Parameterized generics cannot be used with class or instance checks
main.py:4: error: Argument 2 to "isinstance" has incompatible type "object"; expected "Union[type, Tuple[Union[type, Tuple[Any, ...]], ...]]"
Found 2 errors in 1 file (checked 1 source file)

有什么优雅的方法可以解决这个问题吗？

Answer 1

关于：

“不过，if isinstance(x, get_args(ConstData)): return xmypy 无法推断出x 具有正确的返回类型。它抱怨它是Any”

我认为这是reveal_type() 中的一个错误，Mypy 理解了x 类型的缩小。正如您在下面的代码中看到的，当使用x 调用test() 时，它不会引发错误：

def test(k:Union[int, str]) -> None:
    pass


def foo(x: Any) -> Optional[ConstData]:
    if isinstance(x, get_args(ConstData)):
        reveal_type(x)  # Mypy: Revealed type is "Any"
        test(x)  # Mypy: no error
        return x  # Mypy: no error
    return None