]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
[3.7] bpo-34282: Fix Enum._convert method shadowing members named _convert (GH-9034)
authororlnub123 <orlnub123@gmail.com>
Mon, 10 Sep 2018 16:39:48 +0000 (19:39 +0300)
committerEthan Furman <ethan@stoneleaf.us>
Mon, 10 Sep 2018 16:39:48 +0000 (09:39 -0700)
* Fix Enum._convert shadowing members named _convert

Lib/enum.py
Lib/test/test_enum.py
Misc/NEWS.d/next/Library/2018-09-02-13-33-35.bpo-34282.ztyXH8.rst [new file with mode: 0644]

index 576de031805dbc8022da4fc13ff6901cc469dbc2..69b41fe7cbc08cafe1cacbc91f66c44e77b07892 100644 (file)
@@ -171,9 +171,11 @@ class EnumMeta(type):
         enum_class._member_map_ = OrderedDict()      # name->value map
         enum_class._member_type_ = member_type
 
-        # save attributes from super classes so we know if we can take
-        # the shortcut of storing members in the class dict
-        base_attributes = {a for b in enum_class.mro() for a in b.__dict__}
+        # save DynamicClassAttribute attributes from super classes so we know
+        # if we can take the shortcut of storing members in the class dict
+        dynamic_attributes = {k for c in enum_class.mro()
+                              for k, v in c.__dict__.items()
+                              if isinstance(v, DynamicClassAttribute)}
 
         # Reverse value->name map for hashable values.
         enum_class._value2member_map_ = {}
@@ -233,7 +235,7 @@ class EnumMeta(type):
                 enum_class._member_names_.append(member_name)
             # performance boost for any member that would not shadow
             # a DynamicClassAttribute
-            if member_name not in base_attributes:
+            if member_name not in dynamic_attributes:
                 setattr(enum_class, member_name, enum_member)
             # now add to _member_map_
             enum_class._member_map_[member_name] = enum_member
index ef2d1daaf9420dc670cd07ffb4a2fae7e5cad5c3..4b17228946015a58f0f8f5c6aa4b383717a09dae 100644 (file)
@@ -1516,6 +1516,23 @@ class TestEnum(unittest.TestCase):
             yellow = 6
         self.assertEqual(MoreColor.magenta.hex(), '5 hexlified!')
 
+    def test_subclass_duplicate_name(self):
+        class Base(Enum):
+            def test(self):
+                pass
+        class Test(Base):
+            test = 1
+        self.assertIs(type(Test.test), Test)
+
+    def test_subclass_duplicate_name_dynamic(self):
+        from types import DynamicClassAttribute
+        class Base(Enum):
+            @DynamicClassAttribute
+            def test(self):
+                return 'dynamic'
+        class Test(Base):
+            test = 1
+        self.assertEqual(Test.test.test, 'dynamic')
 
     def test_no_duplicates(self):
         class UniqueEnum(Enum):
diff --git a/Misc/NEWS.d/next/Library/2018-09-02-13-33-35.bpo-34282.ztyXH8.rst b/Misc/NEWS.d/next/Library/2018-09-02-13-33-35.bpo-34282.ztyXH8.rst
new file mode 100644 (file)
index 0000000..c1e606a
--- /dev/null
@@ -0,0 +1 @@
+Fix enum members getting shadowed by parent attributes.