如何在Python中基于数据类常量实现动态类型提示并优化常量结构？

嗨，我完全懂你的痛点——既要用集中式的常量避免到处硬编码，又不想在类型提示里重复写字面量，还想把不同类别的常量拆分得更清晰。咱们一步步来解决这个问题：

1. 先优化你的常量结构 #

你想把STATUS、SI_UNIT拆成独立模块的思路非常棒！其实Python的Enum比冻结数据类更适合存这类枚举型常量，它自带类型安全，而且类型检查器对Enum的支持更完善。比如：

from enum import Enum

class Status(Enum):
    SUCCESS = 200
    ERROR = 400

class SIUnit(Enum):
    MASS = "kg"
    LENGTH = "m"

如果你坚持想用数据类的话，也可以按你的想法拆分，结构会更清晰：

from dataclasses import dataclass

@dataclass(frozen=True)
class STATUS:
    SUCCESS = 200
    ERROR = 400

@dataclass(frozen=True)
class SI_UNIT:
    MASS = "kg"
    LENGTH = "m"

@dataclass(frozen=True)
class CONSTANTS:
    STATUS = STATUS
    SI_UNIT = SI_UNIT

2. 动态生成Literal类型，解决重复代码问题 #

你之前遇到的Pylance错误，本质是类型检查器需要在静态分析时就知道Literal的具体值，不能是运行时变量。针对不同的常量结构，有两种解决方案：

方案一：基于Enum的动态类型生成 #

Enum天生自带成员遍历的能力，我们可以直接用它生成对应的Literal类型别名，而且以后加新成员时类型会自动更新：

from typing import Literal

# 生成包含所有Status成员值的Literal类型
StatusValue = Literal[tuple(s.value for s in Status)]
# 生成包含所有SIUnit成员值的Literal类型
SIUnitValue = Literal[tuple(u.value for u in SIUnit)]

然后给函数加类型提示就很简单了：

def func() -> StatusValue:
    something = True
    if something:
        return Status.SUCCESS.value
    else:
        return Status.ERROR.value

方案二：基于数据类的动态类型生成 #

如果你用的是拆分后的冻结数据类，可以借助dataclasses.fields来提取所有字段值，生成动态的Literal类型：

from typing import Literal
from dataclasses import fields

# 提取STATUS类的所有字段值，生成Literal类型
StatusValue = Literal[tuple(getattr(CONSTANTS.STATUS, f.name) for f in fields(CONSTANTS.STATUS))]
# 提取SI_UNIT类的所有字段值，生成Literal类型
SIUnitValue = Literal[tuple(getattr(CONSTANTS.SI_UNIT, f.name) for f in fields(CONSTANTS.SI_UNIT))]

同样，以后给STATUS或SI_UNIT加新常量时，StatusValue和SIUnitValue会自动包含新值，不用手动修改类型提示。

3. 额外小技巧：直接返回Enum成员（如果场景允许） #

如果你的代码可以接受函数返回Enum成员而不是原始的数值/字符串，那类型提示会更简单直接，而且可读性更高：

def func() -> Status:
    something = True
    if something:
        return Status.SUCCESS
    else:
        return Status.ERROR

这种情况下，类型检查器会自动验证返回值的合法性，完全不用额外定义Literal类型。

总结一下：优先用Enum来定义这类常量集合，既能结构化组织，又能轻松实现动态类型提示，彻底避免重复代码。如果一定要用数据类，也可以通过dataclasses.fields来动态生成类型。

备注：内容来源于stack exchange，提问作者Thomas