]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
[Enum] EnumMeta.__prepare__ now accepts **kwds (#23917)
authorEthan Furman <ethan@stoneleaf.us>
Thu, 24 Dec 2020 18:05:02 +0000 (10:05 -0800)
committerGitHub <noreply@github.com>
Thu, 24 Dec 2020 18:05:02 +0000 (10:05 -0800)
Lib/enum.py
Lib/test/test_enum.py
Misc/NEWS.d/next/Library/2020-12-23-19-43-06.bpo-42727.WH3ODh.rst [new file with mode: 0644]

index 0070ebe136353a34679660152e38d11ac1172d26..36a9a400a116204061581d9a2a3430328f29e0e5 100644 (file)
@@ -178,7 +178,7 @@ class EnumMeta(type):
     Metaclass for Enum
     """
     @classmethod
-    def __prepare__(metacls, cls, bases):
+    def __prepare__(metacls, cls, bases, **kwds):
         # check that previous enum members do not exist
         metacls._check_for_existing_members(cls, bases)
         # create the namespace dict
index a83241cbb577a438176eda7fc62c084c9365af3e..19b4fa5464ce7d1df932e12d91b6e7199bc38587 100644 (file)
@@ -2119,7 +2119,7 @@ class TestEnum(unittest.TestCase):
                 one = '1'
                 two = b'2', 'ascii', 9
                 
-    def test_init_subclass(self):
+    def test_init_subclass_calling(self):
         class MyEnum(Enum):
             def __init_subclass__(cls, **kwds):
                 super(MyEnum, cls).__init_subclass__(**kwds)
@@ -2155,6 +2155,16 @@ class TestEnum(unittest.TestCase):
         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)
 
     @unittest.skipUnless(
             sys.version_info[:2] == (3, 9),
diff --git a/Misc/NEWS.d/next/Library/2020-12-23-19-43-06.bpo-42727.WH3ODh.rst b/Misc/NEWS.d/next/Library/2020-12-23-19-43-06.bpo-42727.WH3ODh.rst
new file mode 100644 (file)
index 0000000..a986cb9
--- /dev/null
@@ -0,0 +1,2 @@
+`EnumMeta.__prepare__` now accepts `**kwds` to properly support
+`__init_subclass__`