diff --git a/Lib/plistlib.py b/Lib/plistlib.py index 93f3ef5e38af84..2fc584a5df63a6 100644 --- a/Lib/plistlib.py +++ b/Lib/plistlib.py @@ -388,15 +388,14 @@ def write_bytes(self, data): def write_dict(self, d): if d: self.begin_element("dict") + items = d.items() + if self._skipkeys: + items = [(k, v) for k, v in items if isinstance(k, str)] if self._sort_keys: - items = sorted(d.items()) - else: - items = d.items() + items = sorted(items) for key, value in items: if not isinstance(key, str): - if self._skipkeys: - continue raise TypeError("keys must be strings") self.simple_element("key", key) self.write_value(value) @@ -719,13 +718,13 @@ def _flatten(self, value): keys = [] values = [] items = value.items() + if self._skipkeys: + items = [(k, v) for k, v in items if isinstance(k, str)] if self._sort_keys: items = sorted(items) for k, v in items: if not isinstance(k, str): - if self._skipkeys: - continue raise TypeError("keys must be strings") keys.append(k) values.append(v) @@ -839,15 +838,15 @@ def _write_object(self, value): elif isinstance(value, (dict, frozendict)): keyRefs, valRefs = [], [] + rootItems = value.items() + if self._skipkeys: + rootItems = [(k, v) for k, v in rootItems + if isinstance(k, str)] if self._sort_keys: - rootItems = sorted(value.items()) - else: - rootItems = value.items() + rootItems = sorted(rootItems) for k, v in rootItems: if not isinstance(k, str): - if self._skipkeys: - continue raise TypeError("keys must be strings") keyRefs.append(self._getrefnum(k)) valRefs.append(self._getrefnum(v)) diff --git a/Lib/test/test_plistlib.py b/Lib/test/test_plistlib.py index b9c261310bb567..fd6036c033ffb8 100644 --- a/Lib/test/test_plistlib.py +++ b/Lib/test/test_plistlib.py @@ -722,20 +722,34 @@ def test_skipkeys(self): 'snake': 'aWord', } + for fmt in ALL_FORMATS: + for sort_keys in (False, True): + with self.subTest(fmt=fmt, sort_keys=sort_keys): + data = plistlib.dumps( + pl, fmt=fmt, skipkeys=True, sort_keys=sort_keys) + + pl2 = plistlib.loads(data) + self.assertEqual(pl2, {'snake': 'aWord'}) + + fp = BytesIO() + plistlib.dump( + pl, fp, fmt=fmt, skipkeys=True, sort_keys=sort_keys) + data = fp.getvalue() + pl2 = plistlib.loads(fp.getvalue()) + self.assertEqual(pl2, {'snake': 'aWord'}) + + def test_skipkeys_with_sort_keys_mixed_types(self): + # gh-145856: skipkeys=True + sort_keys=True with mixed key types + # used to raise TypeError because the sort ran before the filter. + pl = {1: 'a', 'z': 'b', 'a': 'c'} + for fmt in ALL_FORMATS: with self.subTest(fmt=fmt): data = plistlib.dumps( - pl, fmt=fmt, skipkeys=True, sort_keys=False) - - pl2 = plistlib.loads(data) - self.assertEqual(pl2, {'snake': 'aWord'}) - - fp = BytesIO() - plistlib.dump( - pl, fp, fmt=fmt, skipkeys=True, sort_keys=False) - data = fp.getvalue() - pl2 = plistlib.loads(fp.getvalue()) - self.assertEqual(pl2, {'snake': 'aWord'}) + pl, fmt=fmt, skipkeys=True, sort_keys=True) + pl2 = plistlib.loads(data, dict_type=collections.OrderedDict) + self.assertEqual(dict(pl2), {'z': 'b', 'a': 'c'}) + self.assertEqual(list(pl2.keys()), ['a', 'z']) def test_tuple_members(self): pl = { diff --git a/Misc/NEWS.d/next/Library/2026-05-20-14-00-00.gh-issue-145856.Pl5kZx.rst b/Misc/NEWS.d/next/Library/2026-05-20-14-00-00.gh-issue-145856.Pl5kZx.rst new file mode 100644 index 00000000000000..9f8cc32fd81c88 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2026-05-20-14-00-00.gh-issue-145856.Pl5kZx.rst @@ -0,0 +1,4 @@ +Fix :func:`plistlib.dumps` and :func:`plistlib.dump` so that ``skipkeys=True`` +together with ``sort_keys=True`` correctly drops non-string keys when the +dictionary contains a mix of string and non-string keys. Previously the sort +ran before the filter and raised :exc:`TypeError`.