119 lines
3.8 KiB
Python
119 lines
3.8 KiB
Python
import types
|
|
|
|
try:
|
|
from collections.abc import Mapping, MutableMapping
|
|
except ImportError:
|
|
from collections import Mapping, MutableMapping
|
|
|
|
|
|
# Shamelessly stolen from https://github.com/kennethreitz/requests/blob/master/requests/structures.py
|
|
class CaseInsensitiveDict(MutableMapping):
|
|
"""
|
|
A case-insensitive ``dict``-like object.
|
|
Implements all methods and operations of
|
|
``collections.abc.MutableMapping`` as well as dict's ``copy``. Also
|
|
provides ``lower_items``.
|
|
All keys are expected to be strings. The structure remembers the
|
|
case of the last key to be set, and ``iter(instance)``,
|
|
``keys()``, ``items()``, ``iterkeys()``, and ``iteritems()``
|
|
will contain case-sensitive keys. However, querying and contains
|
|
testing is case insensitive::
|
|
cid = CaseInsensitiveDict()
|
|
cid['Accept'] = 'application/json'
|
|
cid['aCCEPT'] == 'application/json' # True
|
|
list(cid) == ['Accept'] # True
|
|
For example, ``headers['content-encoding']`` will return the
|
|
value of a ``'Content-Encoding'`` response header, regardless
|
|
of how the header name was originally stored.
|
|
If the constructor, ``.update``, or equality comparison
|
|
operations are given keys that have equal ``.lower()``s, the
|
|
behavior is undefined.
|
|
"""
|
|
|
|
def __init__(self, data=None, **kwargs):
|
|
self._store = dict()
|
|
if data is None:
|
|
data = {}
|
|
self.update(data, **kwargs)
|
|
|
|
def __setitem__(self, key, value):
|
|
# Use the lowercased key for lookups, but store the actual
|
|
# key alongside the value.
|
|
self._store[key.lower()] = (key, value)
|
|
|
|
def __getitem__(self, key):
|
|
return self._store[key.lower()][1]
|
|
|
|
def __delitem__(self, key):
|
|
del self._store[key.lower()]
|
|
|
|
def __iter__(self):
|
|
return (casedkey for casedkey, mappedvalue in self._store.values())
|
|
|
|
def __len__(self):
|
|
return len(self._store)
|
|
|
|
def lower_items(self):
|
|
"""Like iteritems(), but with all lowercase keys."""
|
|
return ((lowerkey, keyval[1]) for (lowerkey, keyval) in self._store.items())
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(other, Mapping):
|
|
other = CaseInsensitiveDict(other)
|
|
else:
|
|
return NotImplemented
|
|
# Compare insensitively
|
|
return dict(self.lower_items()) == dict(other.lower_items())
|
|
|
|
# Copy is required
|
|
def copy(self):
|
|
return CaseInsensitiveDict(self._store.values())
|
|
|
|
def __repr__(self):
|
|
return str(dict(self.items()))
|
|
|
|
|
|
def partition_dict(predicate, dictionary):
|
|
true_dict = {}
|
|
false_dict = {}
|
|
for key, value in dictionary.items():
|
|
this_dict = true_dict if predicate(key, value) else false_dict
|
|
this_dict[key] = value
|
|
return true_dict, false_dict
|
|
|
|
|
|
def compose(*functions):
|
|
def composed(incoming):
|
|
res = incoming
|
|
for function in reversed(functions):
|
|
if function:
|
|
res = function(res)
|
|
return res
|
|
|
|
return composed
|
|
|
|
|
|
def read_body(request):
|
|
if hasattr(request.body, "read"):
|
|
return request.body.read()
|
|
return request.body
|
|
|
|
|
|
def auto_decorate(decorator, predicate=lambda name, value: isinstance(value, types.FunctionType)):
|
|
def maybe_decorate(attribute, value):
|
|
if predicate(attribute, value):
|
|
value = decorator(value)
|
|
return value
|
|
|
|
class DecorateAll(type):
|
|
def __setattr__(cls, attribute, value):
|
|
return super(DecorateAll, cls).__setattr__(attribute, maybe_decorate(attribute, value))
|
|
|
|
def __new__(cls, name, bases, attributes_dict):
|
|
new_attributes_dict = {
|
|
attribute: maybe_decorate(attribute, value) for attribute, value in attributes_dict.items()
|
|
}
|
|
return super(DecorateAll, cls).__new__(cls, name, bases, new_attributes_dict)
|
|
|
|
return DecorateAll
|