diff --git a/app/__init__.py b/app/__init__.py index fad8bd3e5..a83405bce 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -16,6 +16,8 @@ from app.clients.sms.loadtesting import LoadtestingClient from app.clients.email.aws_ses import AwsSesClient from app.clients.statsd.statsd_client import StatsdClient from app.encryption import Encryption +from config import configs + DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%f" DATE_FORMAT = "%Y-%m-%d" @@ -38,7 +40,7 @@ api_user = LocalProxy(lambda: _request_ctx_stack.top.api_user) def create_app(app_name=None): application = Flask(__name__) - application.config.from_object(os.environ['NOTIFY_API_ENVIRONMENT']) + application.config.from_object(configs[os.environ['NOTIFY_ENVIRONMENT']]) if app_name: application.config['NOTIFY_APP_NAME'] = app_name diff --git a/config.py b/config.py index 5f9de8349..4231a3b40 100644 --- a/config.py +++ b/config.py @@ -136,9 +136,9 @@ class Live(Config): configs = { - 'development': 'config.Development', - 'test': 'config.Test', - 'live': 'config.Live', - 'staging': 'config.Staging', - 'preview': 'config.Preview' + 'development': Development, + 'test': Test, + 'live': Live, + 'staging': Staging, + 'preview': Preview }