beauty overhaul of config.py

master
b3yond 2018-11-06 16:23:47 +01:00
parent e8ac1ca1c4
commit ffcb2506f6
1 changed files with 10 additions and 9 deletions

View File

@ -1,6 +1,7 @@
import pytoml as toml import pytoml as toml
import os import os
def load_env(): def load_env():
""" """
load environment variables from the environment. If empty, use default load environment variables from the environment. If empty, use default
@ -9,44 +10,44 @@ def load_env():
:return: config dictionary of dictionaries. :return: config dictionary of dictionaries.
""" """
with open('config.toml.example') as defaultconf: with open('config.toml.example') as defaultconf:
config = toml.load(defaultconf) configdict = toml.load(defaultconf)
try: try:
config['twitter']['consumer_key'] = os.environ['CONSUMER_KEY'] configdict['twitter']['consumer_key'] = os.environ['CONSUMER_KEY']
except KeyError: except KeyError:
pass pass
try: try:
config['twitter']['consumer_secret'] = os.environ['CONSUMER_SECRET'] configdict['twitter']['consumer_secret'] = os.environ['CONSUMER_SECRET']
except KeyError: except KeyError:
pass pass
try: try:
config['web']['host'] = os.environ['HOST'] configdict['web']['host'] = os.environ['HOST']
except KeyError: except KeyError:
pass pass
try: try:
config['web']['port'] = os.environ['PORT'] configdict['web']['port'] = os.environ['PORT']
except KeyError: except KeyError:
pass pass
try: try:
config['web']['contact'] = os.environ['CONTACT'] configdict['web']['contact'] = os.environ['CONTACT']
except KeyError: except KeyError:
pass pass
try: try:
config['mail']['mbox_user'] = os.environ['MBOX_USER'] configdict['mail']['mbox_user'] = os.environ['MBOX_USER']
except KeyError: except KeyError:
pass pass
try: try:
config['database']['db_path'] = os.environ['DB_PATH'] configdict['database']['db_path'] = os.environ['DB_PATH']
except KeyError: except KeyError:
pass pass
return config return configdict
# read config in TOML format (https://github.com/toml-lang/toml#toml) # read config in TOML format (https://github.com/toml-lang/toml#toml)