Bring DB script into line with other prod scripts

This commit is contained in:
Martyn Inglis
2016-03-18 09:55:19 +00:00
parent 8871d4eda6
commit fc7ad7d556

16
db.py
View File

@@ -4,9 +4,19 @@ from app import create_app, db
from credstash import getAllSecrets
import os
secrets = getAllSecrets(region="eu-west-1")
for key, val in secrets.items():
os.environ[key] = val
default_env_file = '/home/ubuntu/environment'
environment = 'live'
if os.path.isfile(default_env_file):
with open(default_env_file, 'r') as environment_file:
environment = environment_file.readline().strip()
# on aws get secrets and export to env
os.environ.update(getAllSecrets(region="eu-west-1"))
from config import configs
os.environ['NOTIFY_API_ENVIRONMENT'] = configs[environment]
application = create_app()