Ticket #10487: conn.2.diff

django/db/__init__.py

@@ -9 +9 @@
 if not settings.DATABASE_ENGINE:
     settings.DATABASE_ENGINE = 'dummy'
 
-try:
-    # Most of the time, the database backend will be one of the official
-    # backends that ships with Django, so look there first.
-    _import_path = 'django.db.backends.'
-    backend = __import__('%s%s.base' % (_import_path, settings.DATABASE_ENGINE), {}, {}, [''])
-except ImportError, e:
-    # If the import failed, we might be looking for a database backend
-    # distributed external to Django. So we'll try that next.
+def load_backend(path):
     try:
-        _import_path = ''
-        backend = __import__('%s.base' % settings.DATABASE_ENGINE, {}, {}, [''])
-    except ImportError, e_user:
-        # The database backend wasn't found. Display a helpful error message
-        # listing all possible (built-in) database backends.
-        backend_dir = os.path.join(__path__[0], 'backends')
+        # Most of the time, the database backend will be one of the official
+        # backends that ships with Django, so look there first.
+        _import_path = 'django.db.backends.'
+        return __import__('%s%s.base' % (_import_path, path), {}, {}, [''])
+    except ImportError, e:
+        # If the import failed, we might be looking for a database backend
+        # distributed external to Django. So we'll try that next.
         try:
-            available_backends = [f for f in os.listdir(backend_dir)
-                    if os.path.isdir(os.path.join(backend_dir, f))]
-        except EnvironmentError:
-            available_backends = []
-        available_backends.sort()
-        if settings.DATABASE_ENGINE not in available_backends:
-            raise ImproperlyConfigured, "%r isn't an available database backend. Available options are: %s\nError was: %s" % \
-                (settings.DATABASE_ENGINE, ", ".join(map(repr, available_backends)), e_user)
-        else:
-            raise # If there's some other error, this must be an error in Django itself.
+            _import_path = ''
+            return __import__('%s.base' % path, {}, {}, [''])
+        except ImportError, e_user:
+            # The database backend wasn't found. Display a helpful error message
+            # listing all possible (built-in) database backends.
+            backend_dir = os.path.join(__path__[0], 'backends')
+            try:
+                available_backends = [f for f in os.listdir(backend_dir)
+                        if os.path.isdir(os.path.join(backend_dir, f))]
+            except EnvironmentError:
+                available_backends = []
+            available_backends.sort()
+            if path not in available_backends:
+                raise ImproperlyConfigured, "%r isn't an available database backend. Available options are: %s\nError was: %s" % \
+                    (path, ", ".join(map(repr, available_backends)), e_user)
+            else:
+                raise # If there's some other error, this must be an error in Django itself.
 
 # `connection`, `DatabaseError` and `IntegrityError` are convenient aliases
 # for backend bits.
 
+backend = load_backend(settings.DATABASE_ENGINE)
+
 # 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.
+
 connection = backend.DatabaseWrapper({
     'DATABASE_HOST': settings.DATABASE_HOST,
     'DATABASE_NAME': settings.DATABASE_NAME,