File size: 2,049 Bytes
485c2ee |
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 |
import json
import os
from copy import deepcopy
from dataclasses import dataclass
@dataclass
class BaseConfig:
def clone(self):
return deepcopy(self)
def inherit(self, another):
"""inherit common keys from a given config"""
common_keys = set(self.__dict__.keys()) & set(another.__dict__.keys())
for k in common_keys:
setattr(self, k, getattr(another, k))
def propagate(self):
"""push down the configuration to all members"""
for k, v in self.__dict__.items():
if isinstance(v, BaseConfig):
v.inherit(self)
v.propagate()
def save(self, save_path):
"""save config to json file"""
dirname = os.path.dirname(save_path)
if not os.path.exists(dirname):
os.makedirs(dirname)
conf = self.as_dict_jsonable()
with open(save_path, 'w') as f:
json.dump(conf, f)
def load(self, load_path):
"""load json config"""
with open(load_path) as f:
conf = json.load(f)
self.from_dict(conf)
def from_dict(self, dict, strict=False):
for k, v in dict.items():
if not hasattr(self, k):
if strict:
raise ValueError(f"loading extra '{k}'")
else:
print(f"loading extra '{k}'")
continue
if isinstance(self.__dict__[k], BaseConfig):
self.__dict__[k].from_dict(v)
else:
self.__dict__[k] = v
def as_dict_jsonable(self):
conf = {}
for k, v in self.__dict__.items():
if isinstance(v, BaseConfig):
conf[k] = v.as_dict_jsonable()
else:
if jsonable(v):
conf[k] = v
else:
# ignore not jsonable
pass
return conf
def jsonable(x):
try:
json.dumps(x)
return True
except TypeError:
return False
|