From 631396e764a87e1254cf9146f56f2badc87d4c1c Mon Sep 17 00:00:00 2001
From: b3yond <b3yond@riseup.net>
Date: Tue, 6 Nov 2018 16:17:47 +0100
Subject: [PATCH] if no config.toml, set config through environment #64

---
 config.py | 56 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 54 insertions(+), 2 deletions(-)

diff --git a/config.py b/config.py
index fa01302..abbae62 100755
--- a/config.py
+++ b/config.py
@@ -1,5 +1,57 @@
 import pytoml as toml
+import os
+
+def load_env():
+    """
+    load environment variables from the environment. If empty, use default
+    values from config.toml.example.
+
+    :return: config dictionary of dictionaries.
+    """
+    with open('config.toml.example') as defaultconf:
+        config = toml.load(configfile)
+
+    try:
+        config['twitter']['consumer_key'] = os.environ['CONSUMER_KEY']
+    except KeyError:
+        pass
+
+    try:
+        config['twitter']['consumer_secret'] = os.environ['CONSUMER_SECRET']
+    except KeyError:
+        pass
+
+    try:
+        config['web']['host'] = os.environ['HOST']
+    except KeyError:
+        pass
+
+    try:
+        config['web']['port'] = os.environ['PORT']
+    except KeyError:
+        pass
+
+    try:
+        config['web']['contact'] = os.environ['CONTACT']
+    except KeyError:
+        pass
+
+    try:
+        config['mail']['mbox_user'] = os.environ['MBOX_USER']
+    except KeyError:
+        pass
+
+    try:
+        config['database']['db_path'] = os.environ['DB_PATH']
+    except KeyError:
+        pass
+
+    return config
+
 
 # read config in TOML format (https://github.com/toml-lang/toml#toml)
-with open('config.toml') as configfile:
-    config = toml.load(configfile)
+try:
+    with open('config.toml') as configfile:
+        config = toml.load(configfile)
+except FileNotFoundError:
+    config = load_env()