updated from Atlas

attic/functions/strkeydict2.py

@@ -0,0 +1,76 @@
+"""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