如何解决TYPE_CHECKING条件下导入模块时cast运行报错问题，或向mypy指定特定场景的返回类型？

我之前也碰到过一模一样的情况——为了避免运行时冗余导入用了TYPE_CHECKING，结果用cast的时候栽了跟头。下面给你几个实用的解决方案，都是我亲测有效的：

方案1：用字符串字面量作为cast的类型参数 #

这是最优雅的解决方式，既保留了cast的简洁性，又完全避免了运行时的模块导入问题。mypy支持PEP 484定义的字符串形式的前向引用，你只需要把bar.Bar改成字符串"bar.Bar"：

from typing import cast

# ... 其他代码 ...

bar_obj: bar.Bar = cast(foo_obj.get_something(), "bar.Bar")

原理： #

• 运行时：cast函数的第二个参数只是一个字符串，Python不会去解析它，自然不会触发bar模块的导入，完全是空操作（nop）。
• 类型检查时：mypy会把这个字符串解析成对应的bar.Bar类型，正确识别你的类型断言。

方案2：用TYPE_CHECKING包裹断言逻辑 #

如果你更倾向于用断言来表达“我确定这里返回的是bar.Bar”，可以把断言放在TYPE_CHECKING的条件块里，这样运行时不会执行断言，也就不会触发isinstance对bar模块的依赖：

from typing import TYPE_CHECKING

something = foo_obj.get_something()
if TYPE_CHECKING:
    # 仅在类型检查时执行，运行时跳过
    assert isinstance(something, bar.Bar)
bar_obj: bar.Bar = something

优点： #

• 代码可读性强，明确传达了你对返回值的假设。
• 运行时没有任何额外开销，也不会报错。

方案3：延迟导入模块（折中方案） #

如果你实在不想用字符串或条件断言，可以用延迟导入的方式在需要的时候才导入bar模块，但只在运行时的cast场景下使用：

from typing import cast, TYPE_CHECKING

if TYPE_CHECKING:
    import bar

def my_func():
    foo_obj: foo.Foo = foo.Foo()
    # 仅在运行时需要cast的参数时导入
    if not TYPE_CHECKING:
        import bar
    bar_obj: bar.Bar = cast(foo_obj.get_something(), bar.Bar)

不过这个方案不如前两个优雅，毕竟还是导入了模块，只是延迟了导入时机，适合一些特殊场景。

为什么原来的cast会报错？ #

你可能好奇为什么之前的cast会触发模块错误——因为Python在运行时会求值cast的第二个参数，也就是bar.Bar，但此时bar模块并没有被实际导入（TYPE_CHECKING块在运行时不会执行），所以会抛出ModuleNotFoundError。而字符串参数不会触发这个求值过程，完美避开了问题。

内容的提问来源于stack exchange，提问作者K.Mulier