Python中如何将类与字符串标签关联并提取定义Literal类型？

你遇到的问题是：想通过类的label属性定义Literal类型，但类型检查器（比如Mypy）不把类属性当成常量，导致报错Parameter 1 of Literal[...] is invalid [valid-type]。下面给你几个可行的解决思路：

方法1：先定义常量字符串，再关联到类 #

先把标签字符串定义成独立的常量，让类引用这些常量，同时用常量来定义Literal类型。类型检查器会识别这些独立常量：

from typing import Literal

# 定义常量标签
APPLE_LABEL = "apple"
MANGO_LABEL = "mango"

class Apple:
    label = APPLE_LABEL

class Mango:
    label = MANGO_LABEL

# 用常量生成Literal类型
FruitName = Literal[APPLE_LABEL, MANGO_LABEL]

方法2：用Final标记类属性为常量 #

使用typing.Final标记类的label属性，明确告诉类型检查器这个属性的值不会被修改，属于常量：

from typing import Literal, Final

class Apple:
    label: Final[str] = "apple"

class Mango:
    label: Final[str] = "mango"

# 直接用标记为Final的类属性定义Literal
FruitName = Literal[Apple.label, Mango.label]

这个方法不需要额外定义独立常量，更贴合你原本的代码结构。

方法3：用枚举类统一管理标签 #

如果需要更规范的标签管理，可以用枚举类来绑定标签和类，同时枚举成员的值可以直接用于Literal类型：

from enum import Enum
from typing import Literal

class FruitLabel(Enum):
    APPLE = "apple"
    MANGO = "mango"

class Apple:
    label = FruitLabel.APPLE.value

class Mango:
    label = FruitLabel.MANGO.value

# 用枚举值定义Literal类型
FruitName = Literal[FruitLabel.APPLE.value, FruitLabel.MANGO.value]

这种方式还能避免标签字符串写错，同时方便后续扩展更多水果类型。

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