scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Convert keys to str

Browse Source
This commit is contained in:
dadav 2020-01-17 19:11:05 +01:00
parent 0764304be9
commit fae6a0942b
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pwnagotchi/utils.py
View File

@ -28,6 +28,26 @@ def merge_config(user, default):
user[k] = merge_config(user[k], v) user[k] = merge_config(user[k], v)
return user return user
def keys_to_str(data):
if isinstance(data,list):
converted_list = list()
for item in data:
if isinstance(item,dict):
converted_list.append(keys_to_str(item))
elif isinstance(item,list):
converted_list.append(keys_to_str(item))
else:
converted_list.append(item)
return converted_list
converted_dict = dict()
for key, value in data.items():
if isinstance(value, list) or isinstance(value, dict):
converted_dict[str(key)] = keys_to_str(value)
else:
converted_dict[str(key)] = value
return converted_dict
def load_config(args): def load_config(args):
default_config_path = os.path.dirname(args.config) default_config_path = os.path.dirname(args.config)
@ -83,6 +103,8 @@ def load_config(args):
logging.info('Old yaml-config found. Converting to toml...') logging.info('Old yaml-config found. Converting to toml...')
with open(args.user_config, 'w') as toml_file, open(yaml_name) as yaml_file: with open(args.user_config, 'w') as toml_file, open(yaml_name) as yaml_file:
user_config = yaml.safe_load(yaml_file) user_config = yaml.safe_load(yaml_file)
# convert int/float keys to str
user_config = keys_to_str(user_config)
# convert to toml but use loaded yaml # convert to toml but use loaded yaml
toml.dump(user_config, toml_file) toml.dump(user_config, toml_file)
elif os.path.exists(args.user_config): elif os.path.exists(args.user_config):