forked from extern/easydiffusion
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
|
import os
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
# The config file is in the same directory as this script
|
||
|
|
config_directory = os.path.dirname(__file__)
|
||
|
|
config_yaml = os.path.join(config_directory, "config.yaml")
|
||
|
|
config_json = os.path.join(config_directory, "config.json")
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Get values from config file')
|
||
|
|
parser.add_argument('--default', dest='default', action='store',
|
||
|
|
help='default value, to be used if the setting is not defined in the config file')
|
||
|
|
parser.add_argument('key', metavar='key', nargs='+',
|
||
|
|
help='config key to return')
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
if os.path.isfile(config_yaml):
|
||
|
|
import yaml
|
||
|
|
with open(config_yaml, 'r') as configfile:
|
||
|
|
try:
|
||
|
|
config = yaml.safe_load(configfile)
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
exit()
|
||
|
|
elif os.path.isfile(config_json):
|
||
|
|
import json
|
||
|
|
with open(config_json, 'r') as configfile:
|
||
|
|
try:
|
||
|
|
config = json.load(configfile)
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
exit()
|
||
|
|
else:
|
||
|
|
config = {}
|
||
|
|
|
||
|
|
for k in args.key:
|
||
|
|
if k in config:
|
||
|
|
config = config[k]
|
||
|
|
else:
|
||
|
|
if args.default != None:
|
||
|
|
print(args.default)
|
||
|
|
exit()
|
||
|
|
|
||
|
|
print(config)
|