33 lines
1020 B
Python
33 lines
1020 B
Python
"""Enum backports from standard lib."""
|
|
from __future__ import annotations
|
|
|
|
from enum import Enum
|
|
from typing import Any
|
|
|
|
from typing_extensions import Self
|
|
|
|
|
|
class StrEnum(str, Enum):
|
|
"""Partial backport of Python 3.11's StrEnum for our basic use cases."""
|
|
|
|
def __new__(cls, value: str, *args: Any, **kwargs: Any) -> Self:
|
|
"""Create a new StrEnum instance."""
|
|
if not isinstance(value, str):
|
|
raise TypeError(f"{value!r} is not a string")
|
|
return super().__new__(cls, value, *args, **kwargs)
|
|
|
|
def __str__(self) -> str:
|
|
"""Return self.value."""
|
|
return str(self.value)
|
|
|
|
@staticmethod
|
|
def _generate_next_value_(
|
|
name: str, start: int, count: int, last_values: list[Any]
|
|
) -> Any:
|
|
"""Make `auto()` explicitly unsupported.
|
|
|
|
We may revisit this when it's very clear that Python 3.11's
|
|
`StrEnum.auto()` behavior will no longer change.
|
|
"""
|
|
raise TypeError("auto() is not supported by this implementation")
|