Есть ли более элегантная перезапись для этой реализации Python Enum value_of? ⇐ Python

[Anonymous]

Я хотел бы получить реализацию value_of для StrEnum (Python 3.9.x). Например:

Код: Выделить всё

from enum import Enum

class StrEnum(str, Enum):
"""Enum with str values"""
pass

class BaseStrEnum(StrEnum):
"""Base Enum"""
@classmethod
def value_of(cls, value):
try:
return cls[value]
except KeyError:
try:
return cls(value)
except ValueError:
return None

а затем можно использовать его следующим образом:

Код: Выделить всё

class Fruits(BaseStrEnum):
BANANA = "Banana"
PEA = "Pea"
APPLE = "Apple"

print(Fruits.value_of('BANANA'))
print(Fruits.value_of('Banana'))

просто вложенная попытка-кроме не выглядит потрясающе, есть ли лучший, более идиоматический перезапись?

Подробнее здесь: https://stackoverflow.com/questions/793 ... ementation