Merge pull request #54 from OzNetNerd/fix/53/choose-config-class

Select prod or debug class based on environment variable - Issue #53

source/flask_app.py

@@ -1,14 +1,23 @@
-from config import DebugConfig
+from config import DebugConfig, ProductionConfig
 from flask import Flask
 from flask_migrate import Migrate
 from importlib import import_module
+from os import environ
 from os.path import abspath, dirname, join, pardir
 import sys
 import logging
 
 # prevent python from writing *.pyc files / __pycache__ folders
 sys.dont_write_bytecode = True
 
+get_config_mode = environ.get('ENMS_CONFIG_MODE', 'Production')
+format_config_mode = "".join([get_config_mode.capitalize(), "Config"])
+
+try:
+    config_mode = getattr(sys.modules[__name__], format_config_mode)
+except AttributeError:
+    sys.exit('Error: Invalid ENMS_CONFIG_MODE environment variable entry.')
+
 path_source = dirname(abspath(__file__))
 path_parent = abspath(join(path_source, pardir))
 if path_source not in sys.path:
@@ -99,7 +108,7 @@ def configure_scheduler(scheduler):
 
 def create_app(test=False):
     app = Flask(__name__, static_folder='base/static')
-    app.config.from_object(DebugConfig)
+    app.config.from_object(config_mode)
     initialize_paths(app)
     register_extensions(app, test)
     register_blueprints(app)