From: Ethan Furman Date: Fri, 25 Apr 2025 06:13:54 +0000 (-0700) Subject: gh-132684: [Enum] only call _missing_ in __contains__ for Flags (GH-132790) X-Git-Tag: v3.14.0b1~305 X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=22bc953aa9be3039629dd1315f856d2522619412;p=thirdparty%2FPython%2Fcpython.git gh-132684: [Enum] only call _missing_ in __contains__ for Flags (GH-132790) Co-authored-by: Bénédikt Tran <10796600+picnixz@users.noreply.github.com> --- diff --git a/Lib/enum.py b/Lib/enum.py index b5f3ca7ae111..01fecca3e5aa 100644 --- a/Lib/enum.py +++ b/Lib/enum.py @@ -731,14 +731,16 @@ class EnumType(type): """ if isinstance(value, cls): return True - try: - cls(value) - return True - except ValueError: - return ( - value in cls._unhashable_values_ # both structures are lists - or value in cls._hashable_values_ - ) + if issubclass(cls, Flag): + try: + result = cls._missing_(value) + return isinstance(result, cls) + except ValueError: + pass + return ( + value in cls._unhashable_values_ # both structures are lists + or value in cls._hashable_values_ + ) def __delattr__(cls, attr): # nicer error message when someone tries to delete an attribute diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py index dde674164f4a..68cedc666a59 100644 --- a/Lib/test/test_enum.py +++ b/Lib/test/test_enum.py @@ -1569,6 +1569,17 @@ class TestSpecial(unittest.TestCase): self.assertIn(IntEnum1.X, IntFlag1) self.assertIn(IntFlag1.X, IntEnum1) + def test_contains_does_not_call_missing(self): + class AnEnum(Enum): + UNKNOWN = None + LUCKY = 3 + @classmethod + def _missing_(cls, *values): + return cls.UNKNOWN + self.assertTrue(None in AnEnum) + self.assertTrue(3 in AnEnum) + self.assertFalse(7 in AnEnum) + def test_inherited_data_type(self): class HexInt(int): __qualname__ = 'HexInt'