mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2024-11-26 02:05:09 +01:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import os
|
|
import argparse
|
|
import sys
|
|
|
|
# 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):
|
|
from ruamel.yaml import YAML
|
|
yaml = YAML(typ='safe')
|
|
with open(config_yaml, 'r') as configfile:
|
|
try:
|
|
config = yaml.load(configfile)
|
|
except Exception as e:
|
|
print(e, file=sys.stderr)
|
|
config = {}
|
|
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, file=sys.stderr)
|
|
config = {}
|
|
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)
|