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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
import json
import jsmin
import os
class Dict(dict):
"""
Dictionary that allows to access per attributes and to except names from being loaded
"""
def __init__(self, dictionary: dict = None):
super(Dict, self).__init__()
if dictionary is not None:
self.load(dictionary)
def __getattr__(self, item):
try:
return self[item] if item in self else getattr(super(Dict, self), item)
except AttributeError:
raise AttributeError(f'This dictionary has no attribute "{item}"')
def load(self, dictionary: dict, name_list: list = None):
"""
Loads a dictionary
:param dictionary: Dictionary to be loaded
:param name_list: List of names to be updated
"""
for name in dictionary:
data = dictionary[name]
if name_list is None or name in name_list:
if isinstance(data, dict):
if name in self:
self[name].load(data)
else:
self[name] = Dict(data)
elif isinstance(data, list):
self[name] = list()
for item in data:
if isinstance(item, dict):
self[name].append(Dict(item))
else:
self[name].append(item)
else:
self[name] = data
def save(self, path):
"""
Saves the dictionary into a json file
:param path: Path of the json file
"""
os.makedirs(path, exist_ok=True)
path = os.path.join(path, 'cfg.json')
with open(path, 'w') as file:
json.dump(self, file, indent=True)
class Configuration(Dict):
"""
Configuration loaded from a json file
"""
def __init__(self, path: str, default_path=None):
super(Configuration, self).__init__()
if default_path is not None:
self.load(default_path)
self.load(path)
def load_model(self, path: str):
self.load(path, name_list=["model"])
def load(self, path: str, name_list: list = None):
"""
Loads attributes from a json file
:param path: Path of the json file
:param name_list: List of names to be updated
:return:
"""
with open(path) as file:
data = json.loads(jsmin.jsmin(file.read()))
super(Configuration, self).load(data, name_list)
|
