summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorEthan Furman <ethan@stoneleaf.us>2021-01-07 21:55:59 (GMT)
committerGitHub <noreply@github.com>2021-01-07 21:55:59 (GMT)
commit9ab4dd452287169f08a8cf4d4c68c2139f8de714 (patch)
treeddca1c5d74936757ff840e24603272b9f2e93f69
parent8c3914aef47e6e5a31b48a0b1f165ec3f4dc4c98 (diff)
downloadcpython-9ab4dd452287169f08a8cf4d4c68c2139f8de714.zip
cpython-9ab4dd452287169f08a8cf4d4c68c2139f8de714.tar.gz
cpython-9ab4dd452287169f08a8cf4d4c68c2139f8de714.tar.bz2
[3.9] bpo-42851: [Enum] remove brittle __init_subclass__ support (GH-24154) (GH-24155)
Solution to support calls to `__init_subclass__` with members defined is too brittle and breaks with certain mixins.. (cherry picked from commit a581a868d97f649aedf868a1d27865a10925c73a)
-rw-r--r--Lib/enum.py31
-rw-r--r--Lib/test/test_enum.py46
-rw-r--r--Misc/NEWS.d/next/Library/2021-01-07-11-44-22.bpo-42851.uyQFyd.rst1
3 files changed, 2 insertions, 76 deletions
diff --git a/Lib/enum.py b/Lib/enum.py
index 1c22380..1fddb1c 100644
--- a/Lib/enum.py
+++ b/Lib/enum.py
@@ -9,14 +9,6 @@ __all__ = [
]
-class _NoInitSubclass:
- """
- temporary base class to suppress calling __init_subclass__
- """
- @classmethod
- def __init_subclass__(cls, **kwds):
- pass
-
def _is_descriptor(obj):
"""
Returns True if obj is a descriptor, False otherwise.
@@ -219,22 +211,7 @@ class EnumMeta(type):
if '__doc__' not in classdict:
classdict['__doc__'] = 'An enumeration.'
- # postpone calling __init_subclass__
- if '__init_subclass__' in classdict and classdict['__init_subclass__'] is None:
- raise TypeError('%s.__init_subclass__ cannot be None')
- # remove current __init_subclass__ so previous one can be found with getattr
- new_init_subclass = classdict.pop('__init_subclass__', None)
- # create our new Enum type
- if bases:
- bases = (_NoInitSubclass, ) + bases
- enum_class = super().__new__(metacls, cls, bases, classdict, **kwds)
- enum_class.__bases__ = enum_class.__bases__[1:] #or (object, )
- else:
- enum_class = super().__new__(metacls, cls, bases, classdict, **kwds)
- old_init_subclass = getattr(enum_class, '__init_subclass__', None)
- # and restore the new one (if there was one)
- if new_init_subclass is not None:
- enum_class.__init_subclass__ = classmethod(new_init_subclass)
+ enum_class = super().__new__(metacls, cls, bases, classdict, **kwds)
enum_class._member_names_ = [] # names in definition order
enum_class._member_map_ = {} # name->value map
enum_class._member_type_ = member_type
@@ -346,9 +323,6 @@ class EnumMeta(type):
if _order_ != enum_class._member_names_:
raise TypeError('member order does not match _order_')
- # finally, call parents' __init_subclass__
- if Enum is not None and old_init_subclass is not None:
- old_init_subclass(**kwds)
return enum_class
def __bool__(self):
@@ -726,9 +700,6 @@ class Enum(metaclass=EnumMeta):
else:
return start
- def __init_subclass__(cls, **kwds):
- super().__init_subclass__(**kwds)
-
@classmethod
def _missing_(cls, value):
return None
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index f6db731..4e22986 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -2065,52 +2065,6 @@ class TestEnum(unittest.TestCase):
except ValueError:
pass
- def test_init_subclass_calling(self):
- class MyEnum(Enum):
- def __init_subclass__(cls, **kwds):
- super(MyEnum, cls).__init_subclass__(**kwds)
- self.assertFalse(cls.__dict__.get('_test', False))
- cls._test1 = 'MyEnum'
- #
- class TheirEnum(MyEnum):
- def __init_subclass__(cls, **kwds):
- super().__init_subclass__(**kwds)
- cls._test2 = 'TheirEnum'
- class WhoseEnum(TheirEnum):
- def __init_subclass__(cls, **kwds):
- pass
- class NoEnum(WhoseEnum):
- ONE = 1
- self.assertEqual(TheirEnum.__dict__['_test1'], 'MyEnum')
- self.assertEqual(WhoseEnum.__dict__['_test1'], 'MyEnum')
- self.assertEqual(WhoseEnum.__dict__['_test2'], 'TheirEnum')
- self.assertFalse(NoEnum.__dict__.get('_test1', False))
- self.assertFalse(NoEnum.__dict__.get('_test2', False))
- #
- class OurEnum(MyEnum):
- def __init_subclass__(cls, **kwds):
- cls._test2 = 'OurEnum'
- class WhereEnum(OurEnum):
- def __init_subclass__(cls, **kwds):
- pass
- class NeverEnum(WhereEnum):
- ONE = 'one'
- self.assertEqual(OurEnum.__dict__['_test1'], 'MyEnum')
- self.assertFalse(WhereEnum.__dict__.get('_test1', False))
- self.assertEqual(WhereEnum.__dict__['_test2'], 'OurEnum')
- self.assertFalse(NeverEnum.__dict__.get('_test1', False))
- self.assertFalse(NeverEnum.__dict__.get('_test2', False))
-
- def test_init_subclass_parameter(self):
- class multiEnum(Enum):
- def __init_subclass__(cls, multi):
- for member in cls:
- member._as_parameter_ = multi * member.value
- class E(multiEnum, multi=3):
- A = 1
- B = 2
- self.assertEqual(E.A._as_parameter_, 3)
- self.assertEqual(E.B._as_parameter_, 6)
class TestOrder(unittest.TestCase):
diff --git a/Misc/NEWS.d/next/Library/2021-01-07-11-44-22.bpo-42851.uyQFyd.rst b/Misc/NEWS.d/next/Library/2021-01-07-11-44-22.bpo-42851.uyQFyd.rst
new file mode 100644
index 0000000..9272835
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2021-01-07-11-44-22.bpo-42851.uyQFyd.rst
@@ -0,0 +1 @@
+remove __init_subclass__ support for Enum members