如何在Python中动态定义字符串排列的Literal Union类型

给定字符串列表：

strings = ['a','b','c']

我想要声明一个由该列表所有有序对拼接字符串组成的Literal联合类型，硬编码的写法如下：

CustomType = Literal[
  'ab', 'ac', 'aa',
  'ba', 'bb', 'bc',
  'ca', 'cb', 'cc'
]

如何无需手动编写每个组合，动态定义CustomType？我尝试了以下写法：

from itertools import permutations
from typing import Literal

CustomType = Literal[*permutations(strings)]

但触发了Pylance的两个错误：

Unpack operator in subscript requires Python 3.11 or newer 
Unpacked arguments cannot be used in type argument lists 

我的目标是让Pylance能静态检测出传入非集合内字符串的情况，例如：

def f(string: CustomType):
    ...

f('foo') # Pylance应报错
f('ba') # Pylance应允许

方法1：静态生成组合+Literal构造 #

类型检查器需要在静态分析阶段确定Literal的成员，所以要先预先生成所有拼接后的字符串，再传递给Literal：

1. 首先生成所有有序对的拼接结果（permutations默认不包含同元素组合，用双重循环可覆盖所有场景）：

from typing import Literal

strings = ['a', 'b', 'c']
# 生成所有x+y的组合，包含aa、bb这类同元素对
all_pairs = [f"{x}{y}" for x in strings for y in strings]

2. 构造CustomType：
   • 若使用Python 3.11+，直接用解包语法即可，Pylance能正确识别：

     CustomType = Literal[*all_pairs]
     

   • 若Python版本低于3.11，或Pylance仍报错，改用Literal.__getitem__方式，兼容Python 3.8+：

     CustomType = Literal.__getitem__(tuple(all_pairs))
     

测试效果：

def f(string: CustomType):
    pass

f('ba') # 正常通过检查
f('foo') # Pylance会报错：Literal['foo']无法赋值给CustomType类型的参数

方法2：TypeVar+运行时校验（不推荐） #

如果上述方法无法使用，也可以用TypeVar加运行时检查，但这种方式无法实现静态校验，只能在运行时抛出错误：

from typing import TypeVar

strings = ['a', 'b', 'c']
valid_pairs = {f"{x}{y}" for x in strings for y in strings}
T = TypeVar('T', bound=str)

def f(string: T) -> None:
    if string not in valid_pairs:
        raise ValueError(f"无效字符串：{string}")

f('ba') # 正常运行
f('foo') # 运行时抛出ValueError，但Pylance不会静态报错

这种方式的静态检查能力远弱于Literal，仅作为备选方案。

注意事项 #

• 必须确保all_pairs是模块加载时就计算出的静态值，不能在函数内部动态生成，否则类型检查器无法识别。
• 如果使用Python 3.11+的解包语法，要确保Pylance的Python版本设置与实际环境一致。

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