From 50b093f5c7060c0b44c264808411346cee7becf0 Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Tue, 2 Jan 2024 21:45:36 +0200 Subject: gh-53502: Fix plistlib.dump() for naive datetime with aware_datetime option (GH-113645) --- Lib/plistlib.py | 4 ++-- Lib/test/test_plistlib.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/Lib/plistlib.py b/Lib/plistlib.py index 6eb70ce..0fc1b5c 100644 --- a/Lib/plistlib.py +++ b/Lib/plistlib.py @@ -155,7 +155,7 @@ def _date_from_string(s, aware_datetime): def _date_to_string(d, aware_datetime): - if aware_datetime and d.tzinfo is not None: + if aware_datetime: d = d.astimezone(datetime.UTC) return '%04d-%02d-%02dT%02d:%02d:%02dZ' % ( d.year, d.month, d.day, @@ -791,7 +791,7 @@ class _BinaryPlistWriter (object): self._fp.write(struct.pack('>Bd', 0x23, value)) elif isinstance(value, datetime.datetime): - if self._aware_datetime and value.tzinfo is not None: + if self._aware_datetime: dt = value.astimezone(datetime.UTC) offset = dt - datetime.datetime(2001, 1, 1, tzinfo=datetime.UTC) f = offset.total_seconds() diff --git a/Lib/test/test_plistlib.py b/Lib/test/test_plistlib.py index 010393a..1d2e14a 100644 --- a/Lib/test/test_plistlib.py +++ b/Lib/test/test_plistlib.py @@ -885,7 +885,8 @@ class TestPlistlib(unittest.TestCase): for fmt in ALL_FORMATS: s = plistlib.dumps(dt, fmt=fmt, aware_datetime=True) parsed = plistlib.loads(s, aware_datetime=False) - self.assertEqual(parsed, dt) + expected = dt.astimezone(datetime.UTC).replace(tzinfo=None) + self.assertEqual(parsed, expected) class TestBinaryPlistlib(unittest.TestCase): -- cgit v0.12