105 lines
4.2 KiB
Python
105 lines
4.2 KiB
Python
from django.conf import settings
|
|
from django.core import signals
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.db.utils import ConnectionHandler, ConnectionRouter, load_backend, DEFAULT_DB_ALIAS, \
|
|
DatabaseError, IntegrityError
|
|
from django.utils.functional import curry
|
|
|
|
__all__ = ('backend', 'connection', 'connections', 'router', 'DatabaseError',
|
|
'IntegrityError', 'DEFAULT_DB_ALIAS')
|
|
|
|
|
|
# For backwards compatibility - Port any old database settings over to
|
|
# the new values.
|
|
if not settings.DATABASES:
|
|
if settings.DATABASE_ENGINE:
|
|
import warnings
|
|
warnings.warn(
|
|
"settings.DATABASE_* is deprecated; use settings.DATABASES instead.",
|
|
DeprecationWarning
|
|
)
|
|
|
|
settings.DATABASES[DEFAULT_DB_ALIAS] = {
|
|
'ENGINE': settings.DATABASE_ENGINE,
|
|
'HOST': settings.DATABASE_HOST,
|
|
'NAME': settings.DATABASE_NAME,
|
|
'OPTIONS': settings.DATABASE_OPTIONS,
|
|
'PASSWORD': settings.DATABASE_PASSWORD,
|
|
'PORT': settings.DATABASE_PORT,
|
|
'USER': settings.DATABASE_USER,
|
|
'TEST_CHARSET': settings.TEST_DATABASE_CHARSET,
|
|
'TEST_COLLATION': settings.TEST_DATABASE_COLLATION,
|
|
'TEST_NAME': settings.TEST_DATABASE_NAME,
|
|
}
|
|
|
|
if DEFAULT_DB_ALIAS not in settings.DATABASES:
|
|
raise ImproperlyConfigured("You must define a '%s' database" % DEFAULT_DB_ALIAS)
|
|
|
|
for alias, database in settings.DATABASES.items():
|
|
if 'ENGINE' not in database:
|
|
raise ImproperlyConfigured("You must specify a 'ENGINE' for database '%s'" % alias)
|
|
if database['ENGINE'] in ("postgresql", "postgresql_psycopg2", "sqlite3", "mysql", "oracle"):
|
|
import warnings
|
|
if 'django.contrib.gis' in settings.INSTALLED_APPS:
|
|
warnings.warn(
|
|
"django.contrib.gis is now implemented as a full database backend. "
|
|
"Modify ENGINE in the %s database configuration to select "
|
|
"a backend from 'django.contrib.gis.db.backends'" % alias,
|
|
DeprecationWarning
|
|
)
|
|
if database['ENGINE'] == 'postgresql_psycopg2':
|
|
full_engine = 'django.contrib.gis.db.backends.postgis'
|
|
elif database['ENGINE'] == 'sqlite3':
|
|
full_engine = 'django.contrib.gis.db.backends.spatialite'
|
|
else:
|
|
full_engine = 'django.contrib.gis.db.backends.%s' % database['ENGINE']
|
|
else:
|
|
warnings.warn(
|
|
"Short names for ENGINE in database configurations are deprecated. "
|
|
"Prepend %s.ENGINE with 'django.db.backends.'" % alias,
|
|
DeprecationWarning
|
|
)
|
|
full_engine = "django.db.backends.%s" % database['ENGINE']
|
|
database['ENGINE'] = full_engine
|
|
|
|
connections = ConnectionHandler(settings.DATABASES)
|
|
|
|
router = ConnectionRouter(settings.DATABASE_ROUTERS)
|
|
|
|
# `connection`, `DatabaseError` and `IntegrityError` are convenient aliases
|
|
# for backend bits.
|
|
|
|
# DatabaseWrapper.__init__() takes a dictionary, not a settings module, so
|
|
# we manually create the dictionary from the settings, passing only the
|
|
# settings that the database backends care about. Note that TIME_ZONE is used
|
|
# by the PostgreSQL backends.
|
|
# we load all these up for backwards compatibility, you should use
|
|
# connections['default'] instead.
|
|
connection = connections[DEFAULT_DB_ALIAS]
|
|
backend = load_backend(connection.settings_dict['ENGINE'])
|
|
|
|
# Register an event that closes the database connection
|
|
# when a Django request is finished.
|
|
def close_connection(**kwargs):
|
|
for conn in connections.all():
|
|
conn.close()
|
|
signals.request_finished.connect(close_connection)
|
|
|
|
# Register an event that resets connection.queries
|
|
# when a Django request is started.
|
|
def reset_queries(**kwargs):
|
|
for conn in connections.all():
|
|
conn.queries = []
|
|
signals.request_started.connect(reset_queries)
|
|
|
|
# Register an event that rolls back the connections
|
|
# when a Django request has an exception.
|
|
def _rollback_on_exception(**kwargs):
|
|
from django.db import transaction
|
|
for conn in connections:
|
|
try:
|
|
transaction.rollback_unless_managed(using=conn)
|
|
except DatabaseError:
|
|
pass
|
|
signals.got_request_exception.connect(_rollback_on_exception)
|