From 54801a7bb8521e7d80ce72124396e83b593eb398 Mon Sep 17 00:00:00 2001 From: John Jarvis Date: Thu, 27 Mar 2014 10:38:10 -0400 Subject: [PATCH] adding overrides for migrations --- xqueue/aws_migrate.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 xqueue/aws_migrate.py diff --git a/xqueue/aws_migrate.py b/xqueue/aws_migrate.py new file mode 100644 index 00000000..65f95662 --- /dev/null +++ b/xqueue/aws_migrate.py @@ -0,0 +1,20 @@ +from .aws_settings import * + +import os +from django.core.exceptions import ImproperlyConfigured + +DB_OVERRIDES = dict( + PASSWORD=os.environ.get('DB_MIGRATION_PASS', None), + ENGINE=os.environ.get('DB_MIGRATION_ENGINE', DATABASES['default']['ENGINE']), + USER=os.environ.get('DB_MIGRATION_USER', DATABASES['default']['USER']), + NAME=os.environ.get('DB_MIGRATION_NAME', DATABASES['default']['NAME']), + HOST=os.environ.get('DB_MIGRATION_HOST', DATABASES['default']['HOST']), + PORT=os.environ.get('DB_MIGRATION_PORT', DATABASES['default']['PORT']), +) + +if DB_OVERRIDES['PASSWORD'] is None: + raise ImproperlyConfigured("No database password was provided for running " + "migrations. This is fatal.") + +for override, value in DB_OVERRIDES.iteritems(): + DATABASES['default'][override] = value