Поведение Other/Default-name на python Enum

Как говорится, если вы хотите что-то сделать... Я создал следующий подкласс перечисления (я не добавлял новых членов, так что это разрешено):

class DefaultNameEnum(Enum):
    """Support for "Other"/default-name values"""

    @classmethod
    def _missing_(cls, value):
        possible_member = cls._value2member_map_.get(value, None)

        if possible_member is None:
            possible_member = cls._create_pseudo_member_(value)

        return possible_member

    @classmethod
    def _create_pseudo_member_(cls, value):
        """
        Create a default-name member.
        """

        default_member = cls._value2member_map_.get(None, None)

        if default_member is None:
            raise ValueError("%r is not a valid %s" % (value, cls.__name__))

        # construct a singleton enum pseudo-member
        other_member = object.__new__(cls)
        other_member._name_ = default_member._name_
        other_member._value_ = value

        # use setdefault in case another thread already created a composite
        # with this value
        other_member = cls._value2member_map_.setdefault(value, other_member)

        return other_member

    def __eq__(self, other):
        """Overrides the default implementation"""
        if isinstance(other, DefaultNameEnum):
            return self._name_ == other._name_
        return False

    def __ne__(self, other):
        return not self == other

Это основано на подклассе перечисления Flag. Его использование на самом деле довольно простое — просто определите как None любое имя, которое вы хотите использовать по умолчанию. Лучше всего это проиллюстрировать на примере — рассмотрим класс:

class ABC(DefaultNameEnum):
    A = 1
    B = 2
    C = 3

    Other = None

Затем следующие консольные вызовы дадут:

>>> print([repr(mem) for mem in ABC])    
... ['<ABC.A: 1>', '<ABC.B: 2>', '<ABC.C: 3>', '<ABC.Other: None>']
>>> ABC(123)
... '<ABC.Other: 123>'
>>> ABC(1) == ABC(2)
... False
>>> ABC(123) == ABC.Other
... True
>>> ABC(123) == ABC(1374)
... True

Если вы хотите взять эту реализацию и использовать ее, обратите внимание на следующие моменты:

1. Поведение в последней строке может быть желательным, а может и нет - в зависимости от вашего использования. Если это нежелательное использование, просто измените метод __eq__, чтобы сравнивать имена, когда self._value_ или other._value_ являются None.

2. Если вы используете этот класс, для удобства представления вы можете захотеть, чтобы значение по умолчанию __repr__ выводило '<ABC.Other>', а не '<ABC.Other: None>' в случае значения None. Этого легко добиться, переопределив метод __repr__.

3. Если вы не определите член по умолчанию, класс вызовет исключение при вызове его для неизвестного значения (как и любой подкласс Enum).

Я также хочу отметить, что в приведенной выше реализации я бы предпочел использовать элемент-разделитель, такой как _default_name_ или _default_value_member_, а не назначать None, но, увы, модуль enum не позволяет определять новый элемент-разделитель для Enum подклассов.

Это похоже на то, что вы просите:

>>> class Yep(enum.IntFlag):
...  A = 1
...  B = 2
... 
>>> Yep(5)
<Yep.4|A: 5>

Таким образом, Yep(5) действителен, но не будет Yep.Other волшебного члена.