File size: 1,352 Bytes
3ef1661 |
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 |
from types import ModuleType
import data_server_info # data infomation on some server
def load_data_info(module_name, data_info={}, db_type='db_info', module=None):
if module is None:
module = globals().get(module_name, None)
if module:
for key, value in module.__dict__.items():
if not (key.startswith('__')) and not (key.startswith('_')):
if key == 'db_info':
data_info.update(value)
elif isinstance(value, ModuleType):
load_data_info(module_name + '.' + key, data_info, module=value)
else:
raise RuntimeError(f'Try to access "db_info", but cannot find {module_name} module.')
def reset_ckpt_path(cfg, data_info):
if isinstance(cfg, dict):
for key in cfg.keys():
if key == 'backbone':
new_ckpt_path = data_info['checkpoint']['db_root'] + '/' + data_info['checkpoint'][cfg.backbone.type]
cfg.backbone.update(checkpoint=new_ckpt_path)
continue
elif isinstance(cfg.get(key), dict):
reset_ckpt_path(cfg.get(key), data_info)
else:
continue
else:
return
if __name__ == '__main__':
db_info_tmp = {}
load_data_info('db_data_info', db_info_tmp)
print('results', db_info_tmp.keys())
|