holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

get_config: return default value if conf file is corrupted

Browse Source
This commit is contained in:
JeLuF 2023-05-22 10:21:19 +02:00
parent 566cb55f36
commit 0f6caaec33
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/get_config.py
View File

@ -1,5 +1,6 @@
import os import os
import argparse import argparse
import sys
# The config file is in the same directory as this script # The config file is in the same directory as this script
config_directory = os.path.dirname(__file__) config_directory = os.path.dirname(__file__)
@ -21,16 +22,16 @@ if os.path.isfile(config_yaml):
try: try:
config = yaml.safe_load(configfile) config = yaml.safe_load(configfile)
except Exception as e: except Exception as e:
print(e) print(e, file=sys.stderr)
exit() config = {}
elif os.path.isfile(config_json): elif os.path.isfile(config_json):
import json import json
with open(config_json, 'r') as configfile: with open(config_json, 'r') as configfile:
try: try:
config = json.load(configfile) config = json.load(configfile)
except Exception as e: except Exception as e:
print(e) print(e, file=sys.stderr)
exit() config = {}
else: else:
config = {} config = {}