gh-132617: Fix dict.update() mutation check (gh-134815)
Use `ma_used` instead of `ma_keys->dk_nentries` for modification check so that we only check if the dictionary is modified, not if new keys are added to a different dictionary that shared the same keys object.
This commit is contained in:
@@ -290,6 +290,38 @@ class DictTest(unittest.TestCase):
|
|||||||
['Cannot convert dictionary update sequence element #0 to a sequence'],
|
['Cannot convert dictionary update sequence element #0 to a sequence'],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_update_shared_keys(self):
|
||||||
|
class MyClass: pass
|
||||||
|
|
||||||
|
# Subclass str to enable us to create an object during the
|
||||||
|
# dict.update() call.
|
||||||
|
class MyStr(str):
|
||||||
|
def __hash__(self):
|
||||||
|
return super().__hash__()
|
||||||
|
|
||||||
|
def __eq__(self, other):
|
||||||
|
# Create an object that shares the same PyDictKeysObject as
|
||||||
|
# obj.__dict__.
|
||||||
|
obj2 = MyClass()
|
||||||
|
obj2.a = "a"
|
||||||
|
obj2.b = "b"
|
||||||
|
obj2.c = "c"
|
||||||
|
return super().__eq__(other)
|
||||||
|
|
||||||
|
obj = MyClass()
|
||||||
|
obj.a = "a"
|
||||||
|
obj.b = "b"
|
||||||
|
|
||||||
|
x = {}
|
||||||
|
x[MyStr("a")] = MyStr("a")
|
||||||
|
|
||||||
|
# gh-132617: this previously raised "dict mutated during update" error
|
||||||
|
x.update(obj.__dict__)
|
||||||
|
|
||||||
|
self.assertEqual(x, {
|
||||||
|
MyStr("a"): "a",
|
||||||
|
"b": "b",
|
||||||
|
})
|
||||||
|
|
||||||
def test_fromkeys(self):
|
def test_fromkeys(self):
|
||||||
self.assertEqual(dict.fromkeys('abc'), {'a':None, 'b':None, 'c':None})
|
self.assertEqual(dict.fromkeys('abc'), {'a':None, 'b':None, 'c':None})
|
||||||
|
|||||||
@@ -0,0 +1,3 @@
|
|||||||
|
Fix :meth:`dict.update` modification check that could incorrectly raise a
|
||||||
|
"dict mutated during update" error when a different dictionary was modified
|
||||||
|
that happens to share the same underlying keys object.
|
||||||
@@ -3858,7 +3858,7 @@ dict_dict_merge(PyInterpreterState *interp, PyDictObject *mp, PyDictObject *othe
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Py_ssize_t orig_size = other->ma_keys->dk_nentries;
|
Py_ssize_t orig_size = other->ma_used;
|
||||||
Py_ssize_t pos = 0;
|
Py_ssize_t pos = 0;
|
||||||
Py_hash_t hash;
|
Py_hash_t hash;
|
||||||
PyObject *key, *value;
|
PyObject *key, *value;
|
||||||
@@ -3892,7 +3892,7 @@ dict_dict_merge(PyInterpreterState *interp, PyDictObject *mp, PyDictObject *othe
|
|||||||
if (err != 0)
|
if (err != 0)
|
||||||
return -1;
|
return -1;
|
||||||
|
|
||||||
if (orig_size != other->ma_keys->dk_nentries) {
|
if (orig_size != other->ma_used) {
|
||||||
PyErr_SetString(PyExc_RuntimeError,
|
PyErr_SetString(PyExc_RuntimeError,
|
||||||
"dict mutated during update");
|
"dict mutated during update");
|
||||||
return -1;
|
return -1;
|
||||||
|
|||||||
Reference in New Issue
Block a user