Merge 4db2778 into ce04c44

lorinkoz · Feb 7, 2022 · d108edf · d108edf
2 parents ce04c44 + 4db2778
commit d108edf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/django_pgschemas/postgresql_backend/introspection.py b/django_pgschemas/postgresql_backend/introspection.py
@@ -2,10 +2,15 @@
 from django.utils.encoding import force_text
 
 from . import _constraints
-from .settings import original_backend
+from .settings import base_backend, original_backend
 
+try:
+    DatabaseIntrospection = original_backend.DatabaseIntrospection
+except AttributeError:
+    DatabaseIntrospection = base_backend.DatabaseIntrospection
 
-class DatabaseSchemaIntrospection(original_backend.DatabaseIntrospection):  # pragma: no cover
+
+class DatabaseSchemaIntrospection(DatabaseIntrospection):  # pragma: no cover
     """
     database schema introspection class
     """

diff --git a/django_pgschemas/postgresql_backend/settings.py b/django_pgschemas/postgresql_backend/settings.py
@@ -2,7 +2,9 @@
 
 from django.conf import settings
 
-ORIGINAL_BACKEND = getattr(settings, "PGSCHEMAS_ORIGINAL_BACKEND", "django.db.backends.postgresql")
+BASE_BACKEND = "django.db.backends.postgresql"
+ORIGINAL_BACKEND = getattr(settings, "PGSCHEMAS_ORIGINAL_BACKEND", BASE_BACKEND)
 EXTRA_SEARCH_PATHS = getattr(settings, "PGSCHEMAS_EXTRA_SEARCH_PATHS", [])
 
+base_backend = import_module(BASE_BACKEND + ".base")
 original_backend = import_module(ORIGINAL_BACKEND + ".base")