Commit a673b1fd authored by Raymond Hettinger's avatar Raymond Hettinger

Fix OrderedDict.setdefault() to work for subclasses that define __missing__().

parent ed13853e
...@@ -171,7 +171,6 @@ class OrderedDict(dict, MutableMapping): ...@@ -171,7 +171,6 @@ class OrderedDict(dict, MutableMapping):
size += sizeof(self.__root) * n # proxy objects size += sizeof(self.__root) * n # proxy objects
return size return size
setdefault = MutableMapping.setdefault
update = MutableMapping.update update = MutableMapping.update
pop = MutableMapping.pop pop = MutableMapping.pop
keys = MutableMapping.keys keys = MutableMapping.keys
...@@ -179,6 +178,13 @@ class OrderedDict(dict, MutableMapping): ...@@ -179,6 +178,13 @@ class OrderedDict(dict, MutableMapping):
items = MutableMapping.items items = MutableMapping.items
__ne__ = MutableMapping.__ne__ __ne__ = MutableMapping.__ne__
def setdefault(self, key, default=None):
'OD.setdefault(k[,d]) -> OD.get(k,d), also set OD[k]=d if k not in OD'
if key in self:
return self[key]
self[key] = default
return default
@_recursive_repr() @_recursive_repr()
def __repr__(self): def __repr__(self):
'od.__repr__() <==> repr(od)' 'od.__repr__() <==> repr(od)'
......
...@@ -976,6 +976,12 @@ class TestOrderedDict(unittest.TestCase): ...@@ -976,6 +976,12 @@ class TestOrderedDict(unittest.TestCase):
# make sure 'x' is added to the end # make sure 'x' is added to the end
self.assertEqual(list(od.items())[-1], ('x', 10)) self.assertEqual(list(od.items())[-1], ('x', 10))
# make sure setdefault still works when __missing__ is defined
class Missing(OrderedDict):
def __missing__(self, key):
return 0
self.assertEqual(Missing().setdefault(5, 9), 9)
def test_reinsert(self): def test_reinsert(self):
# Given insert a, insert b, delete a, re-insert a, # Given insert a, insert b, delete a, re-insert a,
# verify that a is now later than b. # verify that a is now later than b.
......
...@@ -20,6 +20,9 @@ Core and Builtins ...@@ -20,6 +20,9 @@ Core and Builtins
Library Library
------- -------
- Fix collections.OrderedDict.setdefault() so that it works in
subclasses that define __missing__().
- Issue 10786: unittest.TextTestRunner default stream no longer bound at - Issue 10786: unittest.TextTestRunner default stream no longer bound at
import time. `sys.stderr` now looked up at instantiation time. Fix contributed import time. `sys.stderr` now looked up at instantiation time. Fix contributed
by Mark Roddy. by Mark Roddy.
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment