aboutsummaryrefslogtreecommitdiff
path: root/requests/structures.py
blob: 0c82c7b46f7664c42031fb85f7355c995aad76d5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# -*- coding: utf-8 -*-

"""
requests.structures
~~~~~~~~~~~~~~~~~~~

Datastructures that power Requests.

"""

from UserDict import DictMixin


class CaseInsensitiveDict(DictMixin):
    """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())


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


    def __getitem__(self, key):

        if key.lower() 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()