https://github.com/python/cpython/commit/13ffd4bd9f529b6a5fe33741fbd57f14b4b80137
commit: 13ffd4bd9f529b6a5fe33741fbd57f14b4b80137
branch: main
author: Ethan Furman <[email protected]>
committer: ethanfurman <[email protected]>
date: 2024-03-07T13:30:26-08:00
summary:
gh-116040: [Enum] fix by-value calls when second value is falsey; e.g.
Cardinal(1, 0) (GH-116072)
files:
A Misc/NEWS.d/next/Library/2024-02-28-13-10-17.gh-issue-116040.wDidHd.rst
M Lib/enum.py
M Lib/test/test_enum.py
diff --git a/Lib/enum.py b/Lib/enum.py
index 22963cca4466f2..3499cb0b71547c 100644
--- a/Lib/enum.py
+++ b/Lib/enum.py
@@ -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 @@ def __bool__(cls):
"""
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 @@ def __call__(cls, value, names=None, *values,
module=None, qualname=None, type=N
"""
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,
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 27f8bbaf952afc..0a44b61e9049ed 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -3409,6 +3409,15 @@ def __new__(cls, int_value, *value_aliases):
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 @@ def test_no_members(self):
):
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
index 00000000000000..907b58b3a5c206
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2024-02-28-13-10-17.gh-issue-116040.wDidHd.rst
@@ -0,0 +1 @@
+[Enum] fix by-value calls when second value is falsey; e.g. Cardinal(1, 0)
_______________________________________________
Python-checkins mailing list -- [email protected]
To unsubscribe send an email to [email protected]
https://mail.python.org/mailman3/lists/python-checkins.python.org/
Member address: [email protected]
