| 1234567891011121314151617181920212223 |
- from enum import IntEnum
- class ParamEnum(IntEnum):
- """Wraps IntEnum to provide validation of a requested item.
- Intended for enums used for function parameters.
- Use enum.get_value(item) for this behavior instead of builtin enum[item].
- """
- @classmethod
- def get_value(cls, item):
- """Validate item and raise a ValueError with valid options if not present."""
- try:
- return cls[item].value
- except KeyError:
- valid_options = {e.name for e in cls}
- raise ValueError(
- "'{}' is not a valid option, must be one of '{}'".format(
- item, "', '".join(valid_options)
- )
- )
|
