77 lines
1.8 KiB
Python
77 lines
1.8 KiB
Python
"""StrKeyDict always converts non-string keys to `str`
|
|
|
|
Tests for item retrieval using `d[key]` notation::
|
|
|
|
>>> d = StrKeyDict([('2', 'two'), ('4', 'four')])
|
|
>>> d['2']
|
|
'two'
|
|
>>> d[4]
|
|
'four'
|
|
>>> d[1]
|
|
Traceback (most recent call last):
|
|
...
|
|
KeyError: '1'
|
|
|
|
Tests for the `in` operator::
|
|
|
|
>>> 2 in d
|
|
True
|
|
>>> 1 in d
|
|
False
|
|
|
|
Test for item assignment using non-string key::
|
|
|
|
>>> d[0] = 'zero'
|
|
>>> d['0']
|
|
'zero'
|
|
|
|
Tests for update using a `dict` or a sequence of pairs::
|
|
|
|
>>> d.update({6:'six', '8':'eight'})
|
|
>>> sorted(d.keys())
|
|
['0', '2', '4', '6', '8']
|
|
>>> d.update([(10, 'ten'), ('12', 'twelve')])
|
|
>>> sorted(d.keys())
|
|
['0', '10', '12', '2', '4', '6', '8']
|
|
>>> d.update([1, 3, 5])
|
|
Traceback (most recent call last):
|
|
...
|
|
TypeError: 'int' object is not iterable
|
|
|
|
"""
|
|
# BEGIN STRKEYDICT
|
|
|
|
import collections
|
|
import collections.abc
|
|
|
|
|
|
class StrKeyDict(collections.UserDict): # <1>
|
|
|
|
def __init__(self, args, normalize=str, **kwargs):
|
|
super().__init__(self, *args, **kwargs)
|
|
self.normalize = normalize
|
|
|
|
def __missing__(self, key): # <2>
|
|
if self.normalize(key) == key:
|
|
raise KeyError(key)
|
|
return self[self.normalize(key)]
|
|
|
|
def __contains__(self, key):
|
|
return self.normalize(key) in self.data # <3>
|
|
|
|
def __setitem__(self, key, item):
|
|
self.data[self.normalize(key)] = item # <4>
|
|
|
|
def update(self, iterable=None, **kwds):
|
|
if iterable is not None:
|
|
if isinstance(iterable, collections.abc.Mapping): # <5>
|
|
pairs = iterable.items()
|
|
else:
|
|
pairs = ((k, v) for k, v in iterable) # <6>
|
|
for key, value in pairs:
|
|
self[key] = value # <7>
|
|
if kwds:
|
|
self.update(kwds) # <8>
|
|
|
|
# END STRKEYDICT
|