◐ Shell
clean mode source ↗

gh-102558: [Enum] better handling of non-Enum EnumType classes by ethanfurman · Pull Request #103060 · python/cpython

Expand Up @@ -923,7 +923,7 @@ def _convert_(cls, name, module, filter, source=None, *, boundary=None, as_globa def _check_for_existing_members_(mcls, class_name, bases): for chain in bases: for base in chain.__mro__: if issubclass(base, Enum) and base._member_names_: if isinstance(base, EnumType) and base._member_names_: raise TypeError( "<enum %r> cannot extend %r" % (class_name, base) Expand All @@ -942,7 +942,7 @@ def _get_mixins_(mcls, class_name, bases): # ensure final parent class is an Enum derivative, find any concrete # data type, and check that Enum has no members first_enum = bases[-1] if not issubclass(first_enum, Enum): if not isinstance(first_enum, EnumType): raise TypeError("new enumerations should be created as " "`EnumName([mixin_type, ...] [data_type,] enum_type)`") member_type = mcls._find_data_type_(class_name, bases) or object Expand All @@ -954,7 +954,7 @@ def _find_data_repr_(mcls, class_name, bases): for base in chain.__mro__: if base is object: continue elif issubclass(base, Enum): elif isinstance(base, EnumType): # if we hit an Enum, use it's _value_repr_ return base._value_repr_ elif '__repr__' in base.__dict__: Expand All @@ -980,12 +980,12 @@ def _find_data_type_(mcls, class_name, bases): base_chain.add(base) if base is object: continue elif issubclass(base, Enum): elif isinstance(base, EnumType): if base._member_type_ is not object: data_types.add(base._member_type_) break elif '__new__' in base.__dict__ or '__init__' in base.__dict__: if issubclass(base, Enum): if isinstance(base, EnumType): continue data_types.add(candidate or base) break Expand Down Expand Up @@ -1186,8 +1186,6 @@ def _missing_(cls, value): return None
def __repr__(self): if not isinstance(self, Enum): return repr(self) v_repr = self.__class__._value_repr_ or repr return "<%s.%s: %s>" % (self.__class__.__name__, self._name_, v_repr(self._value_))
Expand Down