Make compatible with older pytest-django

businho · Feb 7, 2024 · db37be3 · db37be3
1 parent b2a001b
commit db37be3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/django_migrations_ci/pytest_plugin.py b/django_migrations_ci/pytest_plugin.py
@@ -31,13 +31,16 @@ def pytest_configure(config):
         return
 
     try:
-        from pytest_django.plugin import blocking_manager_key
+        from pytest_django import plugin
 
-        blocking_manager = config.stash[blocking_manager_key]
-        db_unblock = blocking_manager.unblock
+        blocking_manager = config.stash[plugin.blocking_manager_key]
+    except AttributeError:
+        blocking_manager = plugin._blocking_manager
     except ImportError:
         # The pytest-django lib is not installed, do nothing and hope for the best.
         db_unblock = nullcontext
+    else:
+        db_unblock = blocking_manager.unblock
 
     verbosity = config.option.migrateci_verbose
     if verbosity is None: