如何在Python中创建可复用的Substate类型以移除Literal类型首段？

我有一个包含多个句号分隔字符串的Python Literal 类型，希望自动生成对应的Literal类型，其中所有字符串都移除首个句号分隔的部分，不想每次手动维护这个新类型。

示例代码 #

from typing import Literal

# 句号分隔的字符串字面量
ActiveState = Literal['active.green', 'active.yellow', 'active.red']
State = ActiveState | OtherState | AnotherState  # 希望对多个类型都适用

# 移除首个句号分隔部分，仅处理后续内容
def handle_active_state(substate: Literal['green', 'yellow', 'red']) -> None: ...

尝试过的方法 #

我试过用typing.get_args遍历并修改ActiveState里的值，也试过用types.GenericAlias重新生成Literal，但类型检查器（比如pylance、mypy）会把生成的结果识别为值而非类型别名，没法正常使用：

from typing import Literal, get_args
from types import GenericAlias
ActiveState = Literal['active.green', 'active.yellow', 'active.red']

ActiveSubstate = GenericAlias(Literal, tuple('.'.join(state.split('.')[1:]) for state in get_args(ActiveState)))
# 虽然生成了Literal['green', 'yellow', 'red']，但无法使用

def handle_active_state(substate: ActiveSubstate) -> None: ...
# pylance提示：Illegal type annotation: variable not allowed unless it is a type alias
# mypy也有类似报错

目标 #

我想创建可复用的自动处理类型，比如Substate[ActiveState]这种形式。想问这在Python里是否可行？如果不行，原因是什么？

补充说明 #

这类静态类型分析的功能在TypeScript里是可以实现的，示例如下：

type ActiveState = 'active.green' | 'active.yellow' | 'active.red';
type Substate<T> = T extends `${infer First}.${infer After}` ? After : T;
type ActiveSubstate = Substate<ActiveState>;

const good: ActiveSubstate = 'green';
const bad: ActiveSubstate = 'blue'; // 符合预期：类型“blue”无法赋值给ActiveSubstate

在Python的静态类型系统里，原生不支持TypeScript那种基于模板字符串的类型推导（即条件类型+字符串拆分），这是当前Python类型系统的局限性之一。

可行的替代方案 #

如果你想要实现类似Substate[ActiveState]的复用能力，可以考虑以下几种方式：

1. 代码生成工具（推荐）
   既然静态类型系统原生做不到，你可以用代码生成的方式自动维护Substate类型。写一个简单的脚本，读取原Literal的定义，自动生成拆分后的Literal代码，然后在项目里引入生成的文件。
   示例脚本：

   from typing import Literal, get_args
   
   def generate_substate_type(literal_type, new_type_name):
       args = get_args(literal_type)
       sub_args = tuple('.'.join(s.split('.')[1:]) for s in args)
       return f"{new_type_name} = Literal{sub_args}"
   
   # 使用示例
   ActiveState = Literal['active.green', 'active.yellow', 'active.red']
   print(generate_substate_type(ActiveState, 'ActiveSubstate'))
   # 输出：ActiveSubstate = Literal['green', 'yellow', 'red']
   

   你可以把这个脚本集成到项目的构建流程里，每次原Literal更新时重新生成代码，这样就能避免手动维护。

2. 借助mypy的扩展语法（有限支持）
   mypy支持一些PEP之外的扩展语法，比如TypeVarTuple和Unpack，但这种方式无法直接批量处理Literal的所有成员，实用性有限：

   from typing import Literal, TypeVarTuple, Unpack
   
   Ts = TypeVarTuple('Ts')
   
   # 仅能对单个字面量做拆分，无法直接作用于整个Literal类型
   def split_first_part[T: str](_: T) -> Literal[T.split('.')[1]]: ...
   

3. pydantic场景下的折中方案
   如果你的类型用于pydantic模型，可以通过BeforeValidator在运行时处理字符串拆分，但静态类型检查层面还是需要手动定义Literal，可以结合上面的代码生成工具来解决这个问题。

为什么原生Python类型系统做不到？ #

Python的类型系统是声明式的，而TypeScript的类型系统支持编程式的类型推导（比如条件类型、字符串模板类型）。Python的Literal是静态类型构造器，只能接受固定的字面量参数，类型检查器在静态分析阶段无法执行Python代码（比如split('.')）来推导新的字面量集合——你用GenericAlias生成的对象是运行时的值，而非静态类型别名，所以会被类型检查器拒绝。

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