如何让Python类型检查器确认配置字典取值不会为None？

你遇到的问题是：全局定义的配置常量，虽然已经通过if URL is None做了非空校验并抛出异常，但类型检查器依然认为它可能是None，导致传入要求str类型的函数时报错。这是因为类型检查器对全局作用域的控制流分析能力有限，无法像函数内部那样精准跟踪变量类型变化。

下面是几种可行的解决方法：

方法1：用断言确认非空 #

在抛出异常后添加assert语句，明确告诉类型检查器URL不可能为None：

"""Type-checking test"""
from dotenv import dotenv_values

config = dotenv_values(".config")
URL = config.get("BASE_URL")
if URL is None:
    raise ValueError("BASE_URL is not set in .config file")
assert URL is not None  # 类型检查器会据此认定URL为str类型

def do_a_thing(url: str) -> None:
    """Do a thing"""
    print("You passed in a URL:", url)

def main():
    """Main"""
    do_a_thing(URL)

if __name__ == "__main__":
    main()

方法2：使用临时变量+类型注解 #

先通过临时变量处理配置值，再将其赋值给带有明确类型注解的常量：

"""Type-checking test"""
from dotenv import dotenv_values

config = dotenv_values(".config")
# 用临时变量先做非空校验
_temp_url = config.get("BASE_URL")
if _temp_url is None:
    raise ValueError("BASE_URL is not set in .config file")
# 给常量添加明确的str类型注解
URL: str = _temp_url

def do_a_thing(url: str) -> None:
    """Do a thing"""
    print("You passed in a URL:", url)

def main():
    """Main"""
    do_a_thing(URL)

if __name__ == "__main__":
    main()

方法3：用typing.cast强制类型转换 #

导入cast工具，直接告诉类型检查器配置值的类型为str，运行时不会产生额外开销：

"""Type-checking test"""
from dotenv import dotenv_values
from typing import cast

config = dotenv_values(".config")
URL = cast(str, config.get("BASE_URL"))
if URL is None:
    raise ValueError("BASE_URL is not set in .config file")

def do_a_thing(url: str) -> None:
    """Do a thing"""
    print("You passed in a URL:", url)

def main():
    """Main"""
    do_a_thing(URL)

if __name__ == "__main__":
    main()

这三种方法都能让类型检查器确信URL永远不会是None，同时保留全局常量的命名规范。

内容的提问来源于stack exchange，提问作者Gordon Mckeown