Python中如何约束方法仅返回特定集合中的值？

要约束方法返回值只能是特定集合中的值，Python里有几种简洁且符合Python风格的方案：

1. 静态类型约束：使用Literal类型注解 #

如果只需要在开发阶段通过静态检查工具（比如mypy、pyright）来约束，可以用Literal类型。它能明确指定方法允许返回的具体值，不符合的话静态检查工具会直接报错，属于编译前的约束。

from typing import Literal

class Rule:
    def evaluate(self, user_id: int) -> Literal["PASS", "FAIL", "ERROR"]:
        # 示例业务逻辑
        if user_id > 0:
            return "PASS"
        elif user_id == 0:
            return "FAIL"
        else:
            return "ERROR"

注：Literal在Python 3.8及以上版本可直接从typing模块导入，3.8以下版本需要安装typing_extensions库后导入。

2. 运行时验证：确保实际执行时的合法性 #

如果需要在程序运行时也强制约束返回值，可以在方法内添加检查逻辑，或者用装饰器复用验证逻辑，避免代码重复。

方法内直接检查 #

from typing import Literal

class Rule:
    def evaluate(self, user_id: int) -> Literal["PASS", "FAIL", "ERROR"]:
        # 业务逻辑计算结果
        if user_id > 100:
            result = "PASS"
        elif user_id <= 100 and user_id > 0:
            result = "FAIL"
        else:
            result = "ERROR"
        
        # 验证返回值合法性
        allowed_values = {"PASS", "FAIL", "ERROR"}
        if result not in allowed_values:
            raise ValueError(f"返回值必须是{allowed_values}中的一项，当前返回: {result}")
        return result

装饰器复用验证逻辑 #

如果多个方法都需要类似约束，用装饰器能大幅减少重复代码：

from typing import Literal, Callable

def enforce_return(allowed_values: set):
    def decorator(func: Callable):
        def wrapper(*args, **kwargs):
            result = func(*args, **kwargs)
            if result not in allowed_values:
                raise ValueError(f"返回值非法，允许值: {allowed_values}，当前值: {result}")
            return result
        return wrapper
    return decorator

class Rule:
    @enforce_return({"PASS", "FAIL", "ERROR"})
    def evaluate(self, user_id: int) -> Literal["PASS", "FAIL", "ERROR"]:
        # 业务逻辑
        ...

3. 结构化约束：使用枚举（Enum）类型 #

如果希望返回值更结构化，同时兼顾静态检查和运行时约束，可以用枚举类定义允许的返回值。方法返回枚举实例，天然限制了值的范围，还能通过.value属性获取对应的字符串值。

from enum import Enum

class ResultStatus(Enum):
    PASS = "PASS"
    FAIL = "FAIL"
    ERROR = "ERROR"

class Rule:
    def evaluate(self, user_id: int) -> ResultStatus:
        if user_id > 0:
            return ResultStatus.PASS
        elif user_id == 0:
            return ResultStatus.FAIL
        else:
            return ResultStatus.ERROR

# 使用示例
rule = Rule()
status = rule.evaluate(10)
print(status.value)  # 输出 "PASS"

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