https://github.com/python/cpython/commit/7d2f88edf0365c2d9b872dc7847b9a77a5df8aac
commit: 7d2f88edf0365c2d9b872dc7847b9a77a5df8aac
branch: 3.12
author: Miss Islington (bot) <[email protected]>
committer: ethanfurman <[email protected]>
date: 2024-02-08T14:01:38-08:00
summary:
[3.12] gh-114071: [Enum] update docs and code for tuples/subclasses (GH-114871)
(GH-114993)
Update documentation with `__new__` and `__init__` entries.
Support use of `auto()` in tuple subclasses on member assignment lines.
Previously, auto() was only supported on the member definition line either solo
or as part of a tuple:
RED = auto()
BLUE = auto(), 'azul'
However, since Python itself supports using tuple subclasses where tuples are
expected, e.g.:
from collections import namedtuple
T = namedtuple('T', 'first second third')
def test(one, two, three):
print(one, two, three)
test(*T(4, 5, 6))
GH- 4 5 6
it made sense to also support tuple subclasses in enum definitions.
(cherry picked from commit ff7588b729a2a414ea189a2012904da3fbd1401c)
Co-authored-by: Ethan Furman <[email protected]>
files:
A Misc/NEWS.d/next/Library/2024-02-01-10-19-11.gh-issue-114071.vkm2G_.rst
M Doc/library/enum.rst
M Lib/enum.py
M Lib/test/test_enum.py
diff --git a/Doc/library/enum.rst b/Doc/library/enum.rst
index 3b874594b14f9c..72c8bb90e3fe8a 100644
--- a/Doc/library/enum.rst
+++ b/Doc/library/enum.rst
@@ -323,6 +323,17 @@ Data Types
>>> PowersOfThree.SECOND.value
9
+ .. method:: Enum.__init__(self, \*args, \**kwds)
+
+ By default, does nothing. If multiple values are given in the member
+ assignment, those values become separate arguments to ``__init__``; e.g.
+
+ >>> from enum import Enum
+ >>> class Weekday(Enum):
+ ... MONDAY = 1, 'Mon'
+
+ ``Weekday.__init__()`` would be called as ``Weekday.__init__(self, 1,
'Mon')``
+
.. method:: Enum.__init_subclass__(cls, \**kwds)
A *classmethod* that is used to further configure subsequent subclasses.
@@ -350,6 +361,18 @@ Data Types
>>> Build('deBUG')
<Build.DEBUG: 'debug'>
+ .. method:: Enum.__new__(cls, \*args, \**kwds)
+
+ By default, doesn't exist. If specified, either in the enum class
+ definition or in a mixin class (such as ``int``), all values given
+ in the member assignment will be passed; e.g.
+
+ >>> from enum import Enum
+ >>> class MyIntEnum(Enum):
+ ... SEVENTEEN = '1a', 16
+
+ results in the call ``int('1a', 16)`` and a value of ``17`` for the
member.
+
.. method:: Enum.__repr__(self)
Returns the string used for *repr()* calls. By default, returns the
@@ -463,9 +486,9 @@ Data Types
.. class:: Flag
- *Flag* members support the bitwise operators ``&`` (*AND*), ``|`` (*OR*),
- ``^`` (*XOR*), and ``~`` (*INVERT*); the results of those operators are
members
- of the enumeration.
+ ``Flag`` is the same as :class:`Enum`, but its members support the bitwise
+ operators ``&`` (*AND*), ``|`` (*OR*), ``^`` (*XOR*), and ``~`` (*INVERT*);
+ the results of those operators are members of the enumeration.
.. method:: __contains__(self, value)
diff --git a/Lib/enum.py b/Lib/enum.py
index 1502bfe9158520..b9c7f9cead9706 100644
--- a/Lib/enum.py
+++ b/Lib/enum.py
@@ -457,10 +457,11 @@ def __setitem__(self, key, value):
if isinstance(value, auto):
single = True
value = (value, )
- if type(value) is tuple and any(isinstance(v, auto) for v in
value):
+ if isinstance(value, tuple) and any(isinstance(v, auto) for v in
value):
# insist on an actual tuple, no subclasses, in keeping with
only supporting
# top-level auto() usage (not contained in any other data
structure)
auto_valued = []
+ t = type(value)
for v in value:
if isinstance(v, auto):
non_auto_store = False
@@ -475,7 +476,12 @@ def __setitem__(self, key, value):
if single:
value = auto_valued[0]
else:
- value = tuple(auto_valued)
+ try:
+ # accepts iterable as multiple arguments?
+ value = t(auto_valued)
+ except TypeError:
+ # then pass them in singlely
+ value = t(*auto_valued)
self._member_names[key] = None
if non_auto_store:
self._last_values.append(value)
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 23be142c24de0f..ccb71da25d41c9 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -2308,6 +2308,40 @@ class SomeTuple(tuple, Enum):
globals()['SomeTuple'] = SomeTuple
test_pickle_dump_load(self.assertIs, SomeTuple.first)
+ def test_tuple_subclass_with_auto_1(self):
+ from collections import namedtuple
+ T = namedtuple('T', 'index desc')
+ class SomeEnum(T, Enum):
+ __qualname__ = 'SomeEnum' # needed for pickle protocol 4
+ first = auto(), 'for the money'
+ second = auto(), 'for the show'
+ third = auto(), 'for the music'
+ self.assertIs(type(SomeEnum.first), SomeEnum)
+ self.assertEqual(SomeEnum.third.value, (3, 'for the music'))
+ self.assertIsInstance(SomeEnum.third.value, T)
+ self.assertEqual(SomeEnum.first.index, 1)
+ self.assertEqual(SomeEnum.second.desc, 'for the show')
+ globals()['SomeEnum'] = SomeEnum
+ globals()['T'] = T
+ test_pickle_dump_load(self.assertIs, SomeEnum.first)
+
+ def test_tuple_subclass_with_auto_2(self):
+ from collections import namedtuple
+ T = namedtuple('T', 'index desc')
+ class SomeEnum(Enum):
+ __qualname__ = 'SomeEnum' # needed for pickle protocol 4
+ first = T(auto(), 'for the money')
+ second = T(auto(), 'for the show')
+ third = T(auto(), 'for the music')
+ self.assertIs(type(SomeEnum.first), SomeEnum)
+ self.assertEqual(SomeEnum.third.value, (3, 'for the music'))
+ self.assertIsInstance(SomeEnum.third.value, T)
+ self.assertEqual(SomeEnum.first.value.index, 1)
+ self.assertEqual(SomeEnum.second.value.desc, 'for the show')
+ globals()['SomeEnum'] = SomeEnum
+ globals()['T'] = T
+ test_pickle_dump_load(self.assertIs, SomeEnum.first)
+
def test_duplicate_values_give_unique_enum_items(self):
class AutoNumber(Enum):
first = ()
diff --git
a/Misc/NEWS.d/next/Library/2024-02-01-10-19-11.gh-issue-114071.vkm2G_.rst
b/Misc/NEWS.d/next/Library/2024-02-01-10-19-11.gh-issue-114071.vkm2G_.rst
new file mode 100644
index 00000000000000..587ce4d2157637
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2024-02-01-10-19-11.gh-issue-114071.vkm2G_.rst
@@ -0,0 +1 @@
+Support tuple subclasses using auto() for enum member value.
_______________________________________________
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]
