Skip to content

Commit

Permalink
Subclass dict for CaseInsensitiveDict.
Browse files Browse the repository at this point in the history
  • Loading branch information
JNRowe committed May 27, 2011
1 parent 7e3d265 commit 5962c0e
Showing 1 changed file with 4 additions and 40 deletions.
44 changes: 4 additions & 40 deletions requests/structures.py
Expand Up @@ -8,57 +8,21 @@
"""

from UserDict import DictMixin


class CaseInsensitiveDict(DictMixin):
class CaseInsensitiveDict(dict):
"""Case-insensitive Dictionary for :class:`Response <models.Response>` Headers.
For example, ``headers['content-encoding']`` will return the
value of a ``'Content-Encoding'`` response header."""

def __init__(self, *args, **kwargs):
# super(CaseInsensitiveDict, self).__init__()
self.data = dict(*args, **kwargs)

def __repr__(self):
return self.data.__repr__()

def __getstate__(self):
return self.data.copy()

def __setstate__(self, d):
self.data = d

def _lower_keys(self):
return map(str.lower, self.data.keys())
return map(str.lower, self.keys())


def __contains__(self, key):
return key.lower() in self._lower_keys()


def __getitem__(self, key):

if key.lower() in self:
# We allow fall-through here, so values default to None
if key in self:
return self.items()[self._lower_keys().index(key.lower())][1]


def __setitem__(self, key, value):
return self.data.__setitem__(key, value)


def __delitem__(self, key):
return self.data.__delitem__(key)


def __keys__(self):
return self.data.__keys__()


def __iter__(self):
return self.data.__iter__()


def iteritems(self):
return self.data.iteritems()

0 comments on commit 5962c0e

Please sign in to comment.