· 7 years ago · Feb 09, 2018, 02:32 PM
1from django.core.exceptions import ImproperlyConfigured
2
3def get_env_variable(var_name):
4 try:
5 return os.environ[var_name]
6 except KeyError:
7 error_msg = "Set the %s environment variable" % var_name
8 raise ImproperlyConfigured(error_msg)
9
10SECRET_KEY = get_env_variable('DJANGO_SECRET_KEY')
11DEBUG = get_env_variable('DJANGO_DEBUG')
12DATABASES = {
13 'default': {
14 'ENGINE': 'django.db.backends.postgresql_psycopg2',
15 'NAME': get_env_variable('DJANGO_POSTGRE_DB_NAME'),
16 'USER': get_env_variable('DJANGO_POSTGRE_USER'),
17 'PASSWORD': get_env_variable('DJANGO_POSTGRE_PASS'),
18 'HOST': get_env_variable('DJANGO_POSTGRE_HOST'),
19 'PORT': get_env_variable('DJANGO_POSTGRE_PORT'),
20 }
21}