Python：如何实现无需更新的类变量匹配校验方法？

针对你的需求，这里有几种无需修改is_valid方法就能新增支持格式的实现方式：

方案一：自动过滤类中的大写常量 #

利用类的属性列表，筛选出符合常量命名规范（全大写）的类变量，这样新增全大写的格式变量时，is_valid会自动识别：

class DataFormat:
    CP2K_OUTPUT_DIR = 'cp2k/output_dir'
    VASP_OUTPUT_DIR = 'vasp/output_dir'
    LAMMPS_OUTPUT_DIR = 'lammps/output_dir'

    CP2K_OUTPUT = 'cp2k/output'
    VASP_OUTPUT = 'vasp/xml'
    EXTXYZ = 'extxyz'

    @classmethod
    def is_valid(cls, format: str):
        # 筛选出类中全大写的属性值，排除特殊属性和方法
        valid_formats = [
            value for name, value in cls.__dict__.items()
            if name.isupper() and not name.startswith('__')
        ]
        return format in valid_formats

注意：如果类中存在其他全大写的非格式常量，这种方式会误判，适合所有全大写变量都是格式标识的场景。

方案二：集中管理有效值集合 #

专门定义一个类变量存储所有有效格式，其他常量指向集合中的值，新增格式时只需在集合和常量中添加（常量可选，仅为提升代码可读性）：

class DataFormat:
    VALID_FORMATS = {
        'cp2k/output_dir',
        'vasp/output_dir',
        'lammps/output_dir',
        'cp2k/output',
        'vasp/xml',
        'extxyz'
    }

    # 可选：定义常量方便代码中引用
    CP2K_OUTPUT_DIR = 'cp2k/output_dir'
    VASP_OUTPUT_DIR = 'vasp/output_dir'
    LAMMPS_OUTPUT_DIR = 'lammps/output_dir'
    CP2K_OUTPUT = 'cp2k/output'
    VASP_OUTPUT = 'vasp/xml'
    EXTXYZ = 'extxyz'

    @classmethod
    def is_valid(cls, format: str):
        return format in cls.VALID_FORMATS

这种方式更可控，不会误包含其他常量，新增时只需要往VALID_FORMATS里加值即可。

方案三：使用枚举类（最规范） #

Python的enum模块天生适合这种常量集合场景，枚举成员本身就是有效值，检查逻辑更简洁，新增格式只需添加枚举成员：

from enum import Enum

class DataFormat(Enum):
    CP2K_OUTPUT_DIR = 'cp2k/output_dir'
    VASP_OUTPUT_DIR = 'vasp/output_dir'
    LAMMPS_OUTPUT_DIR = 'lammps/output_dir'
    CP2K_OUTPUT = 'cp2k/output'
    VASP_OUTPUT = 'vasp/xml'
    EXTXYZ = 'extxyz'

    @classmethod
    def is_valid(cls, format: str):
        return format in [member.value for member in cls]

甚至可以进一步简化检查逻辑：

@classmethod
def is_valid(cls, format: str):
    try:
        cls(format)
        return True
    except ValueError:
        return False

这种方式不仅自动管理有效值，还能避免魔法字符串，代码可读性和维护性都更强。

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