summaryrefslogtreecommitdiff
path: root/game/python-extra/utils/dicts/limited_dict.py
blob: c690118e8266748e27e12ce9d165f874ac7038f1 (plain) (blame)
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
from collections import MutableMapping


class LimitedDict(MutableMapping):
    def __init__(self, args=None, **kwargs):
        keys = kwargs.pop('keys', [])
        self.__keys = keys

        self.__data = {}

        if args:
            kwargs.update((key, val) for key, val in args)

        for key, val in kwargs.items():
            self[key] = val

    def __setitem__(self, key, val):
        if key not in self.__keys:
            raise KeyError("Illegal key: {}".format(key))

        self.__data[key] = val

    def __getitem__(self, key):
        return self.__data[key]

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

    def __delitem__(self, key):
        del self.__data[key]

    def __len__(self):
        return len(self.__data)

    def __repr__(self):
        return "{}({}, {})".format(self.__class__.__name__, self.defined_keys, self.__data)

    @property
    def defined_keys(self):
        return self.__keys