]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
gh-116040: [Enum] fix by-value calls when second value is falsey; e.g. Cardinal(1...
authorEthan Furman <ethan@stoneleaf.us>
Thu, 7 Mar 2024 21:30:26 +0000 (13:30 -0800)
committerGitHub <noreply@github.com>
Thu, 7 Mar 2024 21:30:26 +0000 (13:30 -0800)
Lib/enum.py
Lib/test/test_enum.py
Misc/NEWS.d/next/Library/2024-02-28-13-10-17.gh-issue-116040.wDidHd.rst [new file with mode: 0644]

index 22963cca4466f2df2b6b0a57138026b4fde0ef18..3499cb0b71547cf23588bba9c4e8a7c2f973edef 100644 (file)
@@ -162,6 +162,13 @@ def _dedent(text):
         lines[j] = l[i:]
     return '\n'.join(lines)
 
+class _not_given:
+    def __repr__(self):
+        return('<not given>')
+    def __bool__(self):
+        return False
+_not_given = _not_given()
+
 class _auto_null:
     def __repr__(self):
         return '_auto_null'
@@ -680,7 +687,7 @@ class EnumType(type):
         """
         return True
 
-    def __call__(cls, value, names=None, *values, module=None, qualname=None, type=None, start=1, boundary=None):
+    def __call__(cls, value, names=_not_given, *values, module=None, qualname=None, type=None, start=1, boundary=None):
         """
         Either returns an existing member, or creates a new enum class.
 
@@ -709,18 +716,18 @@ class EnumType(type):
         """
         if cls._member_map_:
             # simple value lookup if members exist
-            if names:
+            if names is not _not_given:
                 value = (value, names) + values
             return cls.__new__(cls, value)
         # otherwise, functional API: we're creating a new Enum type
-        if names is None and type is None:
+        if names is _not_given and type is None:
             # no body? no data-type? possibly wrong usage
             raise TypeError(
                     f"{cls} has no members; specify `names=()` if you meant to create a new, empty, enum"
                     )
         return cls._create_(
                 class_name=value,
-                names=names,
+                names=names or None,
                 module=module,
                 qualname=qualname,
                 type=type,
index 27f8bbaf952afca97dc15cb3d5035096ede24edb..0a44b61e9049ede33063f485075984777f2e2563 100644 (file)
@@ -3409,6 +3409,15 @@ class TestSpecial(unittest.TestCase):
         self.assertIs(Types(2), Types.NetList)
         self.assertIs(Types('nl'), Types.NetList)
 
+    def test_second_tuple_item_is_falsey(self):
+        class Cardinal(Enum):
+            RIGHT = (1, 0)
+            UP = (0, 1)
+            LEFT = (-1, 0)
+            DOWN = (0, -1)
+        self.assertIs(Cardinal(1, 0), Cardinal.RIGHT)
+        self.assertIs(Cardinal(-1, 0), Cardinal.LEFT)
+
     def test_no_members(self):
         with self.assertRaisesRegex(
                 TypeError,
@@ -3421,6 +3430,20 @@ class TestSpecial(unittest.TestCase):
             ):
             Flag(7)
 
+    def test_empty_names(self):
+        for nothing, e_type in (
+                ('', None),
+                ('', int),
+                ([], None),
+                ([], int),
+                ({}, None),
+                ({}, int),
+            ):
+            empty_enum = Enum('empty_enum', nothing, type=e_type)
+            self.assertEqual(len(empty_enum), 0)
+            self.assertRaises(TypeError, 'has no members', empty_enum, 0)
+
+
 class TestOrder(unittest.TestCase):
     "test usage of the `_order_` attribute"
 
diff --git a/Misc/NEWS.d/next/Library/2024-02-28-13-10-17.gh-issue-116040.wDidHd.rst b/Misc/NEWS.d/next/Library/2024-02-28-13-10-17.gh-issue-116040.wDidHd.rst
new file mode 100644 (file)
index 0000000..907b58b
--- /dev/null
@@ -0,0 +1 @@
+[Enum] fix by-value calls when second value is falsey; e.g. Cardinal(1, 0)