Какой-нибудь причудливый способ раскрутить классы Enum в python?
В настоящее время я кодирую пакет, который содержит несколько классов Enum, и хотел бы использовать символические имена непосредственно из имени пакета, а не из самих классов, как это делает OpenCV. Можно ли в любом случае определить классы, но по-прежнему получать доступ к их символическим именам непосредственно из пакета, или я должен напрямую жестко закодировать их в __init__.py пакета?
На данный момент я нахожусь в таком состоянии:
from enum import Enum
class enum_class1(Enum):
class1_sym1: int = 0
class1_sym2: int = 1
class1_sym3: int = 2
class enum_class2(Enum):
class2_sym1: int = 0
class2_sym2: int = 1
class2_sym3: int = 2
Чем я сейчас занимаюсь:
import mypackage as p
state = p.enum_class2.class_2_sim1
Что я хотел бы сделать:
import mypackage as p
state = p.class_2_sim1
Ответы 4
Вы можете просто определить глобальные переменные со значениями перечисления:
from enum import Enum
class enum_class1(Enum):
class1_sym1: int = 0
class1_sym2: int = 1
class1_sym3: int = 2
class1_sym1: int = enum_class1.class1_sym1
class1_sym2: int = enum_class1.class1_sym2
class1_sym3: int = enum_class1.class1_sym3
class enum_class2(Enum):
class2_sym1: int = 0
class2_sym2: int = 1
class2_sym3: int = 2
class2_sym1: int = enum_class2.class2_sym1
class2_sym2: int = enum_class2.class2_sym2
class2_sym3: int = enum_class2.class2_sym3
Рискуя поставить под угрозу статический анализ, вы можете автоматически определить глобальные переменные для каждого члена Enum (как это предлагается в ответе Бармара ), воспользовавшись отображением Enum.__members__. Данный
class enum_class1(Enum):
class1_sym1: int = 0
class1_sym2: int = 1
class1_sym3: int = 2
у нас есть
>>> enum_class1.__members__
mappingproxy({'class1_sym1': <enum_class1.class1_sym1: 0>,
'class1_sym2': <enum_class1.class1_sym2: 1>,
'class1_sym3': <enum_class1.class1_sym3: 2>})
Это сопоставление можно использовать для заполнения глобальных переменных как
globals().update(enum_class1.__members__)
Так,
class enum_class1(Enum):
class1_sym1: int = 0
class1_sym2: int = 1
class1_sym3: int = 2
globals().update(enum_class1.__members__)
print(repr(class1_sym1))
работает как положено и будет печатать
<enum_class1.class1_sym1: 0>
Более динамичный подход, использующий __getattr__ в модуле, будет выглядеть примерно так:
from enum import Enum
class enum_class1(Enum):
class1_sym1: int = 0
class1_sym2: int = 1
class1_sym3: int = 2
class enum_class2(Enum):
class2_sym1: int = 0
class2_sym2: int = 1
class2_sym3: int = 2
def __getattr__(target):
target_cls, target_attr = target.rsplit('_', 1)
if (cls := globals().get(f'enum_{target_cls}')) is not None:
return getattr(cls, f'{target_cls}_{target_attr}')
In [1]: import mypackage as p
In [2]: p.class2_sym1
Out[2]: <enum_class2.class2_sym1: 0>
У этой реализации есть некоторые интересные побочные эффекты, а именно p.doesnotexist вызовет ValueError, а p._doesnotexist будет оцениваться как None, в то время как оба обычно должны вызывать AttributeError. Также стоит отметить, что это также работает только тогда, когда атрибуты просматриваются в объекте модуля, и не будет работать, когда выполняется прямой поиск сопоставления глобальных переменных. Так, например, эти «глобальные переменные» не будут доступны внутри функций в том же модуле.
Если все ваши члены перечисления имеют разные имена, вы можете сделать:
from enum import Enum, global_enum
@global_enum
class enum_class1(Enum):
class1_sym1: int = 0
class1_sym2: int = 1
class1_sym3: int = 2
Теперь вы можете получить доступ из модуля, а представление обновлено, чтобы не отображать класс:
>>> import mypackage as p
>>> state = p.class1_sym1
>>> state
mypackage.class1_sym1
Вы можете, конечно, написать свой собственный __repr__, если вам не нравится стандартный.
Кроме того, если вам нужно успокоить линтер, добавьте
class1_sym1 = class1_sym2 = class1_sym3 = None
Раскрытие информации: я являюсь автором библиотеки Python stdlib Enum , бэкпорта enum34 и библиотеки Advanced Enumeration (aenum).
Я предполагаю, что это просто происходит от копирования, но аннотации неверны,
class1_sym1: int = enum_class1.class_sym1->class1_sym1: enum_class1 = enum_class1.class_sym1или просто опущены, так как тип должен быть выводимым.