Commit 2a550b1a authored by Kenneth Reitz's avatar Kenneth Reitz

subtle cleanups

parent 0512577a
...@@ -90,18 +90,20 @@ if [ "$NAME" = "Python/Django" ]; then ...@@ -90,18 +90,20 @@ if [ "$NAME" = "Python/Django" ]; then
cat >>$SETTINGS_FILE <<EOF cat >>$SETTINGS_FILE <<EOF
import os, sys, urlparse import os
import sys
import urlparse
urlparse.uses_netloc.append('postgres') urlparse.uses_netloc.append('postgres')
urlparse.uses_netloc.append('mysql') urlparse.uses_netloc.append('mysql')
try: try:
#check to make sure DATABASES is set in settings.py file.If not default to {} # Check to make sure DATABASES is set in settings.py file.
try: # If not default to {}
DATABASES
except NameError: if 'DATABASES' not in locals():
DATABASES = {} DATABASES = {}
if os.environ.has_key('DATABASE_URL'): if 'DATABASE_URL' os.environ:
url = urlparse.urlparse(os.environ['DATABASE_URL']) url = urlparse.urlparse(os.environ['DATABASE_URL'])
# We use update here to preserve other keys we # We use update here to preserve other keys we
...@@ -115,6 +117,7 @@ try: ...@@ -115,6 +117,7 @@ try:
}) })
if url.scheme == 'postgres': if url.scheme == 'postgres':
DATABASES['default']['ENGINE'] = 'django.db.backends.postgresql_psycopg2' DATABASES['default']['ENGINE'] = 'django.db.backends.postgresql_psycopg2'
if url.scheme == 'mysql': if url.scheme == 'mysql':
DATABASES['default']['ENGINE'] = 'django.db.backends.mysql' DATABASES['default']['ENGINE'] = 'django.db.backends.mysql'
except: except:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment