Add compatibility for py-django-debug-toolbar >= 4.4.6

flavors · Oct 7, 2024 · 3a62439 · 3a62439
1 parent 76f7d50
commit 3a62439
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/graphiql_debug_toolbar/middleware.py b/graphiql_debug_toolbar/middleware.py
@@ -4,10 +4,17 @@
 from django.template.loader import render_to_string
 from django.utils.encoding import force_str
 
-from debug_toolbar.middleware import _HTML_TYPES
+from graphene_django.views import GraphQLView
+
+import debug_toolbar
+
+if debug_toolbar.VERSION < "4.4.6":
+    from debug_toolbar.middleware import _HTML_TYPES
+else:
+    from debug_toolbar.utils import _HTML_TYPES
+
 from debug_toolbar.middleware import DebugToolbarMiddleware as BaseMiddleware
 from debug_toolbar.toolbar import DebugToolbar
-from graphene_django.views import GraphQLView
 
 from .serializers import CallableJSONEncoder