mirror of
https://github.com/jazzband/dj-database-url.git
synced 2026-03-16 22:20:24 +00:00
b83dfb52ce
Apparently I was targetting an older version of python 2.7 where urlparse worked differently
104 lines
2.8 KiB
Python
104 lines
2.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('spatialite')
|
|
urlparse.uses_netloc.append('sqlite')
|
|
|
|
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',
|
|
'spatialite': 'django.contrib.gis.db.backends.spatialite',
|
|
'sqlite': 'django.db.backends.sqlite3',
|
|
}
|
|
|
|
|
|
def config(env=DEFAULT_ENV, default=None, engine=None):
|
|
"""Returns configured DATABASE dictionary from DATABASE_URL."""
|
|
|
|
config = {}
|
|
|
|
s = os.environ.get(env, default)
|
|
|
|
if s:
|
|
config = parse(s, engine)
|
|
|
|
return config
|
|
|
|
|
|
def parse(url, engine=None):
|
|
"""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)
|
|
|
|
# Path (without leading '/'), and with no query string
|
|
path = url.path[1:].split('?')[0]
|
|
|
|
# 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': path or '',
|
|
'USER': url.username or '',
|
|
'PASSWORD': url.password or '',
|
|
'HOST': hostname,
|
|
'PORT': url.port or '',
|
|
})
|
|
|
|
# Parse the query string into OPTIONS.
|
|
qs = urlparse.parse_qs(url.query)
|
|
options = {}
|
|
for key, values in qs.iteritems():
|
|
options[key] = values[-1]
|
|
if options:
|
|
config['OPTIONS'] = options
|
|
|
|
if engine:
|
|
config['ENGINE'] = engine
|
|
elif url.scheme in SCHEMES:
|
|
config['ENGINE'] = SCHEMES[url.scheme]
|
|
|
|
return config
|