Dynamic values for root_tag_extra_attrs

Author: gone

debug_toolbar/templates/debug_toolbar/base.html

@@ -16,7 +16,7 @@
      data-render-panel-url="{% url 'djdt:render_panel' %}"
      {% endif %}
      data-default-show="{% if toolbar.config.SHOW_COLLAPSED %}false{% else %}true{% endif %}"
-     {{ toolbar.config.ROOT_TAG_EXTRA_ATTRS|safe }}>
+     {{ toolbar.root_tag_extra_attrs|safe }}>
   <div class="djdt-hidden" id="djDebugToolbar">
     <ul id="djDebugPanelList">
       <li><a id="djHideToolBarButton" href="#" title="{% trans 'Hide toolbar' %}">{% trans "Hide" %} »</a></li>

debug_toolbar/toolbar.py

@@ -34,6 +34,7 @@ def __init__(self, request, get_response):
         self.stats = {}
         self.server_timing_stats = {}
         self.store_id = None
+        self.root_tag_extra_attrs = self.config["ROOT_TAG_EXTRA_ATTRS"]
         self.should_render_css = self.should_render_js = True
 
     # Manage panels

tests/test_integration.py

@@ -121,6 +121,25 @@ def configure_toolbar(self, request, toolbar):
         self.assertNotContains(response, b"<link")
         self.assertNotContains(response, b"<script")
 
+    def test_dynamic_root_extra_attr(self):
+        def get_response(request):
+            return regular_view(request, "nani")
+
+        class DynamicExtraAttrMiddleware(DebugToolbarMiddleware):
+            def configure_toolbar(self, request, toolbar):
+                if request.headers.get("test") == "omae":
+                    toolbar.root_tag_extra_attrs = "omae wa"
+                if request.headers.get("test") == "mou":
+                    toolbar.root_tag_extra_attrs = "mou shindeiru"
+
+        request = rf.get("/", HTTP_TEST="omae")
+        response = DynamicExtraAttrMiddleware(get_response)(request)
+        self.assertContains(response, b"omae wa")
+
+        request = rf.get("/", HTTP_TEST="mou")
+        response = DynamicExtraAttrMiddleware(get_response)(request)
+        self.assertContains(response, b"mou shindeiru")
+
     def test_cache_page(self):
         # Clear the cache before testing the views. Other tests that use cached_view
         # may run earlier and cause fewer cache calls.