mirror of
https://github.com/jazzband/dj-database-url.git
synced 2026-03-16 22:20:24 +00:00
89a55460d6
In addition to postgres, redshift database is also support schema. Respect it by 'currentSchema' param. Also, we need to not include this param as in in Django's DATABASES settings. Otherwise, following error is raised: ``` django.db.utils.OperationalError: invalid connection option "currentSchema" ```
129 lines
3.8 KiB
Python
129 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
|
|
try:
|
|
import urlparse
|
|
except ImportError:
|
|
import urllib.parse as urlparse
|
|
|
|
|
|
# Register database schemes in URLs.
|
|
urlparse.uses_netloc.append('postgres')
|
|
urlparse.uses_netloc.append('postgresql')
|
|
urlparse.uses_netloc.append('pgsql')
|
|
urlparse.uses_netloc.append('postgis')
|
|
urlparse.uses_netloc.append('mysql')
|
|
urlparse.uses_netloc.append('mysql2')
|
|
urlparse.uses_netloc.append('mysqlgis')
|
|
urlparse.uses_netloc.append('mysql-connector')
|
|
urlparse.uses_netloc.append('spatialite')
|
|
urlparse.uses_netloc.append('sqlite')
|
|
urlparse.uses_netloc.append('oracle')
|
|
urlparse.uses_netloc.append('oraclegis')
|
|
urlparse.uses_netloc.append('redshift')
|
|
|
|
DEFAULT_ENV = 'DATABASE_URL'
|
|
|
|
SCHEMES = {
|
|
'postgres': 'django.db.backends.postgresql_psycopg2',
|
|
'postgresql': 'django.db.backends.postgresql_psycopg2',
|
|
'pgsql': 'django.db.backends.postgresql_psycopg2',
|
|
'postgis': 'django.contrib.gis.db.backends.postgis',
|
|
'mysql': 'django.db.backends.mysql',
|
|
'mysql2': 'django.db.backends.mysql',
|
|
'mysqlgis': 'django.contrib.gis.db.backends.mysql',
|
|
'mysql-connector': 'mysql.connector.django',
|
|
'spatialite': 'django.contrib.gis.db.backends.spatialite',
|
|
'sqlite': 'django.db.backends.sqlite3',
|
|
'oracle': 'django.db.backends.oracle',
|
|
'oraclegis': 'django.contrib.gis.db.backends.oracle',
|
|
'redshift': 'django_redshift_backend',
|
|
}
|
|
|
|
|
|
def config(env=DEFAULT_ENV, default=None, engine=None, conn_max_age=0):
|
|
"""Returns configured DATABASE dictionary from DATABASE_URL."""
|
|
|
|
config = {}
|
|
|
|
s = os.environ.get(env, default)
|
|
|
|
if s:
|
|
config = parse(s, engine, conn_max_age)
|
|
|
|
return config
|
|
|
|
|
|
def parse(url, engine=None, conn_max_age=0):
|
|
"""Parses a database URL."""
|
|
|
|
if url == 'sqlite://:memory:':
|
|
# this is a special case, because if we pass this URL into
|
|
# urlparse, urlparse will choke trying to interpret "memory"
|
|
# as a port number
|
|
return {
|
|
'ENGINE': SCHEMES['sqlite'],
|
|
'NAME': ':memory:'
|
|
}
|
|
# note: no other settings are required for sqlite
|
|
|
|
# otherwise parse the url as normal
|
|
config = {}
|
|
|
|
url = urlparse.urlparse(url)
|
|
|
|
# Split query strings from path.
|
|
path = url.path[1:]
|
|
if '?' in path and not url.query:
|
|
path, query = path.split('?', 2)
|
|
else:
|
|
path, query = path, url.query
|
|
query = urlparse.parse_qs(query)
|
|
|
|
# If we are using sqlite and we have no path, then assume we
|
|
# want an in-memory database (this is the behaviour of sqlalchemy)
|
|
if url.scheme == 'sqlite' and path == '':
|
|
path = ':memory:'
|
|
|
|
# Handle postgres percent-encoded paths.
|
|
hostname = url.hostname or ''
|
|
if '%2f' in hostname.lower():
|
|
hostname = hostname.replace('%2f', '/').replace('%2F', '/')
|
|
|
|
# Update with environment configuration.
|
|
config.update({
|
|
'NAME': urlparse.unquote(path or ''),
|
|
'USER': urlparse.unquote(url.username or ''),
|
|
'PASSWORD': urlparse.unquote(url.password or ''),
|
|
'HOST': hostname,
|
|
'PORT': url.port or '',
|
|
'CONN_MAX_AGE': conn_max_age,
|
|
})
|
|
|
|
# Lookup specified engine.
|
|
engine = SCHEMES[url.scheme] if engine is None else engine
|
|
|
|
# Pass the query string into OPTIONS.
|
|
options = {}
|
|
for key, values in query.items():
|
|
if url.scheme == 'mysql' and key == 'ssl-ca':
|
|
options['ssl'] = {'ca': values[-1]}
|
|
continue
|
|
|
|
options[key] = values[-1]
|
|
|
|
# Support for Postgres Schema URLs
|
|
if 'currentSchema' in options and engine in (
|
|
'django.db.backends.postgresql_psycopg2',
|
|
'django_redshift_backend',
|
|
):
|
|
options['options'] = '-c search_path={0}'.format(options.pop('currentSchema'))
|
|
|
|
if options:
|
|
config['OPTIONS'] = options
|
|
|
|
if engine:
|
|
config['ENGINE'] = engine
|
|
|
|
return config
|