-
Notifications
You must be signed in to change notification settings - Fork 0
/
easydict.py
57 lines (46 loc) · 2.12 KB
/
easydict.py
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
class EasyDict:
def _norm(self, k):
if self.normalize:
return k.replace(" ","_")
else:
return k
def _normdic(self, dic):
return {self._norm(k):v for k,v in dic.items()}
def __init__(self, *args, normalize = False, **kwargs):
self.normalize=normalize
args = ( self._normdic(item ) if isinstance(item, dict) else item for item in args)
kwargs = self._normdic(kwargs)
self.__dict__.update(*args, **kwargs)
def __getitem__(self, key):
return self.__dict__[self._norm(key)]
def __setitem__(self, key, value):
self.__dict__[self._norm(key)] = value
def __repr__(self):
return str(self.__dict__)
def __eq__(self, other):
return self.__dict__ == other.__dict__
def __ne__(self, other):
return self.__dict__ != other.__dict__:
def get(self, key, default = None):
return self.__dict__.get(self._norm(key),default)
if __name__ == "__main__":
person = EasyDict({'name': "Trey Hunner", 'location': "San Diego"})
assert person.name == 'Trey Hunner'
person.location = "Portland"
assert person['location'] == 'Portland'
person['location'] = "San Diego"
assert person.location == 'San Diego'
person = EasyDict(name="Trey Hunner", location="San Diego")
assert person.location == 'San Diego'
assert (person == EasyDict(name="Trey", location="San Diego")) == False
assert (person != EasyDict(name="Gordon")) == True
assert (person == EasyDict(name="Trey Hunner", location="San Diego")) == True
assert person.get('profession') == None
assert person.get('profession', 'unknown') == "unknown"
assert person.get('name', 'unknown') == 'Trey Hunner'
person = EasyDict(name="Trey Hunner", location="San Diego", normalize=True)
person['company name'] = "Truthful Technology LLC"
assert person.company_name == "Truthful Technology LLC"
assert person['company name'] == "Truthful Technology LLC"
person = EasyDict({'name':'Gordon Shephard', 'first job':'Mcdonalds'}, normalize=True)
assert person.first_job == 'Mcdonalds'