72 lines
2 KiB
Python
72 lines
2 KiB
Python
|
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:
|
||
|
|
pass
|
||
|
|
return conf
|
||
|
|
|
||
|
|
|
||
|
|
def jsonable(x):
|
||
|
|
try:
|
||
|
|
json.dumps(x)
|
||
|
|
return True
|
||
|
|
except TypeError:
|
||
|
|
return False
|