How do I make a defaultdict safe for unexpecting clients?

Question

Several times (even several in a row) I\'ve been bitten by the defaultdict bug: forgetting that something is actually a defaultdict and treating it like a regular dictionary

Answer 1

Using rkhayrov's idea of resetting self.default_factory, here is a toggleable subclass of defaultdict:

class ToggleableDefaultdict(collections.defaultdict):
    def __init__(self,default_factory):
        self._default_factory=default_factory
        super(ToggleableDefaultdict,self).__init__(default_factory)
    def off(self):
        self.default_factory=None
    def on(self):
        self.default_factory=self._default_factory

For example:

d=ToggleableDefaultdict(list)
d['key'].append(1)
print(d)
# defaultdict(, {'key': [1]})

d.off()
d['newkey'].append(2)
# KeyError: 'newkey'

d.on()
d['newkey'].append(2)
# defaultdict(, {'newkey': [2], 'key': [1]})