加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
config.py 1.66 KB
一键复制 编辑 原始数据 按行查看 历史
程磊 提交于 2022-04-01 15:37 . Initial commit
import argparse
import json
from utils import dotdict
def make_as_dotdict(obj):
if type(obj) is dict:
obj = dotdict(obj)
for key in obj:
if type(obj[key]) is dict:
obj[key] = make_as_dotdict(obj[key])
return obj
def parse():
print('Parsing config file...')
parser = argparse.ArgumentParser(description="config")
parser.add_argument(
"--config",
type=str,
default="configs/base.json",
help="Configuration file to use"
)
cli_args = parser.parse_args()
with open(cli_args.config) as fp:
config = make_as_dotdict(json.loads(fp.read()))
print(json.dumps(config, indent=4, sort_keys=True))
return config
class Singleton(object):
_instance = None
def __new__(cls, *args, **kw):
if not cls._instance:
cls._instance = super(Singleton, cls).__new__(cls, *args, **kw)
return cls._instance
class Config(Singleton):
def __init__(self):
self._cfg = dotdict({})
try:
self._cfg = parse()
except:
pass
def __getattr__(self, name):
if name == '_cfg':
super().__setattr__(name)
else:
return self._cfg.__getattr__(name)
def __setattr__(self, name, val):
if name == '_cfg':
super().__setattr__(name, val)
else:
self._cfg.__setattr__(name, val)
def __delattr__(self, name):
return self._cfg.__delitem__(name)
def copy(self, new_config):
self._cfg = make_as_dotdict(new_config)
cfg = Config()
def parse_from_dict(d):
global cfg
assert type(d) == dict
cfg.copy(d)
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化