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

Added exception handling to config.yml parsing/merging

Browse Source
This commit is contained in:
Dispsylala 2019-10-29 22:38:41 +00:00
parent a897e00c98
commit bcdbf41bb8
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pwnagotchi/utils.py
View File

@ -65,12 +65,16 @@ def load_config(args):
config = yaml.safe_load(fp) config = yaml.safe_load(fp)
# load the user config # load the user config
try:
if os.path.exists(args.user_config): if os.path.exists(args.user_config):
with open(args.user_config) as fp: with open(args.user_config) as fp:
user_config = yaml.safe_load(fp) user_config = yaml.safe_load(fp)
# if the file is empty, safe_load will return None and merge_config will boom. # if the file is empty, safe_load will return None and merge_config will boom.
if user_config: if user_config:
config = merge_config(user_config, config) config = merge_config(user_config, config)
except yaml.YAMLError as ex:
print("There was an error processing the configuration file:\n%s " % ex)
exit(1)
# the very first step is to normalize the display name so we don't need dozens of if/elif around # the very first step is to normalize the display name so we don't need dozens of if/elif around
if config['ui']['display']['type'] in ('inky', 'inkyphat'): if config['ui']['display']['type'] in ('inky', 'inkyphat'):