47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
#
|
|
import os, sys, json
|
|
from pathlib import Path
|
|
from collections import namedtuple
|
|
|
|
support_types = ('str', 'int', 'bool', 'float')
|
|
|
|
def convert_param(original_lists):
|
|
assert isinstance(original_lists, list), 'The type is not right : {:}'.format(original_lists)
|
|
ctype, value = original_lists[0], original_lists[1]
|
|
assert ctype in support_types, 'Ctype={:}, support={:}'.format(ctype, support_types)
|
|
is_list = isinstance(value, list)
|
|
if not is_list: value = [value]
|
|
outs = []
|
|
for x in value:
|
|
if ctype == 'int':
|
|
x = int(x)
|
|
elif ctype == 'str':
|
|
x = str(x)
|
|
elif ctype == 'bool':
|
|
x = bool(int(x))
|
|
elif ctype == 'float':
|
|
x = float(x)
|
|
else:
|
|
raise TypeError('Does not know this type : {:}'.format(ctype))
|
|
outs.append(x)
|
|
if not is_list: outs = outs[0]
|
|
return outs
|
|
|
|
def load_config(path):
|
|
path = str(path)
|
|
assert os.path.exists(path), 'Can not find {:}'.format(path)
|
|
# Reading data back
|
|
with open(path, 'r') as f:
|
|
data = json.load(f)
|
|
f.close()
|
|
content = { k: convert_param(v) for k,v in data.items()}
|
|
Arguments = namedtuple('Configure', ' '.join(content.keys()))
|
|
content = Arguments(**content)
|
|
return content
|