如何使用typing模块定义仅包含特定字符串的类型？

这个问题我刚好熟！要实现你说的这种「只能取特定字符串值」的类型，Python的typing模块里有专门的工具，下面给你两种常用方案：

方案1：用Literal类型（静态类型检查） #

如果你的主要需求是在开发阶段通过静态类型检查工具（比如mypy、pyright）避免传入错误值，那么Literal类型是最合适的选择。它是Python 3.8及以上版本typing模块自带的，3.7及以下版本可以通过typing_extensions包引入。

代码示例 #

# Python 3.8+ 直接从typing导入
from typing import Literal

# 定义只能取指定字符串的类型
CondOperator = Literal['=', '>', '<', '>=', '<=', '<>', '!=']

# 使用这个类型的函数示例
def apply_condition(op: CondOperator, val1: int, val2: int) -> bool:
    match op:
        case '=':
            return val1 == val2
        case '>':
            return val1 > val2
        case '<':
            return val1 < val2
        case '>=':
            return val1 >= val2
        case '<=':
            return val1 <= val2
        case '<>' | '!=':
            return val1 != val2
    return False

# 合法调用（静态检查通过）
apply_condition('>=', 10, 5)

# 非法调用（静态检查工具会报错，比如mypy会提示参数类型不兼容）
# apply_condition('invalid', 10, 5)

注意：Literal只在静态类型检查阶段生效，运行时如果传入非法值，Python不会主动拦截，需要自己额外处理。

方案2：用枚举类（静态检查+运行时约束） #

如果需要在运行时也强制限制取值范围，可以结合Enum类来实现，它既能被静态类型检查工具识别，又能在运行时拦截非法值。

代码示例 #

from enum import Enum

# 定义继承自str和Enum的枚举类，这样既可以用枚举成员，也可以直接用对应的字符串
class CondOperator(str, Enum):
    EQUAL = '='
    GREATER_THAN = '>'
    LESS_THAN = '<'
    GREATER_OR_EQUAL = '>='
    LESS_OR_EQUAL = '<='
    NOT_EQUAL_1 = '<>'
    NOT_EQUAL_2 = '!='

# 使用示例
def apply_condition(op: CondOperator, val1: int, val2: int) -> bool:
    match op:
        case CondOperator.EQUAL:
            return val1 == val2
        case CondOperator.GREATER_THAN:
            return val1 > val2
        # 其他操作符逻辑...
        case CondOperator.NOT_EQUAL_1 | CondOperator.NOT_EQUAL_2:
            return val1 != val2
    return False

# 两种合法调用方式
apply_condition(CondOperator.GREATER_THAN, 10, 5)
apply_condition('>', 10, 5)  # 因为继承了str，所以直接传字符串也能被识别

# 非法调用（运行时会报错：ValueError: 'invalid' is not a valid CondOperator）
# apply_condition('invalid', 10, 5)

这种方案兼顾了开发阶段的类型提示和运行时的参数校验，适合对参数合法性要求较高的场景。

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