如何让Python函数单参数接收多个指定合法字符串值，有哪些更Pythonic的实现？

原代码的问题 #

原代码功能可以正常运行，但存在几个不符合Python工程规范的可优化点：

• 抛出通用Exception而非专门的异常类型：会导致上层捕获异常时无法区分是参数不合法还是其他运行时错误，违反异常处理的最佳实践
• 用列表存储合法参数做成员校验：列表的in查询时间复杂度为O(n)，数据量小时差异不大，但语义和性能都不如适合做成员校验的集合
• 合法参数硬编码在函数内部：如果后续其他逻辑也需要用到合法参数列表，会出现重复编码，维护成本高

轻量化优化方案（适合简单场景） #

如果只是小范围使用这个参数校验逻辑，不需要跨模块复用，可以用以下更简洁符合Pythonic规范的实现：

# 合法参数抽为全局常量，用集合存储优化查询效率
ACCEPTABLE_PARAMS = {'first', 'second', 'third'}

def test_func(param_1: str) -> None:
    if param_1 not in ACCEPTABLE_PARAMS:
        # 抛出ValueError专门对应参数值不合法的场景
        raise ValueError(f"'{param_1}' is not an acceptable parameter, allowed values: {sorted(ACCEPTABLE_PARAMS)}")
    print(param_1)

枚举类优化方案（适合复杂工程场景） #

如果合法参数需要跨模块复用、或者后续有扩展需求，推荐使用Python标准库的enum.Enum实现，静态检查阶段就能拦截大部分非法传值：

from enum import Enum

# 定义字符串枚举类，集中管理合法参数
class ValidParam(str, Enum):
    FIRST = "first"
    SECOND = "second"
    THIRD = "third"

def test_func(param_1: ValidParam) -> None:
    # 类型提示直接限定为枚举类型，IDE可自动补全，mypy等静态检查工具可提前发现非法传值
    print(param_1.value)

# 如果需要支持传入字符串再做运行时校验，可调整函数实现为：
def test_func(param_1: str) -> None:
    try:
        valid_param = ValidParam(param_1)
    except ValueError:
        allowed = [item.value for item in ValidParam]
        raise ValueError(f"'{param_1}' is not an acceptable parameter, allowed values: {allowed}")
    print(valid_param.value)

优化后的优势 #

• 异常类型更精准，上层可以精准捕获参数校验错误，不会和其他运行异常混淆
• 合法参数管理更集中，后续修改合法值只需要修改常量或枚举类即可，不需要改动函数逻辑
• 枚举方案支持静态类型检查，提前发现错误，IDE补全更友好，避免硬编码字符串的笔误问题

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