[Python-checkins] gh-93820: Pickle enum.Flag by name (GH-93891)
serhiy-storchaka
webhook-mailer at python.org
Sun Jun 26 03:54:21 EDT 2022
https://github.com/python/cpython/commit/536985814a7116f14c9bc90aa1b3e3d36d5b2367
commit: 536985814a7116f14c9bc90aa1b3e3d36d5b2367
branch: main
author: Serhiy Storchaka <storchaka at gmail.com>
committer: serhiy-storchaka <storchaka at gmail.com>
date: 2022-06-26T10:54:00+03:00
summary:
gh-93820: Pickle enum.Flag by name (GH-93891)
files:
A Misc/NEWS.d/next/Library/2022-06-16-11-16-53.gh-issue-93820.00X0Y5.rst
M Lib/enum.py
M Lib/test/test_enum.py
diff --git a/Lib/enum.py b/Lib/enum.py
index 15c8d88016710..69216c97fa86b 100644
--- a/Lib/enum.py
+++ b/Lib/enum.py
@@ -1273,7 +1273,21 @@ class Flag(Enum, boundary=STRICT):
"""
def __reduce_ex__(self, proto):
- return self.__class__, (self._value_, )
+ cls = self.__class__
+ unknown = self._value_ & ~cls._flag_mask_
+ member_value = self._value_ & cls._flag_mask_
+ if unknown and member_value:
+ return _or_, (cls(member_value), unknown)
+ for val in _iter_bits_lsb(member_value):
+ rest = member_value & ~val
+ if rest:
+ return _or_, (cls(rest), cls._value2member_map_.get(val))
+ else:
+ break
+ if self._name_ is None:
+ return cls, (self._value_,)
+ else:
+ return getattr, (cls, self._name_)
_numeric_repr_ = repr
diff --git a/Lib/test/test_enum.py b/Lib/test/test_enum.py
index 1968c8abd50dc..93b7a52bec280 100644
--- a/Lib/test/test_enum.py
+++ b/Lib/test/test_enum.py
@@ -66,10 +66,27 @@ class FloatStooges(float, Enum):
class FlagStooges(Flag):
LARRY = 1
CURLY = 2
- MOE = 3
+ MOE = 4
except Exception as exc:
FlagStooges = exc
+class FlagStoogesWithZero(Flag):
+ NOFLAG = 0
+ LARRY = 1
+ CURLY = 2
+ MOE = 4
+
+class IntFlagStooges(IntFlag):
+ LARRY = 1
+ CURLY = 2
+ MOE = 4
+
+class IntFlagStoogesWithZero(IntFlag):
+ NOFLAG = 0
+ LARRY = 1
+ CURLY = 2
+ MOE = 4
+
# for pickle test and subclass tests
class Name(StrEnum):
BDFL = 'Guido van Rossum'
@@ -2999,9 +3016,32 @@ def test_programatic_function_from_dict(self):
def test_pickle(self):
if isinstance(FlagStooges, Exception):
raise FlagStooges
- test_pickle_dump_load(self.assertIs, FlagStooges.CURLY|FlagStooges.MOE)
+ test_pickle_dump_load(self.assertIs, FlagStooges.CURLY)
+ test_pickle_dump_load(self.assertEqual,
+ FlagStooges.CURLY|FlagStooges.MOE)
+ test_pickle_dump_load(self.assertEqual,
+ FlagStooges.CURLY&~FlagStooges.CURLY)
test_pickle_dump_load(self.assertIs, FlagStooges)
+ test_pickle_dump_load(self.assertIs, FlagStoogesWithZero.CURLY)
+ test_pickle_dump_load(self.assertEqual,
+ FlagStoogesWithZero.CURLY|FlagStoogesWithZero.MOE)
+ test_pickle_dump_load(self.assertIs, FlagStoogesWithZero.NOFLAG)
+
+ test_pickle_dump_load(self.assertIs, IntFlagStooges.CURLY)
+ test_pickle_dump_load(self.assertEqual,
+ IntFlagStooges.CURLY|IntFlagStooges.MOE)
+ test_pickle_dump_load(self.assertEqual,
+ IntFlagStooges.CURLY|IntFlagStooges.MOE|0x30)
+ test_pickle_dump_load(self.assertEqual, IntFlagStooges(0))
+ test_pickle_dump_load(self.assertEqual, IntFlagStooges(0x30))
+ test_pickle_dump_load(self.assertIs, IntFlagStooges)
+
+ test_pickle_dump_load(self.assertIs, IntFlagStoogesWithZero.CURLY)
+ test_pickle_dump_load(self.assertEqual,
+ IntFlagStoogesWithZero.CURLY|IntFlagStoogesWithZero.MOE)
+ test_pickle_dump_load(self.assertIs, IntFlagStoogesWithZero.NOFLAG)
+
def test_contains_tf(self):
Open = self.Open
Color = self.Color
diff --git a/Misc/NEWS.d/next/Library/2022-06-16-11-16-53.gh-issue-93820.00X0Y5.rst b/Misc/NEWS.d/next/Library/2022-06-16-11-16-53.gh-issue-93820.00X0Y5.rst
new file mode 100644
index 0000000000000..70bb1e6c0cd76
--- /dev/null
+++ b/Misc/NEWS.d/next/Library/2022-06-16-11-16-53.gh-issue-93820.00X0Y5.rst
@@ -0,0 +1 @@
+Pickle :class:`enum.Flag` by name.
More information about the Python-checkins
mailing list