mirror of
https://github.com/home-assistant/core
synced 2024-09-06 10:29:55 +02:00
29 lines
702 B
Python
29 lines
702 B
Python
"""Helpers for working with enums."""
|
|
from collections.abc import Callable
|
|
import contextlib
|
|
from enum import Enum
|
|
from typing import TYPE_CHECKING, Any, TypeVar
|
|
|
|
# https://github.com/python/mypy/issues/5107
|
|
if TYPE_CHECKING:
|
|
_LruCacheT = TypeVar("_LruCacheT", bound=Callable)
|
|
|
|
def lru_cache(func: _LruCacheT) -> _LruCacheT:
|
|
"""Stub for lru_cache."""
|
|
|
|
else:
|
|
from functools import lru_cache
|
|
|
|
_EnumT = TypeVar("_EnumT", bound=Enum)
|
|
|
|
|
|
@lru_cache
|
|
def try_parse_enum(cls: type[_EnumT], value: Any) -> _EnumT | None:
|
|
"""Try to parse the value into an Enum.
|
|
|
|
Return None if parsing fails.
|
|
"""
|
|
with contextlib.suppress(ValueError):
|
|
return cls(value)
|
|
return None
|