gh-100272: Fix JSON serialization of OrderedDict (GH-100273)

It now preserves the order of keys.
This commit is contained in:
Serhiy Storchaka 2022-12-17 14:32:48 +02:00 committed by GitHub
parent 2b38a9aa74
commit 0fe61d0838
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 4 deletions

View File

@ -1,3 +1,4 @@
import collections
from test.test_json import PyTest, CTest
@ -7,6 +8,16 @@ class TestDefault:
self.dumps(type, default=repr),
self.dumps(repr(type)))
def test_ordereddict(self):
od = collections.OrderedDict(a=1, b=2, c=3, d=4)
od.move_to_end('b')
self.assertEqual(
self.dumps(od),
'{"a": 1, "c": 3, "d": 4, "b": 2}')
self.assertEqual(
self.dumps(od, sort_keys=True),
'{"a": 1, "b": 2, "c": 3, "d": 4}')
class TestPyDefault(TestDefault, PyTest): pass
class TestCDefault(TestDefault, CTest): pass

View File

@ -0,0 +1 @@
Fix JSON serialization of OrderedDict. It now preserves the order of keys.

View File

@ -1570,10 +1570,9 @@ encoder_listencode_dict(PyEncoderObject *s, _PyUnicodeWriter *writer,
*/
}
if (s->sort_keys) {
items = PyDict_Items(dct);
if (items == NULL || PyList_Sort(items) < 0)
if (s->sort_keys || !PyDict_CheckExact(dct)) {
items = PyMapping_Items(dct);
if (items == NULL || (s->sort_keys && PyList_Sort(items) < 0))
goto bail;
for (Py_ssize_t i = 0; i < PyList_GET_SIZE(items); i++) {