如何结合字符串使用typing.Literal以适配mypy静态类型检查？

当你把字符串字面量赋值给变量后传入要求Literal类型参数的函数时，mypy会报错，原因是默认情况下mypy会将普通变量的类型推断为宽泛的str，而非你需要的Literal["best", "worst"]。以下是几种可行的解决方法：

方法1：给变量显式标注Literal类型 #

直接为变量指定Literal["best", "worst"]类型，让mypy明确变量的类型范围：

from typing import Literal

def literal_func(string_input: Literal["best", "worst"]) -> int:
    if string_input == "best":
        return 1
    elif string_input == "worst":
        return 0

# 显式标注变量类型
input_string: Literal["best", "worst"] = "best"
literal_func(string_input=input_string)  # mypy无报错

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

如果无法直接给变量标注类型（比如变量值来自其他逻辑），可以用cast告诉mypy变量的实际类型：

from typing import Literal, cast

def literal_func(string_input: Literal["best", "worst"]) -> int:
    if string_input == "best":
        return 1
    elif string_input == "worst":
        return 0

input_string = "best"
# 强制转换为目标Literal类型
literal_func(string_input=cast(Literal["best", "worst"], input_string))  # mypy无报错

注意：cast仅在类型检查阶段生效，不会改变变量的实际运行时类型，所以要确保变量值确实在Literal的允许范围内，避免运行时逻辑错误。

方法3：用typing.Final标记不可变变量（Python 3.8+） #

如果变量不会被重新赋值，可以用Final标记，mypy会保留变量的字面量类型（推断为Literal["best"]，而它是Literal["best", "worst"]的子类型，符合参数要求）：

from typing import Literal, Final

def literal_func(string_input: Literal["best", "worst"]) -> int:
    if string_input == "best":
        return 1
    elif string_input == "worst":
        return 0

# 用Final标记变量，mypy会推断其为Literal["best"]
input_string: Final = "best"
literal_func(string_input=input_string)  # mypy无报错

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