如何修正Python类型提示以消除Optional[str]无split属性警告？

问题背景 #

我编写了一个处理AWS Lambda上下文的函数，用于查找查询字符串参数，返回模式参考Go语言的常见风格：成功时返回(None, 查询字符串值)，失败时返回(Exception, None)。

触发警告的代码行：

file_name = file_path.split("/")[-1]

完整代码如下：

from typing import Dict, Union, Tuple
import json

class QSException(Exception):
    pass


def get_query_string(
    event: Dict, query_string: str
) -> Union[Tuple[QSException, None], Tuple[None, str]]:
    error = QSException()
    #* [...省略其他逻辑...]
    if query_string not in event["queryStringParameters"]:
        return (error, None)
    return (None, event["queryStringParameters"][query_string])


def get_file(event: Dict, context: Dict) -> Dict:
    err, file_path = get_query_string(event, "file")
    if err is not None:
        message = {"message": "No file specified."}
        return {"statusCode": 403, "body": json.dumps(message)}

    # 从这里开始是正常流程
    file_name = file_path.split("/")[-1]

    #* [...省略其他逻辑...]
    return {
        #* [...返回结果...]
    }

虽然已经提前处理异常并返回403，正常流程中file_path必然是字符串类型，split方法实际可正常调用，但类型检查工具仍抛出警告：

Item "None" of "Optional[str]" has no attribute "split" [union-attr]

解决方案 #

方法1：使用assert断言（简单直接） #

在调用split前添加断言，明确告诉类型检查工具此处file_path不可能为None：

assert file_path is not None, "file_path shouldn't be None in happy path"
file_name = file_path.split("/")[-1]

方法2：用cast强制类型转换 #

通过typing.cast强制指定file_path的类型，适用于你能100%确认变量类型的场景：

from typing import cast

# ...
file_path = cast(str, file_path)
file_name = file_path.split("/")[-1]

方法3：添加兜底类型检查（更健壮） #

虽然正常流程不会走到，但添加类型检查既能消除警告，也能增强代码鲁棒性：

if not isinstance(file_path, str):
    return {"statusCode": 500, "body": json.dumps({"message": "Invalid file path format"})}
file_name = file_path.split("/")[-1]

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