String lower方法无法保留Literal类型引发类型匹配错误求助

在Python类型注解场景中，调用字符串的lower()方法后，静态类型检查器无法保留原字面量类型，导致传入要求Literal["apple", "orange"]类型参数的函数时触发类型错误。

问题代码示例 #

from typing import Literal

def cart(fruit: Literal["apple", "orange"]):
    print(fruit)

my_fruit = "Apple"
cart(my_fruit.lower())  # 触发类型错误

错误提示 #

Argument of type "LiteralString" cannot be assigned to parameter "fruit" of type "Literal['apple', 'orange']" in function "cart"
  Type "LiteralString" cannot be assigned to type "Literal['apple', 'orange']"
    "LiteralString" cannot be assigned to type "Literal['apple']"
    "LiteralString" cannot be assigned to type "Literal['orange']"

尝试过的无效写法 #

即使将转换结果赋值给变量，类型错误依然存在：

my_fruit = "Apple"
my_fruit_in_lowercase = my_fruit.lower()
cart(my_fruit_in_lowercase)  # 同样触发类型错误

此时my_fruit_in_lowercase被类型检查器推断为LiteralString，但实际我们期望它是Literal['apple']。

解决方案 #

使用typing.cast明确指定转换后的字面量类型，帮助类型检查器识别正确的类型：

from typing import Literal, cast

def cart(fruit: Literal["apple", "orange"]):
    print(fruit)

my_fruit = "Apple"
# 通过cast告知类型检查器转换后的具体字面量类型
cart(cast(Literal["apple"], my_fruit.lower()))

原因说明 #

静态类型检查器无法自动追踪lower()这类字符串方法对字面量类型的转换——因为这些方法在运行时理论上可以返回任意小写字符串，所以默认只能推断为宽泛的LiteralString或str类型，无法关联到原字面量的小写形式。通过cast可以手动为类型检查器提供明确的类型信息。

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