Merge pull request #63 from alphagov/read-environment-from-file

Added code to read environemnt from a file

config.py

@@ -36,12 +36,22 @@ class Test(Development):
     SQLALCHEMY_DATABASE_URI = 'postgresql://localhost/test_notification_api'
 
 
+class Preview(Config):
+    pass
+
+
+class Staging(Config):
+    pass
+
+
 class Live(Config):
     pass
 
 
 configs = {
     'development': Development,
+    'preview': Preview,
+    'staging': Staging,
     'test': Test,
     'live': Live,
 }

wsgi.py

@@ -1,9 +1,17 @@
 from app import create_app
 from credstash import getAllSecrets
+import os
+
+config = 'live'
+default_env_file = '/home/ubuntu/environment'
+
+if os.path.isfile(default_env_file):
+        environment = open(default_env_file, 'r')
+        config = environment.readline().strip()
 
 secrets = getAllSecrets(region="eu-west-1")
 
-application = create_app('live', secrets)
+application = create_app(config, secrets)
 
 if __name__ == "__main__":
         application.run()